diff --git a/pytorch_lightning/loggers/neptune.py b/pytorch_lightning/loggers/neptune.py index a4f0a3821adef..75ac422d945d0 100644 --- a/pytorch_lightning/loggers/neptune.py +++ b/pytorch_lightning/loggers/neptune.py @@ -169,7 +169,11 @@ def log_hyperparams(self, params: argparse.Namespace): self.experiment.set_property(f'param__{key}', val) @rank_zero_only - def log_metrics(self, metrics: Dict[str, Union[torch.Tensor, float]], step: Optional[int] = None): + def log_metrics( + self, + metrics: Dict[str, Union[torch.Tensor, float]], + step: Optional[int] = None + ): """Log metrics (numeric values) in Neptune experiments Args: @@ -198,7 +202,12 @@ def version(self) -> str: return self.experiment.id @rank_zero_only - def log_metric(self, metric_name: str, metric_value: Union[torch.Tensor, float, str], step: Optional[int] = None): + def log_metric( + self, + metric_name: str, + metric_value: Union[torch.Tensor, float, str], + step: Optional[int] = None + ): """Log metrics (numeric values) in Neptune experiments Args: diff --git a/pytorch_lightning/loggers/tensorboard.py b/pytorch_lightning/loggers/tensorboard.py index c444668e58006..837e53b37e575 100644 --- a/pytorch_lightning/loggers/tensorboard.py +++ b/pytorch_lightning/loggers/tensorboard.py @@ -44,7 +44,10 @@ class TensorBoardLogger(LightningLoggerBase): """ NAME_CSV_TAGS = 'meta_tags.csv' - def __init__(self, save_dir: str, name: Optional[str] = "default", version: Optional[Union[int, str]] = None, **kwargs): + def __init__( + self, save_dir: str, name: Optional[str] = "default", + version: Optional[Union[int, str]] = None, **kwargs + ): super().__init__() self.save_dir = save_dir self._name = name diff --git a/tests/loggers/test_tensorboard.py b/tests/loggers/test_tensorboard.py index cc41381c39f48..b3f3d19242c8c 100644 --- a/tests/loggers/test_tensorboard.py +++ b/tests/loggers/test_tensorboard.py @@ -76,7 +76,8 @@ def test_tensorboard_named_version(tmpdir): logger.log_hyperparams({"a": 1, "b": 2}) # Force data to be written assert logger.version == expected_version - # Could also test existence of the directory but this fails in the "minimum requirements" test setup + # Could also test existence of the directory but this fails + # in the "minimum requirements" test setup def test_tensorboard_no_name(tmpdir):