fix docs.yml by installing torchmetrics (pytorch#1433)

Summary: manually install torchmetrics. for some reason this problem isn't countered in build-wheels-linux Differential Revision: D50091290
henrylhtsang · Oct 9, 2023 · 7c6a5cd · 7c6a5cd
1 parent ee359ee
commit 7c6a5cd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml
@@ -56,6 +56,9 @@ jobs:
     - name: Install fbgemm
       run: |
         conda run -n build_binary pip install fbgemm-gpu --index-url https://download.pytorch.org/whl/nightly/cpu
+    - name: Install torchmetrics
+      run: |
+        conda run -n build_binary pip install torchmetrics==1.0.3
     - name: Install TorchRec
       run: |
         conda run -n build_binary pip install torchrec --index-url https://download.pytorch.org/whl/nightly/cpu