update script

pytorch · Mar 7, 2024 · 2d97ee5 · 2d97ee5
1 parent 62399f1
commit 2d97ee5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/tpu_ci.yml b/.github/workflows/tpu_ci.yml
@@ -32,8 +32,8 @@ jobs:
         run: |
           pip install fsspec
           pip install rich
-          pip uninstall torch
-          pip uninstall torchvision
+          yes | pip uninstall torch
+          yes | pip uninstall torchvision
           pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
           cd pytorch/xla
           test/tpu/run_tests.sh