Fix torch install in test action

.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/