.

Author: ahmadsharif1

.github/workflows/docs.yaml

@@ -1,9 +1,10 @@
 name: Docs
 
 on:
-  push:
-    branches: [ main ]
-  pull_request:
+  workflow_run:
+    workflows: ["Build and test Linux CUDA wheels"]
+    types:
+      - completed
 
 defaults:
   run:
@@ -26,9 +27,11 @@ jobs:
           python-version: '3.12'
       - name: Update pip
         run: python -m pip install --upgrade pip
-      - name: Install torchcodec from nightly
+      - name: Install torchcodec from the wheel
         run: |
-          pip3 install --pre torch torchvision torchaudio torchcodec --index-url https://download.pytorch.org/whl/nightly/cu124
+          wheel_path=`find pytorch/torchcodec/dist -type f -name "*.whl"`
+          echo Installing $wheel_path
+          ${CONDA_RUN} python -m pip install $wheel_path -vvv
       - name: Install FFMPEG and other deps
         run: |
           conda install cuda-nvrtc=12.4 libnpp -c nvidia