Pin numpy<2 as required by tensorflow

huggingface · Sep 12, 2024 · 32e2c67 · 32e2c67
1 parent c476380
commit 32e2c67
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -79,6 +79,7 @@
     "pytest-datadir",
     "pytest-xdist",
     # optional dependencies
+    "numpy<2.0.0",  # tensorflow requires numpy < 2
     "tensorflow>=2.3,!=2.6.0,!=2.6.1, <=2.10",
     "torch",
     # metrics dependencies