diff --git a/py_neuromodulation/nm_bursts.py b/py_neuromodulation/nm_bursts.py index e810120e..0d837501 100644 --- a/py_neuromodulation/nm_bursts.py +++ b/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 diff --git a/py_neuromodulation/nm_sharpwaves.py b/py_neuromodulation/nm_sharpwaves.py index 8a3e0eab..efe1626a 100644 --- a/py_neuromodulation/nm_sharpwaves.py +++ b/py_neuromodulation/nm_sharpwaves.py @@ -1,4 +1,3 @@ -import numpy as np from collections.abc import Sequence from collections import defaultdict from itertools import product @@ -6,7 +5,12 @@ 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 diff --git a/pyproject.toml b/pyproject.toml index 10fb7d7c..d6536578 100644 --- a/pyproject.toml +++ b/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",