From 72ebdd531e72dda1743e9808f2217aa73215a209 Mon Sep 17 00:00:00 2001 From: Francois Lanusse Date: Sun, 5 Feb 2023 14:45:22 +0100 Subject: [PATCH] updating black version --- jax_cosmo/angular_cl.py | 6 ++++-- jax_cosmo/transfer.py | 1 - 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/jax_cosmo/angular_cl.py b/jax_cosmo/angular_cl.py index b842076..9e36b04 100644 --- a/jax_cosmo/angular_cl.py +++ b/jax_cosmo/angular_cl.py @@ -41,8 +41,8 @@ def find_index(a, b): return cl_index.index((b, a)) cov_blocks = [] - for (i, j) in cl_index: - for (m, n) in cl_index: + for i, j in cl_index: + for m, n in cl_index: cov_blocks.append( (find_index(i, m), find_index(j, n), find_index(i, n), find_index(j, m)) ) @@ -83,6 +83,7 @@ def integrand(a): # Define an ordering for the blocks of the signal vector cl_index = np.array(_get_cl_ordering(probes)) + # Compute all combinations of tracers def combine_kernels(inds): return kernels[inds[0]] * kernels[inds[1]] @@ -109,6 +110,7 @@ def noise_cl(ell, probes): noise = np.concatenate([p.noise() for p in probes]) # Define an ordering for the blocks of the signal vector cl_index = np.array(_get_cl_ordering(probes)) + # Only include a noise contribution for the auto-spectra def get_noise_cl(inds): i, j = inds diff --git a/jax_cosmo/transfer.py b/jax_cosmo/transfer.py index ce66c80..6aec63c 100644 --- a/jax_cosmo/transfer.py +++ b/jax_cosmo/transfer.py @@ -97,7 +97,6 @@ def Eisenstein_Hu(cosmo, k, type="eisenhu_osc"): ) if type == "eisenhu": - q = k * np.power(const.tcmb / 2.7, 2) / gamma_eff # EH98 (29) #