From 80ece45d709ddd36c5e9fe063cc882c0fd8dc36d Mon Sep 17 00:00:00 2001
From: orisenbazuru <orisenbazuru@gmail.com>
Date: Tue, 28 Jul 2020 19:09:55 +0200
Subject: [PATCH] added visualization packages to setup

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

diff --git a/setup.py b/setup.py
index 2d7303f..bc1497a 100644
--- a/setup.py
+++ b/setup.py
@@ -11,6 +11,8 @@
             'pandas',
             'scipy',
             'scikit-learn',
-            'torch'
+            'torch',
+            'matplotlib',
+            'seaborn'
       ],
       zip_safe=False)
\ No newline at end of file