fix plot_3d trying to connect end surfaces for zeta not 2pi cases

PlasmaControl · Oct 20, 2023 · 4d200ce · 4d200ce
1 parent b0794c9
commit 4d200ce
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/desc/plotting.py b/desc/plotting.py
@@ -854,7 +854,10 @@ def plot_3d(
 
     if grid.num_rho == 1:
         n1, n2 = grid.num_theta, grid.num_zeta
-        p1, p2 = True, True
+        if not grid.nodes[-1][2] == 2 * np.pi:
+            p1, p2 = True, False
+        else:
+            p1, p2 = True, True
     elif grid.num_theta == 1:
         n1, n2 = grid.num_rho, grid.num_zeta
         p1, p2 = False, True