Skip to content

Commit ae14612

Browse files
authored
[CI] uninstall deps properly from pr gpu tests. (#11102)
uninstall deps properly from pr gpu tests.
1 parent 0ab8fe4 commit ae14612

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

.github/workflows/pr_tests_gpu.yml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,6 +177,7 @@ jobs:
177177

178178
torch_cuda_tests:
179179
name: Torch CUDA Tests
180+
needs: [check_code_quality, check_repository_consistency]
180181
runs-on:
181182
group: aws-g4dn-2xlarge
182183
container:
@@ -245,7 +246,7 @@ jobs:
245246

246247
run_examples_tests:
247248
name: Examples PyTorch CUDA tests on Ubuntu
248-
pip uninstall transformers -y && python -m uv pip install -U transformers@git+https://github.com/huggingface/transformers.git --no-deps
249+
needs: [check_code_quality, check_repository_consistency]
249250
runs-on:
250251
group: aws-g4dn-2xlarge
251252

@@ -264,6 +265,7 @@ jobs:
264265
- name: Install dependencies
265266
run: |
266267
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
268+
pip uninstall transformers -y && python -m uv pip install -U transformers@git+https://github.com/huggingface/transformers.git --no-deps
267269
python -m uv pip install -e [quality,test,training]
268270
269271
- name: Environment

0 commit comments

Comments
 (0)