Add compatibility with Numpy 1.26 and 2.0

py_neuromodulation/nm_bursts.py

@@ -1,5 +1,8 @@
 import numpy as np
-from numpy.lib._function_base_impl import _quantile as np_quantile  # type:ignore
+if np.__version__ >= "2.0.0":
+    from numpy.lib._function_base_impl import _quantile as np_quantile  # type:ignore
+else:
+    from numpy.lib.function_base import _quantile as np_quantile  # type:ignore
 from collections.abc import Sequence
 from itertools import product
 

py_neuromodulation/nm_sharpwaves.py

@@ -1,12 +1,16 @@
-import numpy as np
 from collections.abc import Sequence
 from collections import defaultdict
 from itertools import product
 
 from py_neuromodulation.nm_types import NMBaseModel
 from pydantic import model_validator
 from typing import TYPE_CHECKING, Any, Callable
-from numpy._core._methods import _mean as np_mean
+
+import numpy as np
+if np.__version__ >= "2.0.0":
+    from numpy._core._methods import _mean as np_mean # type: ignore
+else:
+    from numpy.core._methods import _mean as np_mean
 
 from py_neuromodulation.nm_features import NMFeature
 from py_neuromodulation.nm_types import BoolSelector, FrequencyRange

pyproject.toml

@@ -38,8 +38,8 @@ dependencies = [
     "mne-connectivity",
     "mrmr_selection",
     "nolds",
-    "numpy>=2.0.0",
-    "pandas>=2.2.2",
+    "numpy >= 1.21.2",
+    "pandas >= 2.0.0",
     "scikit-image",
     "scikit-learn >= 0.24.2",
     "scikit-optimize",