Merge pull request #639 from mv1388/pytorch-min-version

Specify minimum pytorch version
mv1388 · Nov 1, 2020 · 7bba82d · 7bba82d
2 parents dc52418 + 74fc837
commit 7bba82d
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/dist/aitoolbox-1.2.0-py3-none-any.whl b/dist/aitoolbox-1.2.0-py3-none-any.whl
diff --git a/dist/aitoolbox-1.2.0.tar.gz b/dist/aitoolbox-1.2.0.tar.gz
diff --git a/requirements.txt b/requirements.txt
@@ -3,7 +3,7 @@ pandas
 scikit-learn
 matplotlib
 seaborn
-torch
+torch>=1.6.0
 torchvision
 torchtext
 pytorch-nlp

diff --git a/setup.py b/setup.py
@@ -24,7 +24,7 @@
         'matplotlib',
         'seaborn',
 
-        'torch',
+        'torch>=1.6.0',
         'torchvision',
         'torchtext',
         'pytorch-nlp',