Merge pull request #274 from Jammy2211/feature/feature_basis_plot

Feature/feature basis plot
Jammy2211 · May 15, 2024 · 166f4bd · 166f4bd
2 parents 3f7cee8 + b79e290
commit 166f4bd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/autolens/imaging/plot/fit_imaging_plotters.py b/autolens/imaging/plot/fit_imaging_plotters.py
@@ -588,6 +588,9 @@ def subplot_tracer(self):
 
     def subplot_mappings_of_plane(self, plane_index: Optional[int] = None, auto_filename: str = "subplot_mappings"):
 
+        if not self.fit.inversion.has(cls=aa.Pixelization):
+            return
+
         plane_indexes = self.plane_indexes_from(plane_index=plane_index)
 
         for plane_index in plane_indexes:

diff --git a/autolens/plot/__init__.py b/autolens/plot/__init__.py
@@ -71,6 +71,7 @@
 from autogalaxy.plot.visuals.one_d import Visuals1D
 from autogalaxy.plot.visuals.two_d import Visuals2D
 
+from autogalaxy.profiles.plot.basis_plotters import BasisPlotter
 from autogalaxy.profiles.plot.light_profile_plotters import LightProfilePlotter
 from autogalaxy.profiles.plot.light_profile_plotters import LightProfilePDFPlotter
 from autogalaxy.profiles.plot.mass_profile_plotters import MassProfilePlotter