if an ax is not provided, need to get fig from ax

adrn · Feb 14, 2024 · c4595f3 · c4595f3
1 parent 72480aa
commit c4595f3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/gala/potential/potential/core.py b/gala/potential/potential/core.py
@@ -757,6 +757,8 @@ def plot_rotation_curve(self, R_grid, t=0.0, ax=None, labels=None, **plot_kwargs
 
         if ax is None:
             fig, ax = plt.subplots()
+        else:
+            fig = ax.figure
 
         if labels is not False:
             ax.set_xlabel(labels[0])