From e20d3e1272e15b1980074142197652dcddf0b6a8 Mon Sep 17 00:00:00 2001
From: MilesCranmer <miles.cranmer@gmail.com>
Date: Sun, 24 Nov 2024 20:01:00 +0000
Subject: [PATCH] test: fix mypy error

---
 pysr/sr.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pysr/sr.py b/pysr/sr.py
index 3d7b31d1..5ce2fd47 100644
--- 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: