Fix DifferentiableUniverseInitiative#99

ajouellette · May 1, 2024 · e6f1be5 · e6f1be5
1 parent 5c3a0f7
commit e6f1be5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/jax_cosmo/angular_cl.py b/jax_cosmo/angular_cl.py
@@ -185,15 +185,16 @@ def gaussian_cl_covariance_and_mean(
     return_cls: (returns signal + noise cl, covariance)
     """
     ell = np.atleast_1d(ell)
-    n_ell = len(ell)
 
     # Compute signal vectors
     cl_signal = angular_cl(
         cosmo, ell, probes, transfer_fn=transfer_fn, nonlinear_fn=nonlinear_fn
     )
     cl_noise = noise_cl(ell, probes)
 
+    return_cls = cl_signal + cl_noise
+
     # retrieve the covariance
     cov_mat = gaussian_cl_covariance(ell, probes, cl_signal, cl_noise, f_sky, sparse)
 
-    return cl_signal.flatten(), cov_mat
+    return return_cls.flatten(), cov_mat