Add back in torch

scikit-hep · Mar 7, 2024 · f79700c · f79700c
1 parent 9e549a4
commit f79700c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -140,8 +140,8 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip setuptools wheel
-        # No PyTorch or TensorFlow for Python 3.12 yet
-        python -m pip install --upgrade ".[jax,minuit,xmlio,contrib,shellcomplete,test]"
+        # No TensorFlow for Python 3.12 yet
+        python -m pip install --upgrade ".[jax,torch,minuit,xmlio,contrib,shellcomplete,test]"
 
     - name: List installed Python packages
       run: python -m pip list