Merge pull request #24 from AutoResearch/23-bug-test-imports

cli: bugfix import in tests
AutoResearch · Nov 10, 2023 · 0994c52 · 0994c52
2 parents fe114f9 + a726cdb
commit 0994c52
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/autora/experimentalist/falsification/__init__.py b/src/autora/experimentalist/falsification/__init__.py
@@ -412,7 +412,7 @@ def falsification_score_sample_from_predictions(
 falsification_pooler = deprecated_alias(falsification_pool, "falsification_pooler")
 
 falsification_sample = sample
-falsification_pool.__doc__ = """Alias for pool"""
+falsification_pool.__doc__ = """Alias for sample"""
 falsification_sampler = deprecated_alias(falsification_sample, "falsification_sampler")
 falsification_score_sampler = deprecated_alias(falsification_score_sample, "falsification_score_sampler")
 falsification_score_sampler_from_predictions = deprecated_alias(falsification_score_sample_from_predictions, "falsification_score_sampler_from_predictions")
diff --git a/tests/test_exp_falsification_sampler.py b/tests/test_exp_falsification_sampler.py
@@ -5,7 +5,7 @@
 from sklearn.linear_model import LinearRegression, LogisticRegression
 
 from autora.experimentalist.pipeline import Pipeline
-from autora.experimentalist.pooler.grid import grid_pool
+from autora.experimentalist.grid import pool
 from autora.experimentalist.falsification import (
     falsification_sample,
     falsification_score_sample,
@@ -309,7 +309,7 @@ def test_iterator_input(synthetic_linr_model):
         dependent_variables=[dv],
     )
 
-    X = grid_pool(metadata.independent_variables)
+    X = pool(metadata)
 
     new_conditions = falsification_sample(
                 conditions=X,