Update setup.py

ksh981214 · Oct 16, 2018 · 9042837 · 9042837
1 parent 4e9f1c1
commit 9042837
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@
           'pandas==0.23.4',
           'scipy==1.1.0',
           'tensorflow==1.11.0',
-          'pytorch==0.4.1',
+          'torch==0.4.1',
           # rdkit
       ],
       description='MOSES: A benchmarking platform for molecular generation models',