Merge pull request #3 from mwiesenberger/itertools

Fix itertools bug
PlasmaFAIR · Nov 6, 2023 · af40866 · af40866
2 parents 2f87ce2 + 7001e99
commit af40866
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pyloidal/cocos.py b/pyloidal/cocos.py
@@ -148,10 +148,8 @@ def identify_cocos(
     if minor_radii is None:
         # Return both variants if not provided with minor radii
         return tuple(
-            itertools.chain.from_iterable(
-                sigma_to_cocos(sigma_bp, sigma_rpz, sigma_rtp, phi_by_2pi=x)
+                sigma_to_cocos(sigma_bp, sigma_rpz, sigma_rtp, psi_by_2pi=x)
                 for x in (True, False)
-            )
         )
 
     index = np.argmin(np.abs(safety_factor))