diff --git a/setup.py b/setup.py index 45cbd58..07037f1 100644 --- a/setup.py +++ b/setup.py @@ -107,13 +107,13 @@ def get_blas_lib_dir(cls): # Requirements numpy_requires = [ 'numpy<1.20.0; python_version<"3.7"', # setup_requires needs correct version for <3.7 - 'numpy>=1.19.5; python_version>="3.7"' + 'numpy>=1.19.5,<2.0.0; python_version>="3.7"' ] setup_requires = numpy_requires + [ 'pytest-runner' ] install_requires = numpy_requires + [ - 'scipy>=1.4.1', + 'scipy>=1.4.1,<1.14.0', 'scikit-learn>=0.24.1', 'torch==1.13; python_version<"3.8"', 'torch>=2.0; python_version>="3.8"',