Update

[ghstack-poisoned]
pytorch · Feb 5, 2025 · 8fa3925 · 8fa3925
1 parent 34fbfcd
commit 8fa3925
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
@@ -76,10 +76,13 @@ jobs:
         git version
         
         # 5. Install PyTorch
-        python3 -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U --quiet --root-user-action=ignore
-        
-        # 6. Install tensordict
-        python3 -m pip install git+https://github.com/pytorch/tensordict.git --quiet --root-user-action=ignore
+        if [[ ${{ github.event_name }} == push && (${{ github.ref_type }} == tag || (${{ github.ref_type }} == branch && ${{ github.ref_name }} == release/*)) ]]; then
+          python3 -m pip install torch torchvision
+          python3 -m pip install tensordict
+        else
+          python3 -m pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cpu -U --quiet --root-user-action=ignore
+          python3 -m pip install git+https://github.com/pytorch/tensordict.git --quiet --root-user-action=ignore
+        fi
         
         # 7. Install TorchRL
         python3 setup.py develop