Merge pull request #103 from mmcdermott/dev

Makes meds-transform and polars requirements more flexible.
mmcdermott · Dec 10, 2024 · 374e2f5 · 374e2f5
2 parents 97d5f5a + 01c0a52
commit 374e2f5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,8 +15,8 @@ classifiers = [
     "Operating System :: OS Independent",
 ]
 dependencies = [
-  "polars==1.6.0", "pyarrow", "loguru", "hydra-core==1.3.2", "numpy", "scipy<1.14.0", "pandas", "tqdm", "xgboost",
-  "scikit-learn", "hydra-optuna-sweeper", "hydra-joblib-launcher", "ml-mixins", "meds==0.3.3", "meds-transforms==0.0.7",
+  "polars>=1.6.0,<=1.17.1", "pyarrow", "loguru", "hydra-core==1.3.2", "numpy", "scipy<1.14.0", "pandas", "tqdm", "xgboost",
+  "scikit-learn", "hydra-optuna-sweeper", "hydra-joblib-launcher", "ml-mixins", "meds>=0.3.3", "meds-transforms>=0.0.7",
 ]
 
 [tool.setuptools_scm]

diff --git a/src/MEDS_tabular_automl/mapper.py b/src/MEDS_tabular_automl/mapper.py
@@ -163,10 +163,9 @@ def wrap(
         ...     lambda df: df.with_columns(pl.col("c") * 2),
         ...     lambda df: df.filter(pl.col("d") > 4)
         ... ]
-        >>> wrap(in_fp, out_fp, read_fn, write_fn, *transform_fns)
-        Traceback (most recent call last):
-            ...
-        polars.exceptions.ColumnNotFoundError: unable to find column "d"; valid columns: ["a", "b", "c"]
+        >>> import pytest
+        >>> with pytest.raises(Exception):
+        ...     wrap(in_fp, out_fp, read_fn, write_fn, *transform_fns)
         >>> assert cache_directory.is_dir()
         >>> cache_fp = cache_directory / "step_0.output"
         >>> pl.read_csv(cache_fp)