fix mypy

scikit-hep · Jul 23, 2024 · 9e2464e · 9e2464e
1 parent 3cfa932
commit 9e2464e
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/src/vector/__init__.py b/src/vector/__init__.py
@@ -32,7 +32,7 @@
     Vector4D,
     dim,
 )
-from vector._version import __version__  # type: ignore[import-not-found]
+from vector._version import __version__
 from vector.backends.awkward_constructors import Array, zip
 from vector.backends.awkward_constructors import Array as awk
 from vector.backends.numpy import (
@@ -87,9 +87,9 @@ def _import_awkward() -> None:
     from vector.backends.awkward import VectorAwkward
 
 try:
-    import sympy
+    import sympy  # type: ignore[import-untyped]
 except ImportError:
-    sympy = None  # type: ignore[assignment]
+    sympy = None
     if not typing.TYPE_CHECKING:
         VectorSympy = None
 else:

diff --git a/src/vector/_lib.py b/src/vector/_lib.py
@@ -3,7 +3,7 @@
 import typing
 
 import numpy
-import sympy
+import sympy  # type: ignore[import-untyped]
 
 if typing.TYPE_CHECKING:
     import sympy
@@ -15,10 +15,10 @@ def nan_to_num(self, val: sympy.Expr, **kwargs: typing.Any) -> sympy.Expr:
         return val
 
     def maximum(self, val1: sympy.Expr | int, val2: sympy.Expr | int) -> sympy.Expr:
-        return val1 if isinstance(val1, sympy.Expr) else val2  # type: ignore[return-value]
+        return val1 if isinstance(val1, sympy.Expr) else val2
 
     def minimum(self, val1: sympy.Expr | int, val2: sympy.Expr | int) -> sympy.Expr:
-        return val1 if isinstance(val1, sympy.Expr) else val2  # type: ignore[return-value]
+        return val1 if isinstance(val1, sympy.Expr) else val2
 
     def arcsin(self, val: sympy.Expr) -> sympy.Expr:
         return sympy.asin(val)
@@ -50,7 +50,7 @@ def isclose(
         val2: sympy.Expr,
         *args: float,
     ) -> sympy.Equality:
-        return sympy.Eq(val1, val2)  # type: ignore[no-untyped-call]
+        return sympy.Eq(val1, val2)
 
     def copysign(self, val1: sympy.Expr, val2: sympy.Expr) -> sympy.Expr:
         return val1
@@ -64,7 +64,7 @@ def sign(self, val: int | float) -> sympy.Expr:
         return numpy.sign(val)
 
     def sqrt(self, val: sympy.Expr) -> sympy.Expr:
-        return sympy.sqrt(val)  # type: ignore[no-untyped-call]
+        return sympy.sqrt(val)
 
     def exp(self, val: sympy.Expr) -> sympy.Expr:
         return sympy.exp(val)

diff --git a/src/vector/backends/sympy.py b/src/vector/backends/sympy.py
@@ -3,7 +3,7 @@
 import typing
 
 import numpy
-import sympy
+import sympy  # type: ignore[import-untyped]
 
 import vector
 from vector._lib import SympyLib