From 537b9b254d511285fbadfc06c24939998c7c4340 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Tue, 18 Mar 2025 14:41:00 +0530 Subject: [PATCH] uninstall deps properly from pr gpu tests. --- .github/workflows/pr_tests_gpu.yml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/pr_tests_gpu.yml b/.github/workflows/pr_tests_gpu.yml index d86eccc28bb5..87d51773888e 100644 --- a/.github/workflows/pr_tests_gpu.yml +++ b/.github/workflows/pr_tests_gpu.yml @@ -177,6 +177,7 @@ jobs: torch_cuda_tests: name: Torch CUDA Tests + needs: [check_code_quality, check_repository_consistency] runs-on: group: aws-g4dn-2xlarge container: @@ -245,7 +246,7 @@ jobs: run_examples_tests: name: Examples PyTorch CUDA tests on Ubuntu - pip uninstall transformers -y && python -m uv pip install -U transformers@git+https://github.com/huggingface/transformers.git --no-deps + needs: [check_code_quality, check_repository_consistency] runs-on: group: aws-g4dn-2xlarge @@ -264,6 +265,7 @@ jobs: - name: Install dependencies run: | python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH" + pip uninstall transformers -y && python -m uv pip install -U transformers@git+https://github.com/huggingface/transformers.git --no-deps python -m uv pip install -e [quality,test,training] - name: Environment