Pin torch version

.github/actions/install-python-and-package/action.yml

@@ -36,11 +36,12 @@ runs:
         python -m pip install --upgrade pip setuptools wheel
 
     # only necessary on linux to avoid bloated installs
+    # pining the version of torch is temporary, see see #829
     - name: Install tensorflow/pytorch cpu version
       if: runner.os == 'Linux' && steps.cache-python-env.outputs.cache-hit != 'true'
       shell: bash {0}
       run: |
-        python -m pip install torch --index-url https://download.pytorch.org/whl/cpu
+        python -m pip install torch==2.3.1 --index-url https://download.pytorch.org/whl/cpu
         python -m pip install tensorflow-cpu
 
     - name: Install DIANNA

setup.cfg

@@ -49,6 +49,7 @@ install_requires =
     scikit-learn
     tqdm
     xarray
+    torch <=2.3.1  # this is temporary, see #829
     # tf and tfprob are required but not declared by onnx_tf
     tensorflow >= 2.12,<2.16
     tensorflow_probability <0.24