From be3c601618a80398d87ea5987a64957a2f12cd27 Mon Sep 17 00:00:00 2001 From: jaimerzp Date: Mon, 9 Mar 2026 15:54:45 +0000 Subject: [PATCH 1/2] working --- heracles/twopoint.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/heracles/twopoint.py b/heracles/twopoint.py index 5f7bd97..401b182 100644 --- a/heracles/twopoint.py +++ b/heracles/twopoint.py @@ -465,7 +465,7 @@ def invert_mixing_matrix( return inv_M -def apply_mixing_matrix(d, M, lmax=None): +def apply_mixing_matrix(d, M): """ Apply mixing matrix to the data Cl. Args: @@ -479,7 +479,6 @@ def apply_mixing_matrix(d, M, lmax=None): if lmax is None: *_, lmax = d[key].shape dtype = d[key].array.dtype - ell_mask = M[key].ell s1, s2 = d[key].spin _d = np.atleast_2d(d[key].array) _M = M[key].array @@ -495,7 +494,5 @@ def apply_mixing_matrix(d, M, lmax=None): _corr_d.append(_M @ cl) _corr_d = np.squeeze(_corr_d) _corr_d = np.array(list(_corr_d), dtype=dtype) - corr_d[key] = replace(d[key], array=_corr_d, ell=ell_mask) - # truncate - corr_d = binned(corr_d, np.arange(0, lmax + 1)) + corr_d[key] = replace(d[key], array=_corr_d) return corr_d From a845afadee6c98da9d99307b284e1030d344f235 Mon Sep 17 00:00:00 2001 From: jaimerzp Date: Mon, 9 Mar 2026 15:56:18 +0000 Subject: [PATCH 2/2] ruff --- heracles/twopoint.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/heracles/twopoint.py b/heracles/twopoint.py index 401b182..637bad5 100644 --- a/heracles/twopoint.py +++ b/heracles/twopoint.py @@ -476,8 +476,6 @@ def apply_mixing_matrix(d, M): """ corr_d = {} for key in d.keys(): - if lmax is None: - *_, lmax = d[key].shape dtype = d[key].array.dtype s1, s2 = d[key].spin _d = np.atleast_2d(d[key].array)