From f334968ef84a9909da6e87ccaad377b6cf867e2c Mon Sep 17 00:00:00 2001 From: bdvllrs Date: Wed, 8 Nov 2023 13:32:02 +0000 Subject: [PATCH] Fix torch install in test action --- .github/workflows/tests.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index 120ac902..83dc1835 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -22,7 +22,7 @@ jobs: run: | poetry install --with test poetry run python -m pip install --upgrade pip - poetry run pip install torch==2.0.1+cpu --index-url https://download.pytorch.org/whl/cpu + poetry run pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu - name: Test with pytest run: | poetry run pytest tests/