Skip to content

Commit e9a3e5a

Browse files
authored
Apply_mixing_matrix adds wrong ell (#383)
1 parent 59c2b0d commit e9a3e5a

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

heracles/twopoint.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -465,7 +465,7 @@ def invert_mixing_matrix(
465465
return inv_M
466466

467467

468-
def apply_mixing_matrix(d, M, lmax=None):
468+
def apply_mixing_matrix(d, M):
469469
"""
470470
Apply mixing matrix to the data Cl.
471471
Args:
@@ -476,10 +476,7 @@ def apply_mixing_matrix(d, M, lmax=None):
476476
"""
477477
corr_d = {}
478478
for key in d.keys():
479-
if lmax is None:
480-
*_, lmax = d[key].shape
481479
dtype = d[key].array.dtype
482-
ell_mask = M[key].ell
483480
s1, s2 = d[key].spin
484481
_d = np.atleast_2d(d[key].array)
485482
_M = M[key].array
@@ -495,7 +492,5 @@ def apply_mixing_matrix(d, M, lmax=None):
495492
_corr_d.append(_M @ cl)
496493
_corr_d = np.squeeze(_corr_d)
497494
_corr_d = np.array(list(_corr_d), dtype=dtype)
498-
corr_d[key] = replace(d[key], array=_corr_d, ell=ell_mask)
499-
# truncate
500-
corr_d = binned(corr_d, np.arange(0, lmax + 1))
495+
corr_d[key] = replace(d[key], array=_corr_d)
501496
return corr_d

0 commit comments

Comments
 (0)