PR L5: Remove Visuals from PyAutoLens plotters; use standalone plot_array() with direct lines=/positions= kwargs

- abstract_plotters.py: Plotter._plot_array(array, auto_labels, lines=, positions=, visuals_2d=)
  uses autoarray.plot.plots.array.plot_array() directly; _to_lines()/_to_positions() helpers
- tracer_util.py: add critical_curves_from(), caustics_from(), lines_of_planes_from()
  computing curves via LensCalc, returning plain numpy arrays; legacy visuals_2d_of_planes_list_from retained
- tracer_plotters.py: cached _tangential/radial_critical_curves/caustics properties;
  figures_2d passes lines= directly; galaxies_plotter_from builds Visuals2D from cached curves
- fit_imaging_plotters.py: lines_of_planes via tracer_util; _lines_for_plane() helper;
  figures_2d/_of_planes pass lines= not visuals_2d=
- fit_interferometer_plotters.py: same pattern; dirty_model_image uses lines=
- conftest.py: PlotPatch also patches matplotlib.figure.Figure.savefig
- pyproject.toml: pin autogalaxy to claude/refactor-plotting-module-3ZdD8

https://claude.ai/code/session_01CzJBy8KvFXiNchoNdk5i9k

Author: claude

autolens/imaging/plot/fit_imaging_plotters.py

@@ -1,6 +1,6 @@
 import copy
 import numpy as np
-from typing import Optional
+from typing import Optional, List
 
 from autoconf import conf
 
@@ -10,7 +10,7 @@
 from autoarray.plot.auto_labels import AutoLabels
 from autoarray.fit.plot.fit_imaging_plotters import FitImagingPlotterMeta
 
-from autolens.plot.abstract_plotters import Plotter
+from autolens.plot.abstract_plotters import Plotter, _to_lines
 from autolens.imaging.fit_imaging import FitImaging
 from autolens.lens.plot.tracer_plotters import TracerPlotter
 
@@ -63,19 +63,45 @@ def __init__(
         self.residuals_symmetric_cmap = residuals_symmetric_cmap
 
         self._visuals_2d_of_planes_list = visuals_2d_of_planes_list
+        self._lines_of_planes = None
 
     @property
-    def visuals_2d_of_planes_list(self):
+    def _lensing_grid(self):
+        return self.fit.grids.lp.mask.derive_grid.all_false
 
-        if self._visuals_2d_of_planes_list is None:
+    @property
+    def lines_of_planes(self) -> List[List]:
+        """Lists of line overlays (numpy arrays) per plane: critical curves for
+        plane 0, caustics for higher planes."""
+        if self._lines_of_planes is None:
+            self._lines_of_planes = tracer_util.lines_of_planes_from(
+                tracer=self.fit.tracer,
+                grid=self._lensing_grid,
+            )
+        return self._lines_of_planes
 
+    @property
+    def visuals_2d_of_planes_list(self):
+        """Legacy property: returns Visuals2D objects per plane for backward-
+        compatible callers (e.g. InversionPlotter)."""
+        if self._visuals_2d_of_planes_list is None:
             self._visuals_2d_of_planes_list = tracer_util.visuals_2d_of_planes_list_from(
                 tracer=self.fit.tracer,
-                grid=self.fit.grids.lp.mask.derive_grid.all_false,
+                grid=self._lensing_grid,
             )
-
         return self._visuals_2d_of_planes_list
 
+    def _lines_for_plane(
+        self, plane_index: int, remove_critical_caustic: bool = False
+    ) -> Optional[List]:
+        """Return the line overlays for a given plane, or None if suppressed."""
+        if remove_critical_caustic:
+            return None
+        try:
+            return self.lines_of_planes[plane_index] or None
+        except IndexError:
+            return None
+
     def visuals_2d_from(
         self, plane_index: Optional[int] = None, remove_critical_caustic: bool = False
     ) -> aplt.Visuals2D:
@@ -255,11 +281,13 @@ def figures_2d_of_planes(
 
                 self._plot_array(
                     array=self.fit.subtracted_images_of_planes_list[plane_index],
-                    visuals_2d=self.visuals_2d_from(
-                        plane_index=plane_index,
-                        remove_critical_caustic=remove_critical_caustic,
-                    ),
                     auto_labels=aplt.AutoLabels(title=title, filename=filename),
+                    lines=_to_lines(
+                        self._lines_for_plane(
+                            plane_index=plane_index,
+                            remove_critical_caustic=remove_critical_caustic,
+                        )
+                    ),
                 )
 
             if model_image:
@@ -281,11 +309,13 @@ def figures_2d_of_planes(
 
                 self._plot_array(
                     array=self.fit.model_images_of_planes_list[plane_index],
-                    visuals_2d=self.visuals_2d_from(
-                        plane_index=plane_index,
-                        remove_critical_caustic=remove_critical_caustic,
-                    ),
                     auto_labels=aplt.AutoLabels(title=title, filename=filename),
+                    lines=_to_lines(
+                        self._lines_for_plane(
+                            plane_index=plane_index,
+                            remove_critical_caustic=remove_critical_caustic,
+                        )
+                    ),
                 )
 
             if plane_image:
@@ -881,7 +911,6 @@ def figures_2d(
 
             self._plot_array(
                 array=self.fit.data,
-                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(title="Data", filename=f"data{suffix}"),
             )
 
@@ -892,7 +921,6 @@ def figures_2d(
 
             self._plot_array(
                 array=self.fit.noise_map,
-                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(
                     title="Noise-Map", filename=f"noise_map{suffix}"
                 ),
@@ -902,7 +930,6 @@ def figures_2d(
 
             self._plot_array(
                 array=self.fit.signal_to_noise_map,
-                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(
                     title="Signal-To-Noise Map",
                     filename=f"signal_to_noise_map{suffix}",
@@ -916,10 +943,10 @@ def figures_2d(
 
             self._plot_array(
                 array=self.fit.model_data,
-                visuals_2d=self.visuals_2d_from(plane_index=0),
                 auto_labels=AutoLabels(
                     title="Model Image", filename=f"model_image{suffix}"
                 ),
+                lines=_to_lines(self._lines_for_plane(plane_index=0)),
             )
 
             if use_source_vmax:
@@ -935,7 +962,6 @@ def figures_2d(
 
             self._plot_array(
                 array=self.fit.residual_map,
-                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(
                     title="Residual Map", filename=f"residual_map{suffix}"
                 ),
@@ -945,7 +971,6 @@ def figures_2d(
 
             self._plot_array(
                 array=self.fit.normalized_residual_map,
-                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(
                     title="Normalized Residual Map",
                     filename=f"normalized_residual_map{suffix}",
@@ -958,7 +983,6 @@ def figures_2d(
 
             self._plot_array(
                 array=self.fit.chi_squared_map,
-                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(
                     title="Chi-Squared Map",
                     filename=f"chi_squared_map{suffix}",
@@ -969,7 +993,6 @@ def figures_2d(
 
             self._plot_array(
                 array=self.fit.residual_flux_fraction_map,
-                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(
                     title="Residual Flux Fraction Map",
                     filename=f"residual_flux_fraction_map{suffix}",

autolens/interferometer/plot/fit_interferometer_plotters.py

@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Optional, List
 
 from autoconf import conf
 
@@ -11,7 +11,7 @@
 from autolens.interferometer.fit_interferometer import FitInterferometer
 from autolens.lens.tracer import Tracer
 from autolens.lens.plot.tracer_plotters import TracerPlotter
-from autolens.plot.abstract_plotters import Plotter
+from autolens.plot.abstract_plotters import Plotter, _to_lines
 
 from autolens.lens import tracer_util
 
@@ -79,20 +79,42 @@ def __init__(
         )
 
         self._visuals_2d_of_planes_list = visuals_2d_of_planes_list
+        self._lines_of_planes = None
 
     @property
-    def visuals_2d_of_planes_list(self):
+    def _lensing_grid(self):
+        return self.fit.grids.lp.mask.derive_grid.all_false
 
+    @property
+    def lines_of_planes(self) -> List[List]:
+        if self._lines_of_planes is None:
+            self._lines_of_planes = tracer_util.lines_of_planes_from(
+                tracer=self.fit.tracer,
+                grid=self._lensing_grid,
+            )
+        return self._lines_of_planes
+
+    @property
+    def visuals_2d_of_planes_list(self):
         if self._visuals_2d_of_planes_list is None:
             self._visuals_2d_of_planes_list = (
                 tracer_util.visuals_2d_of_planes_list_from(
                     tracer=self.fit.tracer,
-                    grid=self.fit.grids.lp.mask.derive_grid.all_false,
+                    grid=self._lensing_grid,
                 )
             )
-
         return self._visuals_2d_of_planes_list
 
+    def _lines_for_plane(
+        self, plane_index: int, remove_critical_caustic: bool = False
+    ) -> Optional[List]:
+        if remove_critical_caustic:
+            return None
+        try:
+            return self.lines_of_planes[plane_index] or None
+        except IndexError:
+            return None
+
     def visuals_2d_from(
         self, plane_index: Optional[int] = None, remove_critical_caustic: bool = False
     ) -> aplt.Visuals2D:
@@ -289,10 +311,10 @@ def figures_2d(
         if dirty_model_image:
             self._plot_array(
                 array=self.fit.dirty_model_image,
-                visuals_2d=self.visuals_2d_of_planes_list[0],
                 auto_labels=AutoLabels(
                     title="Dirty Model Image", filename="dirty_model_image_2d"
                 ),
+                lines=_to_lines(self._lines_for_plane(plane_index=0)),
             )
 
     def figures_2d_of_planes(