mocks

Author: Borda

pytorch_lightning/loggers/__init__.py

@@ -24,25 +24,25 @@
     'CSVLogger',
 ]
 
-from pytorch_lightning.loggers.comet import COMET_AVAILABLE, CometLogger
-from pytorch_lightning.loggers.mlflow import MLFLOW_AVAILABLE, MLFlowLogger
-from pytorch_lightning.loggers.neptune import NEPTUNE_AVAILABLE, NeptuneLogger
-from pytorch_lightning.loggers.test_tube import TESTTUBE_AVAILABLE, TestTubeLogger
-from pytorch_lightning.loggers.wandb import WANDB_AVAILABLE, WandbLogger
+from pytorch_lightning.loggers.comet import _COMET_AVAILABLE, CometLogger
+from pytorch_lightning.loggers.mlflow import _MLFLOW_AVAILABLE, MLFlowLogger
+from pytorch_lightning.loggers.neptune import _NEPTUNE_AVAILABLE, NeptuneLogger
+from pytorch_lightning.loggers.test_tube import _TESTTUBE_AVAILABLE, TestTubeLogger
+from pytorch_lightning.loggers.wandb import _WANDB_AVAILABLE, WandbLogger
 
-if COMET_AVAILABLE:
+if _COMET_AVAILABLE:
     __all__.append('CometLogger')
     # needed to prevent ImportError and duplicated logs.
     environ["COMET_DISABLE_AUTO_LOGGING"] = "1"
 
-if MLFLOW_AVAILABLE:
+if _MLFLOW_AVAILABLE:
     __all__.append('MLFlowLogger')
 
-if NEPTUNE_AVAILABLE:
+if _NEPTUNE_AVAILABLE:
     __all__.append('NeptuneLogger')
 
-if TESTTUBE_AVAILABLE:
+if _TESTTUBE_AVAILABLE:
     __all__.append('TestTubeLogger')
 
-if WANDB_AVAILABLE:
+if _WANDB_AVAILABLE:
     __all__.append('WandbLogger')

pytorch_lightning/loggers/comet.py

@@ -29,9 +29,9 @@
 from pytorch_lightning.utilities import rank_zero_only, _module_available
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 
-COMET_AVAILABLE = _module_available("comet_ml")
+_COMET_AVAILABLE = _module_available("comet_ml")
 
-if COMET_AVAILABLE:
+if _COMET_AVAILABLE:
     import comet_ml
     from comet_ml import ExistingExperiment as CometExistingExperiment
     from comet_ml import Experiment as CometExperiment
@@ -42,6 +42,8 @@
     except ImportError:  # pragma: no-cover
         # For more information, see: https://www.comet.ml/docs/python-sdk/releases/#release-300
         from comet_ml.papi import API  # pragma: no-cover
+else:
+    comet_ml = None  # needed for test mocks, these tests shall be updated
 
 
 class CometLogger(LightningLoggerBase):
@@ -126,7 +128,7 @@ def __init__(
         prefix: str = '',
         **kwargs
     ):
-        if not COMET_AVAILABLE:
+        if not comet_ml:
             raise ImportError(
                 "You want to use `comet_ml` logger which is not installed yet,"
                 " install it with `pip install comet-ml`."

pytorch_lightning/loggers/mlflow.py

@@ -26,10 +26,13 @@
 from pytorch_lightning.loggers.base import LightningLoggerBase, rank_zero_experiment
 from pytorch_lightning.utilities import rank_zero_only, rank_zero_warn, _module_available
 
-MLFLOW_AVAILABLE = _module_available("mlflow")
+_MLFLOW_AVAILABLE = _module_available("mlflow")
 
-if MLFLOW_AVAILABLE:
+if _MLFLOW_AVAILABLE:
+    import mlflow
     from mlflow.tracking import MlflowClient
+else:
+    mlflow = None
 
 
 LOCAL_FILE_URI_PREFIX = "file:"
@@ -90,7 +93,7 @@ def __init__(
         save_dir: Optional[str] = './mlruns',
         prefix: str = '',
     ):
-        if not MLFLOW_AVAILABLE:
+        if not mlflow:
             raise ImportError('You want to use `mlflow` logger which is not installed yet,'
                               ' install it with `pip install mlflow`.')
         super().__init__()

pytorch_lightning/loggers/neptune.py

@@ -26,11 +26,13 @@
 from pytorch_lightning.loggers.base import LightningLoggerBase, rank_zero_experiment
 from pytorch_lightning.utilities import rank_zero_only, _module_available
 
-NEPTUNE_AVAILABLE = _module_available("neptune")
+_NEPTUNE_AVAILABLE = _module_available("neptune")
 
-if NEPTUNE_AVAILABLE:
+if _NEPTUNE_AVAILABLE:
     import neptune
     from neptune.experiments import Experiment as NeptuneExperiment
+else:
+    neptune = None  # needed for test mocks, these tests shall be updated
 
 
 class NeptuneLogger(LightningLoggerBase):
@@ -184,7 +186,7 @@ def __init__(
         prefix: str = '',
         **kwargs
     ):
-        if not NEPTUNE_AVAILABLE:
+        if not neptune:
             raise ImportError('You want to use `neptune` logger which is not installed yet,'
                               ' install it with `pip install neptune-client`.')
         super().__init__()

pytorch_lightning/loggers/test_tube.py

@@ -24,9 +24,9 @@
 from pytorch_lightning.utilities import _module_available
 from pytorch_lightning.utilities.distributed import rank_zero_only, rank_zero_warn
 
-TESTTUBE_AVAILABLE = _module_available("test_tube")
+_TESTTUBE_AVAILABLE = _module_available("test_tube")
 
-if TESTTUBE_AVAILABLE:
+if _TESTTUBE_AVAILABLE:
     from test_tube import Experiment as TTExperiment
 
 
@@ -89,7 +89,7 @@ def __init__(
         log_graph: bool = False,
         prefix: str = '',
     ):
-        if not TESTTUBE_AVAILABLE:
+        if not _TESTTUBE_AVAILABLE:
             raise ImportError('You want to use `test_tube` logger which is not installed yet,'
                               ' install it with `pip install test-tube`.')
         super().__init__()

pytorch_lightning/loggers/wandb.py

@@ -25,11 +25,13 @@
 from pytorch_lightning.loggers.base import LightningLoggerBase, rank_zero_experiment
 from pytorch_lightning.utilities import rank_zero_only, _module_available
 
-WANDB_AVAILABLE = _module_available("wandb")
+_WANDB_AVAILABLE = _module_available("wandb")
 
-if WANDB_AVAILABLE:
+if _WANDB_AVAILABLE:
     import wandb
     from wandb.wandb_run import Run as WBRun
+else:
+    wandb = None  # needed for test mocks, these tests shall be updated
 
 
 class WandbLogger(LightningLoggerBase):
@@ -88,7 +90,7 @@ def __init__(
         prefix: str = '',
         **kwargs
     ):
-        if not WANDB_AVAILABLE:
+        if not wandb:
             raise ImportError('You want to use `wandb` logger which is not installed yet,'  # pragma: no-cover
                               ' install it with `pip install wandb`.')
         super().__init__()