fix type hinting axes

martinvonk · Sep 25, 2023 · 4373fe5 · 4373fe5
1 parent dfea975
commit 4373fe5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/spei/_typing.py b/src/spei/_typing.py
@@ -1,12 +1,12 @@
 # flake8: noqa
 from typing import Any, Union, TypeVar
 
-# from matplotlib.axes import Axes
-from numpy import float64, generic
+from matplotlib.axes import Axes as mplAxes
+from numpy import float64
 from numpy.typing import NDArray
 from scipy.stats._continuous_distns import rv_continuous
 
 ContinuousDist = Union[Any, rv_continuous]
-Axes = TypeVar("Axes", bound=Union[generic, Any])
-NDArrayAxes = NDArray[Axes]
+Axes = TypeVar("Axes", bound=Union[mplAxes, Any])
+NDArrayAxes = NDArray[Any]
 NDArrayFloat = NDArray[float64]