Update environment-aurora.yml

globus-labs · Apr 10, 2024 · 81be3ec · 81be3ec
1 parent a1901bc
commit 81be3ec
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/envs/environment-aurora.yml b/envs/environment-aurora.yml
@@ -41,7 +41,8 @@ dependencies:
 
   - pip
   - pip:
-    - torch_geometric pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv
+    - torch_geometric
+    - torch_scatter
     - pytorch-lightning<2
     - git+https://gitlab.com/ase/ase.git
     - git+https://github.com/exalearn/colmena.git  # Fixes for streaming not yet on PyPI