Add py-torch to container

spack-environment/packages.yaml

@@ -284,6 +284,9 @@ packages:
   py-toml:
     require:
     - '@0.10.2'
+  py-torch:
+    require:
+    - '@2.0.0'  
   py-uproot:
     require:
     - '@5.0.5'