Update gliderpy/plotting.py

Co-authored-by: Filipe <[email protected]>
ioos · Jun 18, 2024 · ebe0451 · ebe0451
1 parent c8d673a
commit ebe0451
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/gliderpy/plotting.py b/gliderpy/plotting.py
@@ -57,10 +57,14 @@ def plot_transect(
     """
     cmap = kw.get("cmap", None)
 
-    if ax is None:
+    fignums = plt.get_fignums()
+    if ax is None and not fignums:
         fig, ax = plt.subplots(figsize=(17, 2))
+    elif ax:
+        fig = ax.get_figure()
     else:
-        fig = ax.figure
+        ax = plt.gca()
+        fig = plt.gcf()
 
     cs = ax.scatter(
         df.index,