Merge pull request #310 from boutproject/fix-fix-gridless-loading

Fix in case 'dz' was actually 0-dimensional
boutproject · Jan 17, 2025 · 64408ea · 64408ea
2 parents 7b4d5d3 + 8b46ff9
commit 64408ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xbout/geometries.py b/xbout/geometries.py
@@ -214,7 +214,7 @@ def apply_geometry(ds, geometry_name, *, coordinates=None, grid=None):
                 dz = updated_ds["dz"]
 
             z0 = 2 * np.pi * updated_ds.metadata["ZMIN"]
-            z1 = z0 + nz * dz.data[0, 0]
+            z1 = z0 + nz * dz.data.flatten()[0]
             if not np.all(
                 np.isclose(
                     z1,