added visualization packages to setup

uzh-dqbm-cmi · Jul 28, 2020 · 80ece45 · 80ece45
1 parent 0d6dc4e
commit 80ece45
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -11,6 +11,8 @@
             'pandas',
             'scipy',
             'scikit-learn',
-            'torch'
+            'torch',
+            'matplotlib',
+            'seaborn'
       ],
       zip_safe=False)