trust remote code for training datasets

huggingface · Jan 13, 2025 · eb25460 · eb25460
1 parent cb9cb7f
commit eb25460
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/.github/workflows/test_onnxruntime_training.yml b/.github/workflows/test_onnxruntime_training.yml
@@ -53,7 +53,14 @@ jobs:
           pip install --no-cache-dir evaluate absl-py rouge_score seqeval sacrebleu nltk scikit-learn
           pip install .[tests,onnxruntime-training]
 
-      - name: Test with pytest
+      - name: Test with pytest (trainer)
         run: |
-          RUN_SLOW=1 pytest tests/onnxruntime/training/nightly_test_trainer.py --durations=0 -s -vvvv
-          RUN_SLOW=1 pytest tests/onnxruntime/training/nightly_test_examples.py --durations=0 -s -vvvv
+          RUN_SLOW=1 pytest tests/onnxruntime/training/test_trainer.py --durations=0 -s -vvvv
+        env:
+          HF_DATASETS_TRUST_REMOTE_CODE: 1
+
+      - name: Test with pytest (examples)
+        run: |
+          RUN_SLOW=1 pytest tests/onnxruntime/training/test_examples.py --durations=0 -s -vvvv
+        env:
+          HF_DATASETS_TRUST_REMOTE_CODE: 1
diff --git a/...runtime/training/nightly_test_examples.py → tests/onnxruntime/training/test_examples.py b/...runtime/training/nightly_test_examples.py → tests/onnxruntime/training/test_examples.py
diff --git a/...xruntime/training/nightly_test_trainer.py → tests/onnxruntime/training/test_trainer.py b/...xruntime/training/nightly_test_trainer.py → tests/onnxruntime/training/test_trainer.py