From 450e674fd238c0e16ea4d111bef5e92424333217 Mon Sep 17 00:00:00 2001 From: MilesCranmer Date: Fri, 6 Dec 2024 22:22:57 +0000 Subject: [PATCH] test: fix mypy errors --- pysr/export_jax.py | 2 +- pysr/export_sympy.py | 2 +- pysr/export_torch.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pysr/export_jax.py b/pysr/export_jax.py index 506ac7d4..4f8ead20 100644 --- 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 index 06996e6a..925c284b 100644 --- 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 index c846d80e..eb3ccd8a 100644 --- 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):