diff --git a/CHANGELOG.rst b/CHANGELOG.rst index efa47e4..91ce981 100644 --- a/CHANGELOG.rst +++ b/CHANGELOG.rst @@ -10,7 +10,7 @@ Changelog 0.6.0 (2024-06-**) ------------------ -* Implemented :class:`metalearners.metalearner_grid_search_cv.MetaLearnerGridSearchCV`. +* Implemented :class:`metalearners.grid_search.MetaLearnerGridSearchCV`. * Added ``scoring`` parameter to :meth:`metalearners.metalearner.MetaLearner.evaluate` and implemented the abstract method for the :class:`metalearners.XLearner` and diff --git a/conda.recipe/recipe.yaml b/conda.recipe/recipe.yaml index 192234c..3594845 100644 --- a/conda.recipe/recipe.yaml +++ b/conda.recipe/recipe.yaml @@ -45,6 +45,7 @@ tests: - metalearners.rlearner - metalearners.drlearner - metalearners.explainer + - metalearners.grid_search pip_check: true about: diff --git a/metalearners/metalearner_grid_search_cv.py b/metalearners/grid_search.py similarity index 99% rename from metalearners/metalearner_grid_search_cv.py rename to metalearners/grid_search.py index e38a99f..78ba23a 100644 --- a/metalearners/metalearner_grid_search_cv.py +++ b/metalearners/grid_search.py @@ -153,10 +153,9 @@ class MetaLearnerGridSearchCV: For how to define ``scoring`` check :meth:`~metalearners.metalearner.MetaLearner.evaluate`. ``verbose`` will be passed to `joblib.Parallel `_. - """ - -# TODO: Add a reference to a docs example once it is written. + + # TODO: Add a reference to a docs example once it is written. def __init__( self, diff --git a/tests/test_metalearner_grid_search_cv.py b/tests/test_grid_search.py similarity index 97% rename from tests/test_metalearner_grid_search_cv.py rename to tests/test_grid_search.py index e992ce5..5fff566 100644 --- a/tests/test_metalearner_grid_search_cv.py +++ b/tests/test_grid_search.py @@ -7,7 +7,7 @@ from sklearn.linear_model import LinearRegression, LogisticRegression from metalearners.drlearner import DRLearner -from metalearners.metalearner_grid_search_cv import MetaLearnerGridSearchCV +from metalearners.grid_search import MetaLearnerGridSearchCV from metalearners.rlearner import RLearner from metalearners.slearner import SLearner from metalearners.tlearner import TLearner