From 32b6c207dcf703f6c695dc9fae3f7615bd5e8f15 Mon Sep 17 00:00:00 2001
From: Jan Petykiewicz <jan@mpxd.net>
Date: Mon, 22 Sep 2025 22:24:15 -0700
Subject: [PATCH] [read] more fixup for visualize_edges

---
 gridlock/read.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/gridlock/read.py b/gridlock/read.py
index 7b4de1e..503e996 100644
--- a/gridlock/read.py
+++ b/gridlock/read.py
@@ -147,6 +147,7 @@ class GridReadMixin(GridPosMixin):
             cell_data: NDArray,
             plane: PlaneProtocol | PlaneDict,
             which_shifts: int = 0,
+            sample_period: int = 1,
             finalize: bool = True,
             contour_args: dict[str, Any] | None = None,
             ax: 'matplotlib.axes.Axes | None' = None,
@@ -207,7 +208,7 @@ class GridReadMixin(GridPosMixin):
             fig, ax = pyplot.subplots()
         else:
             fig = ax.figure
-        xc, yc = (self.shifted_exyz(which_shifts)[a] for a in surface)
+        xc, yc = (self.shifted_xyz(which_shifts)[a] for a in surface)
         xcmesh, ycmesh = numpy.meshgrid(xc, yc, indexing='ij')
 
         mappable = ax.contour(xcmesh, ycmesh, grid_slice, levels=levels, **contour_args)