move ort training tests to onnxruntime-training

huggingface · Jan 13, 2025 · 64e9c86 · 64e9c86
1 parent 0a7a23d
commit 64e9c86
Show file tree

Hide file tree

Showing 7 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/test_onnxruntime_training.yml b/.github/workflows/test_onnxruntime_training.yml
@@ -55,12 +55,12 @@ jobs:
 
       - name: Test with pytest (trainer)
         run: |
-          RUN_SLOW=1 pytest tests/onnxruntime/training/test_trainer.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
+          RUN_SLOW=1 pytest tests/onnxruntime-training/test_examples.py --durations=0 -s -vvvv
         env:
           HF_DATASETS_TRUST_REMOTE_CODE: 1
diff --git a/...me/ds_configs/ds_config_zero_stage_1.json → ...ng/ds_configs/ds_config_zero_stage_1.json b/...me/ds_configs/ds_config_zero_stage_1.json → ...ng/ds_configs/ds_config_zero_stage_1.json
diff --git a/...me/ds_configs/ds_config_zero_stage_2.json → ...ng/ds_configs/ds_config_zero_stage_2.json b/...me/ds_configs/ds_config_zero_stage_2.json → ...ng/ds_configs/ds_config_zero_stage_2.json
diff --git a/...me/ds_configs/ds_config_zero_stage_3.json → ...ng/ds_configs/ds_config_zero_stage_3.json b/...me/ds_configs/ds_config_zero_stage_3.json → ...ng/ds_configs/ds_config_zero_stage_3.json
diff --git a/...nfigs/ds_config_zero_stage_inifinity.json → ...nfigs/ds_config_zero_stage_inifinity.json b/...nfigs/ds_config_zero_stage_inifinity.json → ...nfigs/ds_config_zero_stage_inifinity.json
diff --git a/tests/onnxruntime/training/test_examples.py → tests/onnxruntime-training/test_examples.py b/tests/onnxruntime/training/test_examples.py → tests/onnxruntime-training/test_examples.py
diff --git a/tests/onnxruntime/training/test_trainer.py → tests/onnxruntime-training/test_trainer.py b/tests/onnxruntime/training/test_trainer.py → tests/onnxruntime-training/test_trainer.py
@@ -730,7 +730,7 @@ def test_trainer_fp16_ds_stage1(self, test_name, model_name, task, data_metric_c
                     weight_decay=self.weight_decay,
                     logging_dir=tmp_dir,
                     fp16=True,
-                    deepspeed="tests/onnxruntime/ds_configs/ds_config_zero_stage_1.json",
+                    deepspeed="tests/onnxruntime-training/ds_configs/ds_config_zero_stage_1.json",
                 )
 
                 trainer, _ = get_ort_trainer(
@@ -769,7 +769,7 @@ def test_trainer_fp16_ds_stage2(self, test_name, model_name, task, data_metric_c
                     weight_decay=self.weight_decay,
                     logging_dir=tmp_dir,
                     fp16=True,
-                    deepspeed="tests/onnxruntime/ds_configs/ds_config_zero_stage_2.json",
+                    deepspeed="tests/onnxruntime-training/ds_configs/ds_config_zero_stage_2.json",
                 )
 
                 trainer, _ = get_ort_trainer(