Allow lists

itpplasma · Nov 19, 2023 · 684dd73 · 684dd73
1 parent ec9f975
commit 684dd73
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/libneo/mgrid.py b/python/libneo/mgrid.py
@@ -116,7 +116,7 @@ def write(self, filename):
             f.createDimension("stringsize", 30)
             f.createDimension("external_coil_groups", len(self.coil_group))
             f.createDimension("dim_00001", 1)
-            f.createDimension("external_coils", self.coil_current.size)
+            f.createDimension("external_coils", len(self.coil_current))
             f.createDimension("rad", self.ir)
             f.createDimension("zee", self.jz)
             f.createDimension("phi", self.kp)
@@ -156,7 +156,7 @@ def write(self, filename):
             f.variables["jz"][:] = self.jz
             f.variables["kp"][:] = self.kp
             f.variables["nfp"][:] = self.nfp
-            f.variables["nextcur"][:] = self.coil_current.size
+            f.variables["nextcur"][:] = len(self.coil_current)
             f.variables["rmin"][:] = self.rmin
             f.variables["zmin"][:] = self.zmin
             f.variables["rmax"][:] = self.rmax