Fallback on import newer numpy.testing module

talmolab · Sep 12, 2023 · f7a3964 · f7a3964
1 parent e0ead82
commit f7a3964
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -19,7 +19,7 @@ classifiers = [
     "Programming Language :: Python :: 3.9"
 ]
 dependencies = [
-    "numpy>=1.19.5",
+    "numpy>=1.19.2",
     "attrs>=21.2.0",
     "h5py>=3.1.0",
     "pynwb",

diff --git a/sleap_io/io/nwb.py b/sleap_io/io/nwb.py
@@ -8,7 +8,11 @@
 
 import pandas as pd  # type: ignore[import]
 import numpy as np
-from numpy.typing import ArrayLike
+
+try:
+    from numpy.typing import ArrayLike
+except ImportError:
+    ArrayLike = np.ndarray
 from pynwb import NWBFile, NWBHDF5IO, ProcessingModule  # type: ignore[import]
 from ndx_pose import PoseEstimationSeries, PoseEstimation  # type: ignore[import]