From 413221af7ba103897db68a02c80e988d58b3aea8 Mon Sep 17 00:00:00 2001
From: Yin Li <eelregit@gmail.com>
Date: Thu, 6 May 2021 12:38:57 -0700
Subject: [PATCH] Add matplotlib and tensorboard requirements

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

diff --git a/setup.py b/setup.py
index 9636a46..3656c05 100644
--- a/setup.py
+++ b/setup.py
@@ -15,8 +15,7 @@ setup(
         'torch>=1.2',
         'numpy',
         'scipy',
+        'matplotlib',
+        'tensorboard',
     ],
-    extras_require={
-        "visualization":  ["tensorboard"],
-    },
 )