test: fix type annotation

MilesCranmer · Nov 30, 2024 · 762d8f9 · 762d8f9
1 parent 95fb9d1
commit 762d8f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysr/sr.py b/pysr/sr.py
@@ -1080,7 +1080,7 @@ def from_file(
             assert unary_operators is None
             assert n_features_in is None
             with open(pkl_filename, "rb") as f:
-                model: "pysr.sr.PySRRegressor" = pkl.load(f)
+                model = cast("PySRRegressor", pkl.load(f))
 
             # Update any parameters if necessary, such as
             # extra_sympy_mappings: