From 7001e99074cee2873ce68e75b800615c66c792d3 Mon Sep 17 00:00:00 2001 From: Matthias Date: Thu, 19 Oct 2023 15:54:39 +0200 Subject: [PATCH] Fix itertools bug --- pyloidal/cocos.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/pyloidal/cocos.py b/pyloidal/cocos.py index 1871b0c..e95cf55 100644 --- 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))