Remove check empty generator

Quantco · Jul 22, 2024 · e34b751 · e34b751
1 parent 46a88cc
commit e34b751
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/metalearners/grid_search.py b/metalearners/grid_search.py
@@ -358,10 +358,7 @@ def fit(
         if self.store_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:
-                    next(self.raw_results_)  # type: ignore
-                except StopIteration:
-                    self.raw_results_ = None
+                # The generator will be empty so we replace it with None
+                self.raw_results_ = None
         else:
             self.results_ = None