test

Author: goanpeca

.github/workflows/tpu-tests.yml

@@ -71,8 +71,8 @@ jobs:
           ## Install torch & xla and torchvision
           uv pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
           ## TODO: download file and rename as valid file with version
-          curl -o torch-2.5.0-cp310-cp310-linux_x86_64.whl https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-nightly-cp310-cp310-linux_x86_64.whl
-          uv pip install torch-2.5.0-cp310-cp310-linux_x86_64.whl
+          curl -o torch_xla-2.5.0-cp310-cp310-linux_x86_64.whl https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-nightly-cp310-cp310-linux_x86_64.whl
+          uv pip install torch_xla-2.5.0-cp310-cp310-linux_x86_64.whl
           # uv pip install https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-nightly-cp310-cp310-linux_x86_64.whl
 
           # Check installation