From 5e34a35682ed961e82776b1df4eadac7b69a757d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Francesc=20Mart=C3=AD=20Escofet?= <154450563+FrancescMartiEscofetQC@users.noreply.github.com> Date: Thu, 4 Jul 2024 12:05:28 +0200 Subject: [PATCH] Update metalearners/grid_search.py Co-authored-by: Kevin Klein <7267523+kklein@users.noreply.github.com> --- metalearners/grid_search.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metalearners/grid_search.py b/metalearners/grid_search.py index 6c7c3c2..90db94b 100644 --- a/metalearners/grid_search.py +++ b/metalearners/grid_search.py @@ -128,7 +128,7 @@ class MetaLearnerGridSearch: :meth:`~metalearners.metalearner.MetaLearner.treatment_model_specifications`. The values should be sequences of model factories. - If models are reused, they should be passed through ``metalearner_params`` and they + If base models are meant to be reused, they should be passed through ``metalearner_params`` and the corresponding keys should not be in ``base_learner_grid``. ``param_grid`` should contain the parameters grid for each type of model used by the