From 139caabe31e83d75996b3969180e5a58276cf7fd Mon Sep 17 00:00:00 2001
From: Yin Li <eelregit@gmail.com>
Date: Mon, 9 Dec 2019 17:21:25 -0500
Subject: [PATCH] Remove pytorch version constraint and Add python 3 constraint
 in setup.py

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index d5e6a11..5527a12 100644
--- a/setup.py
+++ b/setup.py
@@ -8,8 +8,9 @@ setup(
     author='Yin Li et al.',
     author_email='eelregit@gmail.com',
     packages=find_packages(),
+    python_requires='>=3',
     install_requires=[
-        'torch==1.1',
+        'torch',
         'numpy',
         'scipy',
         'tensorboard',