Resolves Pep8 issues
@@ -67,8 +67,8 @@ class WandBLogger(StrategyLogger):
if isinstance(value, Image):
# Confusion Matrix logging
- self.wandb.log({name:
- [self.wandb.Image(value, caption="Confusion Matrix")]}, step=metric_value.x_plot)
+ self.wandb.log({name: [self.wandb.Image(
+ value, caption="Confusion Matrix")]}, step=metric_value.x_plot)
elif isinstance(value, (float, int)):
self.wandb.log({name: value}, step=metric_value.x_plot)
