Use generator_unordered

Quantco · Jul 17, 2024 · 1e945ba · 1e945ba
1 parent 6da4180
commit 1e945ba
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/metalearners/grid_search.py b/metalearners/grid_search.py
@@ -348,9 +348,7 @@ def fit(
                         metalerner_fit_params=kwargs,
                     )
                 )
-        return_as = (
-            "list" if self.store_raw_results else "generator"
-        )  # Can we use generator_unordered?
+        return_as = "list" if self.store_raw_results else "generator_unordered"
         parallel = Parallel(
             n_jobs=self.n_jobs, verbose=self.verbose, return_as=return_as
         )