Temporarily pin numpy<2.1 to fix CI (#7114)

huggingface · Aug 20, 2024 · fb8ae4d · fb8ae4d
1 parent 5f42139
commit fb8ae4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -111,7 +111,7 @@
     # For file locking
     "filelock",
     # We use numpy>=1.17 to have np.random.Generator (Dataset shuffling)
-    "numpy>=1.17",
+    "numpy>=1.17,<2.1",  # Temporarily pin <2.1 (see GH-7111)
     # Backend and serialization.
     # Minimum 15.0.0 to be able to cast dictionary types to their underlying types
     "pyarrow>=15.0.0",