From edb848e0db2cfb06e38d40d1de3662d7b5a84349 Mon Sep 17 00:00:00 2001 From: Ethan Harris Date: Mon, 2 Mar 2020 16:16:31 +0000 Subject: [PATCH] Fix style --- pytorch_lightning/loggers/comet.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/loggers/comet.py b/pytorch_lightning/loggers/comet.py index eaf18621c0ce4..7e146a2037dd9 100644 --- a/pytorch_lightning/loggers/comet.py +++ b/pytorch_lightning/loggers/comet.py @@ -166,7 +166,11 @@ def log_hyperparams(self, params: argparse.Namespace): self.experiment.log_parameters(vars(params)) @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 + ): # Comet.ml expects metrics to be a dictionary of detached tensors on CPU for key, val in metrics.items(): if is_tensor(val):