pip install pytorch geometric nightly instead

choderalab · Jan 26, 2024 · be05591 · be05591
1 parent 81f8818
commit be05591
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/devtools/conda-envs/test_env.yaml b/devtools/conda-envs/test_env.yaml
@@ -1,10 +1,10 @@
 name: test
 channels:
-  - pytorch-nightly
   - conda-forge
 dependencies:
+  - pip
   - pytorch
-  - pytorch_geometric
+  # - pytorch_geometric
   - pytorch_cluster
   - pytorch_scatter
   - pytorch_sparse
@@ -20,3 +20,6 @@ dependencies:
   - pytest-cov
   - codecov
   - pydantic >=1.10.8,<2.0.0a0
+
+  - pip:
+    - pyg-nightly