fix

Author: Borda

pytorch_lightning/loggers/test_tube.py

@@ -27,7 +27,7 @@
 TESTTUBE_AVAILABLE = _module_available("test_tube")
 
 if TESTTUBE_AVAILABLE:
-    from test_tube import Experiment
+    from test_tube import Experiment as TTExperiment
 
 
 class TestTubeLogger(LightningLoggerBase):
@@ -105,7 +105,7 @@ def __init__(
 
     @property
     @rank_zero_experiment
-    def experiment(self) -> Experiment:
+    def experiment(self) -> "TTExperiment":
         r"""
 
         Actual TestTube object. To use TestTube features in your
@@ -119,7 +119,7 @@ def experiment(self) -> Experiment:
         if self._experiment is not None:
             return self._experiment
 
-        self._experiment = Experiment(
+        self._experiment = TTExperiment(
             save_dir=self.save_dir,
             name=self._name,
             debug=self.debug,

pytorch_lightning/loggers/wandb.py

@@ -29,7 +29,7 @@
 
 if WANDB_AVAILABLE:
     import wandb
-    from wandb.wandb_run import Run
+    from wandb.wandb_run import Run as WBRun
 
 
 class WandbLogger(LightningLoggerBase):
@@ -116,7 +116,7 @@ def __getstate__(self):
 
     @property
     @rank_zero_experiment
-    def experiment(self) -> Run:
+    def experiment(self) -> "WBRun":
         r"""
 
         Actual wandb object. To use wandb features in your