Add private aliases in array.py; allow save_figure to accept multiple formats

- autoarray/plot/array.py: add module-level aliases _zoom_array_2d and
  _mask_edge_coords pointing at the imported zoom_array / auto_mask_edge
  helpers for use by downstream packages (e.g. autogalaxy)
- autoarray/plot/utils.py: save_figure now accepts format as either a
  str or a list/tuple of strings; iterates over all formats so a single
  call can write png + pdf (or any combination) simultaneously

https://claude.ai/code/session_01B9sVEV54XWCa2LJw1C8gvv

Author: claude

autoarray/plot/array.py

@@ -20,6 +20,9 @@
     numpy_positions,
 )
 
+_zoom_array_2d = zoom_array
+_mask_edge_coords = auto_mask_edge
+
 
 def plot_array(
     array,

autoarray/plot/utils.py

@@ -345,7 +345,9 @@ def save_figure(
     filename
         File name without extension.
     format
-        File format passed to ``fig.savefig`` (e.g. ``"png"``, ``"pdf"``).
+        File format(s) passed to ``fig.savefig``.  Either a single string
+        (e.g. ``"png"``) or a list/tuple of strings (e.g. ``["png", "pdf"]``)
+        to save in multiple formats in one call.
     dpi
         Resolution in dots per inch.
     structure
@@ -355,29 +357,31 @@ def save_figure(
     """
     if path:
         os.makedirs(path, exist_ok=True)
-        if format == "fits":
-            if structure is not None and hasattr(structure, "output_to_fits"):
-                structure.output_to_fits(
-                    file_path=os.path.join(path, f"{filename}.fits"),
-                    overwrite=True,
-                )
+        formats = format if isinstance(format, (list, tuple)) else [format]
+        for fmt in formats:
+            if fmt == "fits":
+                if structure is not None and hasattr(structure, "output_to_fits"):
+                    structure.output_to_fits(
+                        file_path=os.path.join(path, f"{filename}.fits"),
+                        overwrite=True,
+                    )
+                else:
+                    logger.warning(
+                        f"save_figure: fits format requested for {filename} but no "
+                        "compatible structure was provided; skipping."
+                    )
             else:
-                logger.warning(
-                    f"save_figure: fits format requested for {filename} but no "
-                    "compatible structure was provided; skipping."
-                )
-        else:
-            try:
-                fig.savefig(
-                    os.path.join(path, f"{filename}.{format}"),
-                    dpi=dpi,
-                    bbox_inches="tight",
-                    pad_inches=0.1,
-                )
-            except Exception as exc:
-                logger.warning(
-                    f"save_figure: could not save {filename}.{format}: {exc}"
-                )
+                try:
+                    fig.savefig(
+                        os.path.join(path, f"{filename}.{fmt}"),
+                        dpi=dpi,
+                        bbox_inches="tight",
+                        pad_inches=0.1,
+                    )
+                except Exception as exc:
+                    logger.warning(
+                        f"save_figure: could not save {filename}.{fmt}: {exc}"
+                    )
     else:
         plt.show()