ci: require numpy<2.0.0 for tensorflow compatibility

deel-ai · Sep 30, 2024 · b3192bb · b3192bb
1 parent 61c8b8f
commit b3192bb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -12,7 +12,7 @@
     author="Thomas FEL",
     author_email="[email protected]",
     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"],