Fix tests

etr2460 · etr2460 · commit 75b69212d67c · 2024-04-30T16:56:42.000-07:00
diff --git a/.github/workflows/run_tests.yaml b/.github/workflows/run_tests.yaml
@@ -29,7 +29,7 @@ jobs:
         python -m pip install --upgrade pip
         pip install pyyaml
         pip install pytest
-        pip install -e .
+        pip install -e .[torch]
 
     - name: Run unit tests
       env:
