Fix keras bug

capitalone · Mar 22, 2024 · 8e3c439 · 8e3c439
1 parent d8a6db9
commit 8e3c439
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -80,11 +80,11 @@ repos:
 
             # requirements-ml.txt
             scikit-learn>=0.23.2,
-            keras>=2.4.3,
+            'keras>=2.4.3,<3.0.0',
             rapidfuzz>=2.6.1,
-            tensorflow>=2.6.4; sys.platform != 'darwin',
-            tensorflow>=2.6.4; sys_platform == 'darwin' and platform_machine != 'arm64',
-            tensorflow-macos>=2.6.4; sys_platform == 'darwin' and platform_machine == 'arm64',
+            "tensorflow>=2.6.4,<2.15.0; sys.platform != 'darwin'",
+            "tensorflow>=2.6.4,<2.15.0; sys_platform == 'darwin' and platform_machine != 'arm64'",
+            "tensorflow-macos>=2.6.4,<2.15.0; sys_platform == 'darwin' and platform_machine == 'arm64'",
             tqdm>=4.0.0,
 
             # requirements-reports.txt

diff --git a/requirements-ml.txt b/requirements-ml.txt
@@ -1,7 +1,7 @@
 scikit-learn>=0.23.2
-keras>=2.4.3, <3.1.1
+keras>=2.4.3,<3.0.0
 rapidfuzz>=2.6.1
-tensorflow>=2.6.4, <2.16.1; sys.platform != 'darwin'
-tensorflow>=2.6.4, <2.16.1; sys_platform == 'darwin' and platform_machine != 'arm64'
-tensorflow-macos>=2.6.4, <2.16.1; sys_platform == 'darwin' and platform_machine == 'arm64'
+tensorflow>=2.6.4,<2.15.0; sys.platform != 'darwin'
+tensorflow>=2.6.4,<2.15.0; sys_platform == 'darwin' and platform_machine != 'arm64'
+tensorflow-macos>=2.6.4,<2.15.0; sys_platform == 'darwin' and platform_machine == 'arm64'
 tqdm>=4.0.0
diff --git a/requirements-test.txt b/requirements-test.txt
@@ -1,5 +1,5 @@
 coverage>=5.0.1
-dask>=2.29.0, <2024.2.0
+dask>=2.29.0,<2024.2.0
 fsspec>=0.3.3
 pytest>=6.0.1
 pytest-cov>=2.8.1