Fix issue 322

gwmod · Feb 8, 2024 · 4bc5b85 · 4bc5b85
1 parent 9cc59d3
commit 4bc5b85
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/nlmod/plot/plot.py b/nlmod/plot/plot.py
@@ -40,7 +40,11 @@ def modelgrid(ds, ax=None, **kwargs):
         _, ax = plt.subplots(figsize=(10, 10))
         ax.axis("scaled")
     modelgrid = modelgrid_from_ds(ds)
+    extent = None if ax.get_autoscale_on() else ax.axis()
     modelgrid.plot(ax=ax, **kwargs)
+    if extent is not None:
+        ax.axis(extent)
+
     return ax