From 8ef7040bfb28877ba9383134e927335c7c137ac3 Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Mon, 12 Feb 2024 01:53:10 +0100
Subject: [PATCH] Update example diff file for image classification

---
 tests/example_diff/run_image_classification.txt | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/example_diff/run_image_classification.txt b/tests/example_diff/run_image_classification.txt
index dd919e03c9..127ac9bf7d 100644
--- a/tests/example_diff/run_image_classification.txt
+++ b/tests/example_diff/run_image_classification.txt
@@ -30,6 +30,10 @@
 > # Will error if the minimal version of Transformers and Optimum Habana are not installed. Remove at your own risks.
 > check_min_version("4.34.0")
 > check_optimum_habana_min_version("1.8.1")
+62c70
+< 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")
 191c199
 <     parser = HfArgumentParser((ModelArguments, DataTrainingArguments, TrainingArguments))
 ---