fix: torch-geometric version typo in tox.ini

OmicsML · Aug 18, 2022 · 6f2ea20 · 6f2ea20
1 parent 027fe98
commit 6f2ea20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tox.ini b/tox.ini
@@ -18,7 +18,7 @@ commands =
     osx: pip install torch==1.12.1 torchvision==0.13.1
     # Custom installation for PyG, cf. https://github.com/rusty1s/pytorch_scatter/pull/268
     gpu: pip install torch-scatter torch-sparse torch-cluster torch-geometric -f https://data.pyg.org/whl/torch-1.12.1+cu102.html
-    {cpu,osx}: pip install torch-geometric==2.10.0 torch-scatter torch-sparse torch-cluster -f https://data.pyg.org/whl/torch-1.12.1+cpu.html
+    {cpu,osx}: pip install torch-geometric==2.1.0 torch-scatter torch-sparse torch-cluster -f https://data.pyg.org/whl/torch-1.12.1+cpu.html
     # Custom installation for DGL
     gpu: pip install dgl-cu102 -f https://data.dgl.ai/wheels/repo.html
     {cpu,osx}: pip install dgl -f https://data.dgl.ai/wheels/repo.html