Formatting.

LCAV · Apr 30, 2024 · 8cdb1eb · 8cdb1eb
1 parent a1a94c1
commit 8cdb1eb
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/lensless/eval/benchmark.py b/lensless/eval/benchmark.py
@@ -188,7 +188,9 @@ def benchmark(
 
                         if use_wandb:
                             assert epoch is not None, "epoch must be provided for wandb logging"
-                            log_key = f"{_batch_idx}_{label}" if label is not None else f"{_batch_idx}"
+                            log_key = (
+                                f"{_batch_idx}_{label}" if label is not None else f"{_batch_idx}"
+                            )
                             wandb.log({log_key: wandb.Image(fp)}, step=epoch)
 
             # normalization

diff --git a/lensless/recon/utils.py b/lensless/recon/utils.py
@@ -459,7 +459,7 @@ def __init__(
             plot_image(psf_np, gamma=self.gamma)
             fp_plot = "psf_original_plot.png"
             plt.savefig(fp_plot)
-            
+
             if self.use_wandb:
                 wandb.log({"psf": wandb.Image(fp)}, step=0)
                 wandb.log({"psf_plot": wandb.Image(fp_plot)}, step=0)

diff --git a/lensless/utils/dataset.py b/lensless/utils/dataset.py
@@ -1023,7 +1023,7 @@ def __init__(
         split,
         n_files=None,
         psf=None,
-        rotate=False,   # just the lensless image
+        rotate=False,  # just the lensless image
         downsample=1,
         downsample_lensed=1,
         display_res=None,