Skip to content

Commit

Permalink
Allow numpy-2.1 and test it without audio extra (huggingface#7118)
Browse files Browse the repository at this point in the history
* Revert "Temporarily pin numpy<2.1 to fix CI"

This reverts commit fb8ae4d.

* Do not test audio with numpy-2.1
  • Loading branch information
albertvillanova authored Aug 21, 2024
1 parent c3aaf21 commit 0a0fa48
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -111,7 +111,7 @@
# For file locking
"filelock",
# We use numpy>=1.17 to have np.random.Generator (Dataset shuffling)
"numpy>=1.17,<2.1", # Temporarily pin <2.1 (see GH-7111)
"numpy>=1.17",
# Backend and serialization.
# Minimum 15.0.0 to be able to cast dictionary types to their underlying types
"pyarrow>=15.0.0",
Expand Down Expand Up @@ -194,6 +194,7 @@

NUMPY2_INCOMPATIBLE_LIBRARIES = [
"faiss-cpu",
"librosa", # librosa -> numba-0.60.0 requires numpy < 2.1 (see GH-7111)
"tensorflow",
"transformers",
]
Expand Down

0 comments on commit 0a0fa48

Please sign in to comment.