fix ploting bug

Author: Jammy2211

autoarray/inversion/plot/mapper_plotters.py

@@ -134,16 +134,16 @@ def plot_source_from(
         auto_labels
             The labels given to the figure.
         """
-        # try:
-        self.mat_plot_2d.plot_mapper(
-            mapper=self.mapper,
-            visuals_2d=self.visuals_2d,
-            auto_labels=auto_labels,
-            pixel_values=pixel_values,
-            zoom_to_brightest=zoom_to_brightest,
-            interpolate_to_uniform=interpolate_to_uniform,
-        )
-        # except ValueError:
-        #     logger.info(
-        #         "Could not plot the source-plane via the Mapper because of a ValueError."
-        #     )
+        try:
+            self.mat_plot_2d.plot_mapper(
+                mapper=self.mapper,
+                visuals_2d=self.visuals_2d,
+                auto_labels=auto_labels,
+                pixel_values=pixel_values,
+                zoom_to_brightest=zoom_to_brightest,
+                interpolate_to_uniform=interpolate_to_uniform,
+            )
+        except ValueError:
+            logger.info(
+                "Could not plot the source-plane via the Mapper because of a ValueError."
+            )

autoarray/plot/wrap/two_d/delaunay_drawer.py

@@ -74,6 +74,11 @@ def draw_delaunay_pixels(
 
         simplices = mapper.delaunay.simplices
 
+        # Remove padded -1 values required for JAX
+        simplices = np.asarray(simplices)
+        valid_mask = np.all(simplices >= 0, axis=1)
+        simplices = simplices[valid_mask]
+
         facecolors = facecolors_from(values=pixel_values, simplices=simplices)
 
         norm = cmap.norm_from(array=pixel_values, use_log10=use_log10)