From acd0df41557de139d91b30b2bd2a039eb5ce98a9 Mon Sep 17 00:00:00 2001 From: Henry Tsang Date: Mon, 9 Oct 2023 11:55:47 -0700 Subject: [PATCH] fix docs.yml by installing torchmetrics (#1433) Summary: manually install torchmetrics. for some reason this problem isn't countered in build-wheels-linux Differential Revision: D50091290 --- .github/workflows/docs.yml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/.github/workflows/docs.yml b/.github/workflows/docs.yml index 39897956c..4d48a0afe 100644 --- 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