test: fix mypy errors with sympy

pysr/export_jax.py

@@ -1,5 +1,5 @@
 import numpy as np  # noqa: F401
-import sympy
+import sympy  # type: ignore
 
 # Special since need to reduce arguments.
 MUL = 0

pysr/export_latex.py

@@ -3,8 +3,8 @@
 from typing import List, Optional, Tuple
 
 import pandas as pd
-import sympy
-from sympy.printing.latex import LatexPrinter
+import sympy  # type: ignore
+from sympy.printing.latex import LatexPrinter  # type: ignore
 
 
 class PreciseLatexPrinter(LatexPrinter):

pysr/export_numpy.py

@@ -6,7 +6,7 @@ from typing import List, Union
 import numpy as np
 import pandas as pd
 from numpy.typing import NDArray
-from sympy import Expr, Symbol, lambdify
+from sympy import Expr, Symbol, lambdify  # type: ignore
 
 
 def sympy2numpy(eqn, sympy_symbols, *, selection=None):

pysr/export_sympy.py

@@ -2,7 +2,7 @@
 
 from typing import Callable, Dict, List, Optional
 
-import sympy
+import sympy  # type: ignore
 from sympy import sympify
 
 from .utils import ArrayLike

pysr/export_torch.py

@@ -4,7 +4,7 @@ import collections as co
 import functools as ft
 
 import numpy as np  # noqa: F401
-import sympy
+import sympy  # type: ignore
 
 
 def _reduce(fn):

pysr/test/test.py

@@ -9,7 +9,7 @@ from pathlib import Path
 
 import numpy as np
 import pandas as pd
-import sympy
+import sympy  # type: ignore
 from sklearn.utils.estimator_checks import check_estimator
 
 from pysr import PySRRegressor, install, jl

pysr/test/test_jax.py

@@ -3,7 +3,7 @@ from functools import partial
 
 import numpy as np
 import pandas as pd
-import sympy
+import sympy  # type: ignore
 
 import pysr
 from pysr import PySRRegressor, sympy2jax
@@ -102,7 +102,7 @@ class TestJAX(unittest.TestCase):
         )
 
     def test_issue_656(self):
-        import sympy
+        import sympy  # type: ignore
 
         E_plus_x1 = sympy.exp(1) + sympy.symbols("x1")
         f, params = pysr.export_jax.sympy2jax(E_plus_x1, [sympy.symbols("x1")])

pysr/test/test_torch.py

@@ -2,7 +2,7 @@ import unittest
 
 import numpy as np
 import pandas as pd
-import sympy
+import sympy  # type: ignore
 
 import pysr
 from pysr import PySRRegressor, sympy2torch