From 55df468646b3da507b3e0616d9cbbfa6c5eb9965 Mon Sep 17 00:00:00 2001 From: Cemberk Date: Fri, 23 Aug 2024 08:45:05 -0500 Subject: [PATCH] Update fork-maintenance.yml --- .github/workflows/fork-maintenance.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/fork-maintenance.yml b/.github/workflows/fork-maintenance.yml index 097c16cd4aea50..ed6a9fbd05d376 100644 --- a/.github/workflows/fork-maintenance.yml +++ b/.github/workflows/fork-maintenance.yml @@ -34,6 +34,8 @@ jobs: ] } pr_branch_prefix: "scheduled-merge" + requirements_command: > + sed -i 's$torchaudio$$g' examples/pytorch/_tests_requirements.txt && pip3 install -r examples/pytorch/_tests_requirements.txt && pip3 install --no-cache-dir GPUtil azureml azureml-core tokenizers ninja cerberus sympy sacremoses sacrebleu==1.5.1 sentencepiece scipy scikit-learn "urllib3<2" && python3 -m pip install huggingface_hub datasets && apt-get install -y sshpass sshfs netcat && python3 -m pip uninstall -y tensorflow flax unit_test_command: > cd tests; folders=($(python3 -c 'import os; tests = os.getcwd(); model_tests = os.listdir(os.path.join(tests, "models")); d1 = sorted(list(filter(os.path.isdir, os.listdir(tests)))); d2 = sorted(list(filter(os.path.isdir, [f"models/{x}" for x in model_tests]))); d1.remove("models"); d = d2 + d1; print("\n".join(d))')); cd ..; for folder in "${folders[@]}"; do pytest tests/${folder} -v --make-reports="huggingface_unit_tests_run_models_gpu_${folder}" -rfEsx --continue-on-collection-errors -m "not not_device_test" -p no:cacheprovider; done; allstats=($(find reports -name stats.txt)); for stat in "${allstats[@]}"; do echo "$stat"; cat $stat; done performance_test_command: >