test: fix mypy error

MilesCranmer · Nov 24, 2024 · e20d3e1 · e20d3e1
1 parent 5a63e88
commit e20d3e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysr/sr.py b/pysr/sr.py
@@ -2461,7 +2461,7 @@ def get_hof(self, search_output: Optional[Any] = None):
         expression_options = self.expression_options or ExpressionOptions()
         ret_outputs = [
             expression_options.create_exports(self, output, search_output)
-            for output in self.equation_file_contents_
+            for output in cast(List[pd.DataFrame], self.equation_file_contents_)
         ]
 
         if self.nout_ > 1: