test: fix mypy errors

MilesCranmer · Dec 6, 2024 · 450e674 · 450e674
1 parent 678fce0
commit 450e674
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pysr/export_jax.py b/pysr/export_jax.py
@@ -1,6 +1,6 @@
 import numpy as np  # noqa: F401
 import sympy  # type: ignore
-from sympy.codegen.cfunctions import log2, log10
+from sympy.codegen.cfunctions import log2, log10  # type: ignore
 
 # Special since need to reduce arguments.
 MUL = 0

diff --git a/pysr/export_sympy.py b/pysr/export_sympy.py
@@ -4,7 +4,7 @@
 
 import sympy  # type: ignore
 from sympy import sympify
-from sympy.codegen.cfunctions import log2, log10
+from sympy.codegen.cfunctions import log2, log10  # type: ignore
 
 from .utils import ArrayLike
 

diff --git a/pysr/export_torch.py b/pysr/export_torch.py
@@ -5,7 +5,7 @@
 
 import numpy as np  # noqa: F401
 import sympy  # type: ignore
-from sympy.codegen.cfunctions import log2, log10
+from sympy.codegen.cfunctions import log2, log10  # type: ignore
 
 
 def _reduce(fn):