From a726cdbf26ae1371dcf7718cfce9591614f62f62 Mon Sep 17 00:00:00 2001 From: Younes Strittmatter Date: Fri, 10 Nov 2023 09:44:18 -0500 Subject: [PATCH] cli: bugfix import in tests --- src/autora/experimentalist/falsification/__init__.py | 2 +- tests/test_exp_falsification_sampler.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/autora/experimentalist/falsification/__init__.py b/src/autora/experimentalist/falsification/__init__.py index 635bf31..b8cf830 100644 --- 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") \ No newline at end of file diff --git a/tests/test_exp_falsification_sampler.py b/tests/test_exp_falsification_sampler.py index 8bc7a59..5a71a88 100644 --- 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,