add docstring

Author: edward-io

tests/loggers/test_all.py

@@ -31,7 +31,7 @@
     TestTubeLogger,
     WandbLogger,
 )
-from pytorch_lightning.loggers.base import DummyExperiment, LightningLoggerBase
+from pytorch_lightning.loggers.base import DummyExperiment
 from tests.helpers import BoringModel
 from tests.helpers.runif import RunIf
 from tests.loggers.test_comet import _patch_comet_atexit

tests/trainer/logging_/test_distributed_logging.py

@@ -23,6 +23,11 @@
 
 
 class AllRankLogger(LightningLoggerBase):
+    """
+    Logger to test all-rank logging (i.e. not just rank 0).
+    Logs are saved to local variable `logs`.
+    """
+
     def __init__(self):
         super().__init__()
         self.logs = {}
@@ -32,9 +37,7 @@ def experiment(self) -> Any:
         return self.exp
 
     def log_metrics(self, metrics: Dict[str, float], step: Optional[int] = None):
-        print("appending", metrics, step)
         self.logs.update(metrics)
-        print(self.logs)
 
     def version(self) -> Union[int, str]:
         return 1