Change tensorboard to optional dependency

2020-01-23 09:55:01 -05:00 · 2020-01-23 09:55:01 -05:00 · 581ad5d60b
commit 581ad5d60b
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',
    ]