Hiblert works for data resolution swap

Jammy2211 · Apr 30, 2024 · 3fb1a57 · 3fb1a57
1 parent 767b86d
commit 3fb1a57
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/autoarray/inversion/pixelization/image_mesh/hilbert.py b/autoarray/inversion/pixelization/image_mesh/hilbert.py
@@ -161,7 +161,12 @@ def image_and_grid_from(image, mask, mask_radius, pixel_scales, hilbert_length):
     image associated to that grid.
     """
 
-    shape_nnn = np.shape(mask)[0]
+    # For multi wavelength fits the input image may be a different resolution than the mask.
+
+    try:
+        shape_nnn = np.shape(image.native)[0]
+    except AttributeError:
+        shape_nnn = np.shape(mask)[0]
 
     grid = Grid2D.uniform(
         shape_native=(shape_nnn, shape_nnn),