feat: use tight_layout wrapper for PYAUTO_FAST_PLOTS support

Replace all plt.tight_layout() calls with the wrapper from
autoarray.plot.utils that skips the expensive layout pass when
PYAUTO_FAST_PLOTS=1 is set.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: 2 people

autogalaxy/ellipse/plot/fit_ellipse_plots.py

@@ -5,7 +5,7 @@
 
 import autoarray as aa
 from autoarray import plot as aplt
-from autoarray.plot.utils import conf_subplot_figsize
+from autoarray.plot.utils import conf_subplot_figsize, tight_layout
 
 from autogalaxy.ellipse.plot import fit_ellipse_plot_util
 from autogalaxy.ellipse.fit_ellipse import FitEllipse
@@ -159,7 +159,7 @@ def subplot_fit_ellipse(
     )
     _plot_ellipse_residuals(fit_list=fit_list, for_subplot=True, ax=axes[1])
 
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, "fit_ellipse", output_format)
 
 
@@ -242,5 +242,5 @@ def subplot_ellipse_errors(
             ax=axes_flat[i],
         )
 
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, "ellipse_errors", output_format)

autogalaxy/galaxy/plot/adapt_plots.py

@@ -3,7 +3,7 @@
 from typing import Dict
 
 import autoarray as aa
-from autoarray.plot.utils import conf_subplot_figsize
+from autoarray.plot.utils import conf_subplot_figsize, tight_layout
 
 from autogalaxy.galaxy.galaxy import Galaxy
 from autogalaxy.plot.plot_utils import plot_array, _save_subplot
@@ -58,7 +58,7 @@ def subplot_adapt_images(
             ax=axes_list[i],
         )
 
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, "adapt_images", output_format)
 
 

autogalaxy/galaxy/plot/galaxies_plots.py

@@ -5,7 +5,7 @@
 
 from autogalaxy.galaxy.galaxies import Galaxies
 from autogalaxy.plot.plot_utils import _to_lines, _to_positions, plot_array, plot_grid, _save_subplot, _critical_curves_from
-from autoarray.plot.utils import hide_unused_axes, conf_subplot_figsize
+from autoarray.plot.utils import hide_unused_axes, conf_subplot_figsize, tight_layout
 from autogalaxy import exc
 
 
@@ -156,7 +156,7 @@ def _defl_x():
         )
 
     hide_unused_axes(axes_flat)
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, auto_filename, output_format)
 
 
@@ -213,7 +213,7 @@ def subplot_galaxy_images(
             ax=axes_flat[i],
         )
 
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, "galaxy_images", output_format)
 
 

autogalaxy/galaxy/plot/galaxy_plots.py

@@ -3,7 +3,7 @@
 from typing import TYPE_CHECKING
 
 import autoarray as aa
-from autoarray.plot.utils import conf_subplot_figsize
+from autoarray.plot.utils import conf_subplot_figsize, tight_layout
 
 from autogalaxy.galaxy.galaxy import Galaxy
 from autogalaxy.profiles.light.abstract import LightProfile
@@ -60,7 +60,7 @@ def subplot_of_light_profiles(
             ax=axes_flat[i],
         )
 
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, "image", output_format)
 
 
@@ -142,5 +142,5 @@ def _deflections_x(mp):
                 ax=axes_flat[i],
             )
 
-        plt.tight_layout()
+        tight_layout()
         _save_subplot(fig, output_path, name, output_format)

autogalaxy/imaging/plot/fit_imaging_plots.py

@@ -3,7 +3,7 @@
 
 import autoarray as aa
 from autoconf.fitsable import hdu_list_for_output_from
-from autoarray.plot.utils import conf_subplot_figsize
+from autoarray.plot.utils import conf_subplot_figsize, tight_layout
 
 from autogalaxy.imaging.fit_imaging import FitImaging
 from autogalaxy.plot.plot_utils import plot_array, _save_subplot
@@ -64,7 +64,7 @@ def subplot_fit(
             ax=axes_flat[i],
         )
 
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, "fit", output_format)
 
 
@@ -128,7 +128,7 @@ def subplot_of_galaxy(
             ax=axes_flat[i],
         )
 
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, f"of_galaxy_{galaxy_index}", output_format)
 
 
@@ -165,7 +165,7 @@ def subplot_fit_imaging_list(
         plot_array(array=fit.model_data, title="Model Image", ax=axes[i][2])
         plot_array(array=fit.normalized_residual_map, title="Normalized Residual Map", cb_unit=r"$\sigma$", ax=axes[i][3])
         plot_array(array=fit.chi_squared_map, title="Chi-Squared Map", cb_unit=r"$\chi^2$", ax=axes[i][4])
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, output_filename, output_format)
 
 

autogalaxy/interferometer/plot/fit_interferometer_plots.py

@@ -5,7 +5,7 @@
 import autoarray as aa
 from autoconf.fitsable import hdu_list_for_output_from
 from autoarray.plot import plot_visibilities_1d
-from autoarray.plot.utils import conf_subplot_figsize
+from autoarray.plot.utils import conf_subplot_figsize, tight_layout
 
 from autogalaxy.interferometer.fit_interferometer import FitInterferometer
 from autogalaxy.galaxy.plot import galaxies_plots
@@ -54,7 +54,7 @@ def subplot_fit(
     for i, (vis, title) in enumerate(panels):
         plot_visibilities_1d(vis, axes_flat[i], title)
 
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, "fit", output_format)
 
 
@@ -110,7 +110,7 @@ def subplot_fit_dirty_images(
             ax=axes_flat[i],
         )
 
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, "fit_dirty_images", output_format)
 
 
@@ -175,7 +175,7 @@ def subplot_fit_real_space(
                 use_log10=use_log10,
                 ax=axes_flat[i],
             )
-        plt.tight_layout()
+        tight_layout()
         _save_subplot(fig, output_path, "fit_real_space", output_format)
 
 

autogalaxy/profiles/plot/basis_plots.py

@@ -2,7 +2,7 @@
 import numpy as np
 
 import autoarray as aa
-from autoarray.plot.utils import conf_subplot_figsize
+from autoarray.plot.utils import conf_subplot_figsize, tight_layout
 
 from autogalaxy.profiles.basis import Basis
 from autogalaxy.plot.plot_utils import _to_positions, plot_array, _save_subplot
@@ -76,5 +76,5 @@ def subplot_image(
             ax=axes_flat[i],
         )
 
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, "basis_image", output_format)

autogalaxy/quantity/plot/fit_quantity_plots.py

@@ -1,7 +1,7 @@
 import matplotlib.pyplot as plt
 
 import autoarray as aa
-from autoarray.plot.utils import conf_subplot_figsize
+from autoarray.plot.utils import conf_subplot_figsize, tight_layout
 
 from autogalaxy.quantity.fit_quantity import FitQuantity
 from autogalaxy.plot.plot_utils import plot_array, _save_subplot
@@ -56,7 +56,7 @@ def _subplot_fit_array(fit, output_path, output_format, colormap, use_log10, pos
             ax=axes_flat[i],
         )
 
-    plt.tight_layout()
+    tight_layout()
     _save_subplot(fig, output_path, filename, output_format)