From d0328b10a4a7314244e00cb52f99de3bd84b8346 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sass=20B=C3=A1lint?= <sassbalint@users.noreply.github.com>
Date: Sun, 24 Jan 2021 09:43:45 +0100
Subject: [PATCH] torch<=1.4.0 needed

as this stuff does not work with torch>1.4.0
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 3aa5393..aef0a78 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
     license='Apache',
     url="https://github.com/charles9n/bert-sklearn",
     packages=find_packages(exclude=['test', 'scripts', 'examples']),
-    install_requires=['torch>=0.4.1',
+    install_requires=['torch>=0.4.1,<=1.4.0',
                        'scikit-learn',
                        'numpy',
                        'pandas',