move version fix to where it was specified before

graphnet-team · May 17, 2022 · a0b7478 · a0b7478
1 parent 764dc87
commit a0b7478
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -22,7 +22,6 @@ def install(package):
     "numpy",
     "timer",
     "tqdm",
-    "torch==1.10.1",
     "torch-cluster==1.5.9",
     "torch-scatter==2.0.9",
     "torch-sparse==0.6.12",
@@ -54,7 +53,7 @@ def install(package):
 try:
     import torch  # pyright: reportMissingImports=false
 except ImportError:
-    install("torch==1.9.0")
+    install("torch==1.10.1")
 
 setup(
     name="graphnet",