Updated requirements for image-classification samples: datasets>=2.14…

….0 (huggingface#28974) Updated datasets requirements. Need a package version >= 2.14.0
sbucaille · Feb 14, 2024 · 77a7957 · 77a7957
1 parent 0081cc8
commit 77a7957
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/pytorch/image-classification/requirements.txt b/examples/pytorch/image-classification/requirements.txt
@@ -1,5 +1,5 @@
 accelerate>=0.12.0
 torch>=1.5.0
 torchvision>=0.6.0
-datasets>=1.17.0
+datasets>=2.14.0
 evaluate
diff --git a/examples/pytorch/image-classification/run_image_classification.py b/examples/pytorch/image-classification/run_image_classification.py
@@ -59,7 +59,7 @@
 # Will error if the minimal version of Transformers is not installed. Remove at your own risks.
 check_min_version("4.38.0.dev0")
 
-require_version("datasets>=1.8.0", "To fix: pip install -r examples/pytorch/image-classification/requirements.txt")
+require_version("datasets>=2.14.0", "To fix: pip install -r examples/pytorch/image-classification/requirements.txt")
 
 MODEL_CONFIG_CLASSES = list(MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING.keys())
 MODEL_TYPES = tuple(conf.model_type for conf in MODEL_CONFIG_CLASSES)