From 542fe9a1e6daf4338593535549fd0d64bc40de9a Mon Sep 17 00:00:00 2001 From: Eric Bezzam Date: Fri, 9 Aug 2024 16:42:21 +0000 Subject: [PATCH] Reformat. --- lensless/utils/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lensless/utils/dataset.py b/lensless/utils/dataset.py index 5d78664e..5832115d 100644 --- a/lensless/utils/dataset.py +++ b/lensless/utils/dataset.py @@ -1547,7 +1547,7 @@ def _get_images_pair(self, idx): lensless_np = np.array(self.dataset[idx]["lensless"]) lensed_np = np.array(self.dataset[idx]["lensed"]) background_np = np.array(self.dataset[idx]["ambient"]) if self.measured_bg else None - + if self.force_rgb: if len(lensless_np.shape) == 2: warnings.warn(f"Converting lensless[{idx}] to RGB")