pin datasets in examples requirements (#850)

* pin datasets in examples requirements * fix notebook test
huggingface · Jul 29, 2024 · 39d6638 · 39d6638
1 parent 0da4f14
commit 39d6638
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/test_openvino_notebooks.yml b/.github/workflows/test_openvino_notebooks.yml
@@ -40,8 +40,8 @@ jobs:
         # ffmpeg, torchaudio and pillow are required for image classification and audio classification pipelines
         sudo apt-get install ffmpeg
         pip install torch torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
-        pip install ".[tests, openvino]" nbval
         pip install -r notebooks/openvino/requirements.txt
+        pip install .[tests,openvino] nbval
 
     - run: free -h
     - run: lscpu

diff --git a/examples/openvino/audio-classification/requirements.txt b/examples/openvino/audio-classification/requirements.txt
@@ -1,5 +1,5 @@
-datasets>=1.14.0
+datasets>=1.14.0,<2.20.0
 evaluate
 librosa
 torchaudio
-accelerate
+accelerate
diff --git a/examples/openvino/image-classification/requirements.txt b/examples/openvino/image-classification/requirements.txt
@@ -1,4 +1,4 @@
-datasets >= 1.8.0
+datasets>=1.14.0,<2.20.0
 torch >= 1.9.0
 torchvision>=0.6.0
 evaluate

diff --git a/examples/openvino/question-answering/requirements.txt b/examples/openvino/question-answering/requirements.txt
@@ -1,4 +1,4 @@
-datasets >= 1.8.0
+datasets>=1.14.0,<2.20.0
 torch >= 1.9.0
 evaluate
 accelerate
diff --git a/examples/openvino/text-classification/requirements.txt b/examples/openvino/text-classification/requirements.txt
@@ -1,4 +1,4 @@
-datasets >= 1.8.0
+datasets>=1.14.0,<2.20.0
 sentencepiece != 0.1.92
 scipy
 scikit-learn