comprehend theta coord list, add a comma to ivb

svalinn · Oct 8, 2024 · ac01cb4 · ac01cb4
1 parent d36017b
commit ac01cb4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/parastell/invessel_build.py b/parastell/invessel_build.py
@@ -827,7 +827,8 @@ def generate_invessel_build():
         invessel_build_dict["poloidal_angles"],
         invessel_build_dict["wall_s"],
         invessel_build_dict["radial_build"],
-        logger=logger**invessel_build_dict,
+        logger=logger,
+        **invessel_build_dict,
     )
 
     invessel_build = InVesselBuild(

diff --git a/parastell/nwl_utils.py b/parastell/nwl_utils.py
@@ -187,7 +187,9 @@ def flux_coords(plas_eq, wall_s, coords, num_threads):
             for theta_coord in theta_coord_chunk
         ]
     # Ensures theta_coords are all positive (add 360 degrees where needed)
-    theta_coords = (theta_coords + 2 * np.pi) % 2 * np.pi
+    theta_coords = [
+        (theta_coord + 2 * np.pi) % 2 * np.pi for theta_coord in theta_coords
+    ]
 
     return phi_coords.tolist(), theta_coords