From c032ffb05a7ed86f8fe5d596f94e8997c558cae8 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Wed, 28 Jul 2021 15:14:55 +0200
Subject: Reformatting with attrs, config for encoder and decoder

---
 training/callbacks/wandb_callbacks.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

(limited to 'training/callbacks')

diff --git a/training/callbacks/wandb_callbacks.py b/training/callbacks/wandb_callbacks.py
index d9d81f6..451b0d5 100644
--- a/training/callbacks/wandb_callbacks.py
+++ b/training/callbacks/wandb_callbacks.py
@@ -178,13 +178,9 @@ class LogReconstuctedImages(Callback):
         experiment.log(
             {
                 f"Reconstructions/{experiment.name}/{stage}": [
-                    [
-                        wandb.Image(img),
-                        wandb.Image(rec),
-                    ]
+                    [wandb.Image(img), wandb.Image(rec),]
                     for img, rec in zip(
-                        imgs[: self.num_samples],
-                        reconstructions[: self.num_samples],
+                        imgs[: self.num_samples], reconstructions[: self.num_samples],
                     )
                 ]
             }
-- 
cgit v1.2.3-70-g09d2