diff --git a/.github/workflows/fork-maintenance.yml b/.github/workflows/fork-maintenance.yml index 3c32f59cb27..09f5385b455 100644 --- a/.github/workflows/fork-maintenance.yml +++ b/.github/workflows/fork-maintenance.yml @@ -36,8 +36,7 @@ jobs: pr_branch_prefix: "scheduled-merge" requirements_command: > sudo sed -i 's/torchaudio//g' examples/pytorch/_tests_requirements.txt && pip install -r examples/pytorch/_tests_requirements.txt && pip install --no-cache-dir GPUtil azureml azureml-core tokenizers ninja cerberus sympy sacremoses sacrebleu==1.5.1 sentencepiece scipy scikit-learn urllib3 && pip install huggingface_hub datasets && pip install parameterized && pip install -e . - unit_test_command: > - pytest + unit_test_command: pytest performance_test_command: > echo "python examples/pytorch/language-modeling/run_mlm.py --model_name_or_path bert-base-uncased --dataset_name wikitext --dataset_config_name wikitext-2-raw-v1 --do_train --do_eval --output_dir /tmp/test-mlm --per_device_train_batch_size 8 --per_device_eval_batch_size 8 --max_steps 500" docker_image: rocm/pytorch:latest