chore: Fix docgen

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

.github/workflows/docgen.yml

@@ -19,10 +19,10 @@ jobs:
           ref: ${{github.head_ref}}
       - name: Select python
         run: |
-          export PATH=/opt/python/cp311-cp311/bin/:$PATH
+          echo "/opt/python/cp311-cp311/bin/" >> $GITHUB_PATH
       - name: Install base deps
         run: |
-          python3 -m pip install pyyaml
+          python3 -m pip install pyyaml torch --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu121
           VERSION_SUFFIX=cu121 ./packaging/pre_build_script.sh
       - name: Get HEAD SHA
         id: vars