diff --git a/requirements.txt b/requirements.txt index da889c05..865df0bf 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ +numpy<2.0.0 tensorflow>=2.1.0 -numpy scikit-learn scikit-image matplotlib diff --git a/setup.py b/setup.py index 04c2f989..08a132ef 100644 --- a/setup.py +++ b/setup.py @@ -12,7 +12,7 @@ author="Thomas FEL", author_email="thomas_fel@brown.edu", license="MIT", - install_requires=['tensorflow>=2.1.0,<2.16.0', 'numpy', 'scikit-learn', 'scikit-image', + install_requires=['numpy<2.0.0', 'tensorflow>=2.1.0,<2.16.0', 'scikit-learn', 'scikit-image', 'matplotlib', 'scipy', 'opencv-python', 'deprecated'], extras_require={ "tests": ["pytest", "pylint"],