From b09efbb8abd2622fefba00b4762ba663671f6389 Mon Sep 17 00:00:00 2001 From: Matt Tancik Date: Thu, 22 Dec 2022 10:05:50 -0800 Subject: [PATCH] lint --- nerfstudio/data/datasets/base_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/data/datasets/base_dataset.py b/nerfstudio/data/datasets/base_dataset.py index a509acd89..44793fbb0 100644 --- a/nerfstudio/data/datasets/base_dataset.py +++ b/nerfstudio/data/datasets/base_dataset.py @@ -66,7 +66,7 @@ class InputDataset(Dataset): pil_image = pil_image.resize(newsize, resample=Image.BILINEAR) image = np.array(pil_image, dtype="uint8") # shape is (h, w) or (h, w, 3 or 4) if len(image.shape) == 2: - image = (image[:, :, None].repeat(3, axis=2)) + image = image[:, :, None].repeat(3, axis=2) assert len(image.shape) == 3 assert image.dtype == np.uint8 assert image.shape[2] in [3, 4], f"Image shape of {image.shape} is in correct."