From 72e5e69afb85a4de44972674f6bfda3b749ddb72 Mon Sep 17 00:00:00 2001 From: Alexey Fadeev Date: Fri, 9 Feb 2024 03:18:04 -0800 Subject: [PATCH] Updated datasets requirements. Need a package version >= 2.14.0 --- examples/image-classification/requirements.txt | 2 +- examples/image-classification/run_image_classification.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/image-classification/requirements.txt b/examples/image-classification/requirements.txt index fc17d6b79f..87694059fe 100644 --- a/examples/image-classification/requirements.txt +++ b/examples/image-classification/requirements.txt @@ -1,5 +1,5 @@ torch>=1.5.0 torchvision>=0.6.0 -datasets>=2.4.0 +datasets>=2.14.0 evaluate scikit-learn diff --git a/examples/image-classification/run_image_classification.py b/examples/image-classification/run_image_classification.py index 080970c344..82b2e1ba3f 100644 --- a/examples/image-classification/run_image_classification.py +++ b/examples/image-classification/run_image_classification.py @@ -67,7 +67,7 @@ def check_optimum_habana_min_version(*a, **b): check_min_version("4.34.0") check_optimum_habana_min_version("1.8.1") -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)