Fix a flaky wget issue. (#193)

* fix a flaky wget issue. * fix a dir issue. * fix comments * fix missing dir
AI-Hypercomputer · Mar 13, 2024 · b2f99eb · b2f99eb
1 parent 4750150
commit b2f99eb
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/dags/legacy_test/tests/pytorch/nightly/common.libsonnet b/dags/legacy_test/tests/pytorch/nightly/common.libsonnet
@@ -140,9 +140,8 @@ local volumes = import 'templates/volumes.libsonnet';
         pip3 uninstall -y torch torchvision
         pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu
 
-        mkdir pytorch
-        wget https://github.com/pytorch/xla/archive/refs/heads/master.tar.gz -O - | tar xzf -
-        mv xla-master pytorch/xla
+        mkdir -p pytorch/xla
+        git clone --depth=1 https://github.com/pytorch/xla.git pytorch/xla
 
         %s