more stuff moved over

Author: Jammy2211

autoarray/plot/wrap/two_d/array_overlay.py

@@ -17,7 +17,8 @@ class ArrayOverlay(AbstractMatWrap2D):
 
     def overlay_array(self, array, figure):
         aspect = figure.aspect_from(shape_native=array.shape_native)
-        extent = array.extent_of_zoomed_array(buffer=0)
+        array_zoom = array.zoomed_around_mask(buffer=0)
+        extent = array_zoom.geometry.extent
 
         plt.imshow(
             X=array.native._array, aspect=aspect, extent=extent, **self.config_dict

autoarray/structures/arrays/uniform_2d.py

@@ -7,6 +7,7 @@
 from autoconf.fitsable import ndarray_via_fits_from, header_obj_from
 
 from autoarray.mask.mask_2d import Mask2D
+from autoarray.mask.derive.zoom_2d import Zoom2D
 from autoarray.structures.abstract_structure import Structure
 from autoarray.structures.header import Header
 from autoarray.structures.arrays.uniform_1d import Array1D