Set minimum version of torch to be 1.8

Using v1.8 features
scikit-hep · Mar 5, 2021 · 3c61d19 · 3c61d19
1 parent 9f0304a
commit 3c61d19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -6,7 +6,7 @@
         'tensorflow~=2.2.0',  # TensorFlow minor releases are as volatile as major
         'tensorflow-probability~=0.10.0',
     ],
-    'torch': ['torch~=1.2'],
+    'torch': ['torch~=1.8'],
     'jax': ['jax~=0.2.4', 'jaxlib~=0.1.56'],
     'xmlio': [
         'uproot3~=3.14',