Fix mypy error with latest version of mypy

eguiraud · Nov 30, 2023 · 07fc72d · 07fc72d
1 parent 7ec4856
commit 07fc72d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/correctionlib_gradients/_typedefs.py b/src/correctionlib_gradients/_typedefs.py
@@ -1,9 +1,9 @@
-from typing import TypeAlias
+from typing import Any, TypeAlias
 
 import jax
-import numpy as np
+import numpy.typing
 
 # TODO: switch to use numpy.array_api.Array as _the_ array type.
 # Must wait for it to be out of experimental.
 # See https://numpy.org/doc/stable/reference/array_api.html.
-Value: TypeAlias = float | np.ndarray | jax.Array | list[float]
+Value: TypeAlias = float | numpy.typing.NDArray[Any] | jax.Array | list[float]