horrible refactor nearly done

Jammy2211 · Dec 15, 2024 · 3e0f442 · 3e0f442
1 parent d4a8ef3
commit 3e0f442
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/autoarray/inversion/pixelization/mappers/rectangular.py b/autoarray/inversion/pixelization/mappers/rectangular.py
@@ -106,8 +106,10 @@ def pix_sub_weights(self) -> PixSubWeights:
             origin=self.source_plane_mesh_grid.origin,
         ).astype("int")
 
+        mappings = mappings.reshape((len(mappings), 1))
+
         return PixSubWeights(
-            mappings=mappings.reshape((len(mappings), 1)),
+            mappings=mappings,
             sizes=np.ones(len(mappings), dtype="int"),
             weights=np.ones(
                 (len(self.source_plane_data_grid.grid_over_sampled), 1), dtype="int"