From 8fa39259d7fc2c53301a99807a10029d347c846a Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Wed, 5 Feb 2025 18:07:59 +0000
Subject: [PATCH] Update

[ghstack-poisoned]
---
 .github/workflows/docs.yml | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
index 6b963ce9ca1..5f99fb12ba6 100644
--- 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