Apply yapf

Lightning-Universe · Jan 17, 2021 · 1bf5420 · 1bf5420
1 parent ed7c4bf
commit 1bf5420
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pl_bolts/callbacks/vision/image_generation.py b/pl_bolts/callbacks/vision/image_generation.py
@@ -90,6 +90,4 @@ def on_epoch_end(self, trainer: Trainer, pl_module: LightningModule) -> None:
             pad_value=self.pad_value,
         )
         str_title = f"{pl_module.__class__.__name__}_images"
-        trainer.logger.experiment.add_image(
-            str_title, grid, global_step=trainer.global_step
-        )
+        trainer.logger.experiment.add_image(str_title, grid, global_step=trainer.global_step)