visual clean up complete

Author: Jammy2211

autoarray/dataset/grids.py

@@ -3,11 +3,9 @@
 from autoarray.mask.mask_2d import Mask2D
 from autoarray.structures.arrays.uniform_2d import Array2D
 from autoarray.structures.arrays.kernel_2d import Kernel2D
-from autoarray.structures.grids.uniform_1d import Grid1D
 from autoarray.structures.grids.uniform_2d import Grid2D
 
 from autoarray.inversion.pixelization.border_relocator import BorderRelocator
-from autoconf import cached_property
 
 from autoarray import exc
 

autoarray/dataset/plot/imaging_plotters.py

@@ -4,15 +4,14 @@
 from autoarray.plot.visuals.two_d import Visuals2D
 from autoarray.plot.mat_plot.two_d import MatPlot2D
 from autoarray.plot.auto_labels import AutoLabels
-from autoarray.plot.abstract_plotters import Plotter
+from autoarray.plot.abstract_plotters import AbstractPlotter
 from autoarray.dataset.imaging.dataset import Imaging
 
 
-class ImagingPlotterMeta(Plotter):
+class ImagingPlotterMeta(AbstractPlotter):
     def __init__(
         self,
         dataset: Imaging,
-        get_visuals_2d: Callable,
         mat_plot_2d: MatPlot2D = None,
         visuals_2d: Visuals2D = None,
     ):
@@ -31,8 +30,6 @@ def __init__(
         ----------
         dataset
             The imaging dataset the plotter plots.
-        get_visuals_2d
-            A function which extracts from the `Imaging` the 2D visuals which are plotted on figures.
         mat_plot_2d
             Contains objects which wrap the matplotlib function calls that make 2D plots.
         visuals_2d
@@ -42,7 +39,6 @@ def __init__(
         super().__init__(mat_plot_2d=mat_plot_2d, visuals_2d=visuals_2d)
 
         self.dataset = dataset
-        self.get_visuals_2d = get_visuals_2d
 
     @property
     def imaging(self):
@@ -84,21 +80,21 @@ def figures_2d(
         if data:
             self.mat_plot_2d.plot_array(
                 array=self.dataset.data,
-                visuals_2d=self.get_visuals_2d(),
+                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(title=title_str or f" Data", filename="data"),
             )
 
         if noise_map:
             self.mat_plot_2d.plot_array(
                 array=self.dataset.noise_map,
-                visuals_2d=self.get_visuals_2d(),
+                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(title_str or f"Noise-Map", filename="noise_map"),
             )
 
         if psf:
             self.mat_plot_2d.plot_array(
                 array=self.dataset.psf,
-                visuals_2d=self.get_visuals_2d(),
+                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(
                     title=title_str or f"Point Spread Function",
                     filename="psf",
@@ -109,7 +105,7 @@ def figures_2d(
         if signal_to_noise_map:
             self.mat_plot_2d.plot_array(
                 array=self.dataset.signal_to_noise_map,
-                visuals_2d=self.get_visuals_2d(),
+                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(
                     title=title_str or f"Signal-To-Noise Map",
                     filename="signal_to_noise_map",
@@ -120,7 +116,7 @@ def figures_2d(
         if over_sample_size_lp:
             self.mat_plot_2d.plot_array(
                 array=self.dataset.grids.over_sample_size_lp,
-                visuals_2d=self.get_visuals_2d(),
+                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(
                     title=title_str or f"Over Sample Size (Light Profiles)",
                     filename="over_sample_size_lp",
@@ -131,7 +127,7 @@ def figures_2d(
         if over_sample_size_pixelization:
             self.mat_plot_2d.plot_array(
                 array=self.dataset.grids.over_sample_size_pixelization,
-                visuals_2d=self.get_visuals_2d(),
+                visuals_2d=self.visuals_2d,
                 auto_labels=AutoLabels(
                     title=title_str or f"Over Sample Size (Pixelization)",
                     filename="over_sample_size_pixelization",
@@ -220,7 +216,7 @@ def subplot_dataset(self):
         self.mat_plot_2d.use_log10 = use_log10_original
 
 
-class ImagingPlotter(Plotter):
+class ImagingPlotter(AbstractPlotter):
     def __init__(
         self,
         dataset: Imaging,
@@ -254,14 +250,10 @@ def __init__(
 
         self._imaging_meta_plotter = ImagingPlotterMeta(
             dataset=self.dataset,
-            get_visuals_2d=self.get_visuals_2d,
             mat_plot_2d=self.mat_plot_2d,
             visuals_2d=self.visuals_2d,
         )
 
         self.figures_2d = self._imaging_meta_plotter.figures_2d
         self.subplot = self._imaging_meta_plotter.subplot
         self.subplot_dataset = self._imaging_meta_plotter.subplot_dataset
-
-    def get_visuals_2d(self):
-        return self.get_2d.via_mask_from(mask=self.dataset.mask)