removed checks for image dimensions

LSST-strong-lensing · Feb 24, 2025 · 1802e37 · 1802e37
1 parent 1595f65
commit 1802e37
Showing 1 changed file with 0 additions and 10 deletions.
diff --git a/slsim/Sources/source.py b/slsim/Sources/source.py
@@ -587,16 +587,6 @@ def kwargs_extended_source_light(self, center_lens, draw_area, band=None):
             z_image = self.source_dict["z_data"]
             pixel_width = self.source_dict["pixel_width_data"]
             pixel_width *= z_scale_factor(z_old=z_image, z_new=self.redshift)
-
-            # Ensure the image is 2D
-            image = self.source_dict["image"]
-            if image.ndim > 2:
-                image = image.squeeze()  # Remove any singleton dimensions
-            if image.ndim != 2:
-                raise ValueError(
-                    f"Interpolated image must be 2D, got shape {image.shape}"
-                )
-
             kwargs_extended_source = [
                 {
                     "magnitude": mag_source,