Merge pull request #210 from timephy/patch-1

Specify torch version, which fixed #199
graphnet-team · May 19, 2022 · a05f45e · a05f45e
2 parents 03f48cb + a0b7478
commit a05f45e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -53,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",