added contour to Grid2D

Jammy2211 · Jan 4, 2024 · e7d256c · e7d256c
1 parent d974108
commit e7d256c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/autoarray/plot/mat_plot/two_d.py b/autoarray/plot/mat_plot/two_d.py
@@ -448,6 +448,8 @@ def plot_grid(
             self.yticks.set(min_value=extent[2], max_value=extent[3], units=self.units)
             self.xticks.set(min_value=extent[0], max_value=extent[1], units=self.units)
 
+        self.contour.set(array=color_array, extent=extent, use_log10=self.use_log10)
+
         visuals_2d.plot_via_plotter(plotter=self, grid_indexes=grid)
 
         if not self.is_for_subplot: