working JAX LH Function

Author: Jammy2211

autoarray/inversion/plot/inversion_plotters.py

@@ -376,14 +376,14 @@ def subplot_of_mapper(
         )
 
         self.set_title(label="Regularization Weights (Unzoomed)")
-        # try:
-        self.figures_2d_of_pixelization(
-            pixelization_index=mapper_index,
-            regularization_weights=True,
-            zoom_to_brightest=False,
-        )
-        # except IndexError:
-        #     pass
+        try:
+            self.figures_2d_of_pixelization(
+                pixelization_index=mapper_index,
+                regularization_weights=True,
+                zoom_to_brightest=False,
+            )
+        except IndexError:
+            pass
         self.set_title(label=None)
 
         self.figures_2d_of_pixelization(

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."
+        #     )