diff --git a/spack-environment/packages.yaml b/spack-environment/packages.yaml
index dda0113580f9c13a1116a8d6cc17438836ae734d..5f3a4c4ecd5bf152a387af353505af4492a69b79 100644
--- a/spack-environment/packages.yaml
+++ b/spack-environment/packages.yaml
@@ -303,6 +303,9 @@ packages:
   py-toml:
     require:
     - '@0.10.2'
+  py-torch:
+    require:
+    - '@2.0.0'  
   py-uproot:
     require:
     - '@5.0.5'