fix auto-imports

flatironinstitute · Nov 28, 2024 · 7cec3c0 · 7cec3c0
1 parent 9985784
commit 7cec3c0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/nemos/typing.py b/src/nemos/typing.py
@@ -6,7 +6,7 @@
 import jaxopt
 import pynapple as nap
 from jax.typing import ArrayLike
-from statsmodels.tools.typing import NDArray
+from numpy.typing import NDArray
 
 from .pytrees import FeaturePytree