Change tensorboard to optional dependency

2020-01-23 09:55:01 -05:00 · 2020-01-23 09:55:01 -05:00 · 6252629631
commit 6252629631
parent 6c030f1eb1
1 changed files with 3 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@ -13,8 +13,10 @@ setup(
        'torch',
        'numpy',
        'scipy',
-        'tensorboard',
    ],
+    extras_require={
+        "visualization":  ["tensorboard"],
+    }
    scripts=[
        'scripts/m2m.py',
    ]