Fix

Quantco · Jul 17, 2024 · b1fd5b8 · b1fd5b8
1 parent 4682e53
commit b1fd5b8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: