Replace tensorboardx with torch.utils.tensorboard (#158)

* Replace tensorboardx with torch.utils.tensorboard

* Add tensorboard to dependencies

clinicadl/clinicadl/tools/deep_learning/autoencoder_utils.py

@@ -43,7 +43,7 @@ def train(
         options: (Namespace) ensemble of other options given to the main script.
         logger: (logging object) writer to stdout and stderr
     """
-    from tensorboardX import SummaryWriter
+    from torch.utils.tensorboard import SummaryWriter
 
     if logger is None:
         logger = logging

clinicadl/clinicadl/tools/deep_learning/cnn_utils.py

@@ -51,7 +51,7 @@ def train(
     """
     from time import time
 
-    from tensorboardX import SummaryWriter
+    from torch.utils.tensorboard import SummaryWriter
 
     if logger is None:
         logger = logging

clinicadl/setup.py

@@ -1,5 +1,6 @@
-from os.path import dirname, join, abspath, pardir
-from setuptools import setup, find_packages
+from os.path import abspath, dirname, join, pardir
+
+from setuptools import find_packages, setup
 
 with open(join(dirname(__file__), "clinicadl/VERSION"), "rb") as f:
     version = f.read().decode("ascii").strip()
@@ -35,6 +36,6 @@
         "Intended Audience :: Developers",
         "Programming Language :: Python",
     ],
-    install_requires=["numpy>=1.17", "clinica>=0.3.8", "tensorboardX"],
+    install_requires=["numpy>=1.17", "clinica>=0.3.8", "torch", "tensorboard"],
     python_requires=">=3.6",
 )

requirements.txt

@@ -7,6 +7,7 @@
 clinica>=0.3.6
 nilearn
 nipy
-tensorboardX
+torch
 pandas
 colorama
+tensorboard