From 38c9102d135ddac126e1bf02ea823b2f9e9b4ac4 Mon Sep 17 00:00:00 2001
From: William Falcon <waf2107@columbia.edu>
Date: Sat, 29 Jun 2019 15:35:05 -0400
Subject: [PATCH] required tensorflow for tensorboardx install

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

diff --git a/setup.py b/setup.py
index 81ed622e2c..88f021b592 100755
--- a/setup.py
+++ b/setup.py
@@ -19,7 +19,8 @@ setup(
     install_requires=[
         "torch>=1.0.0",
         "tqdm",
-        "test-tube",
+        "test-tube>=0.641",
+        "tensorflow>=1.14.0"
     ],
     packages=find_packages(),
     long_description=open("README.md", encoding="utf-8").read(),