diff --git a/.github/workflows/run_tests.yaml b/.github/workflows/run_tests.yaml
index d36981bd98..11a6151d7c 100644
--- 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: