Reformatting with attrs, config for encoder and decoder

1 files changed, 2 insertions, 6 deletions

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],
                     )
                 ]
             }