From b1fd5b8ad9f2d9970f803866269e89ddd205a417 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Francesc=20Mart=C3=AD=20Escofet?= Date: Wed, 17 Jul 2024 17:11:31 +0200 Subject: [PATCH] Fix --- metalearners/grid_search.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/metalearners/grid_search.py b/metalearners/grid_search.py index 304beb5..16b120d 100644 --- a/metalearners/grid_search.py +++ b/metalearners/grid_search.py @@ -352,9 +352,9 @@ def fit( parallel = Parallel( n_jobs=self.n_jobs, verbose=self.verbose, return_as=return_as ) - self.raw_results = parallel(delayed(_fit_and_score)(job) for job in jobs) + self.raw_results_ = parallel(delayed(_fit_and_score)(job) for job in jobs) if self.store_results: - self.results_ = _format_results(results=self.raw_results) + self.results_ = _format_results(results=self.raw_results_) # type: ignore if not self.store_raw_results: # This just checks that the generator is empty try: