add trust remote code for latest datasets release

huggingface · Jun 14, 2024 · 7e75328 · 7e75328
1 parent 23a6d30
commit 7e75328
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/openvino/test_training.py b/tests/openvino/test_training.py
@@ -618,7 +618,7 @@ def test_training(self, _, desc: OVTrainerTestDescriptor):
         self.run_ovtrainer_training_checks(desc)
 
     def prepare_model_and_dataset(self, desc: OVTrainerTestDescriptor):
-        self.dataset = load_dataset("hf-internal-testing/cats_vs_dogs_sample")
+        self.dataset = load_dataset("hf-internal-testing/cats_vs_dogs_sample", trust_remote_code=True)
         self.num_labels = len(self.dataset["train"].features["labels"].names)
 
         self.feature_extractor = AutoImageProcessor.from_pretrained(desc.model_id)