Fix DifferentiableUniverseInitiative#99

andyyPark · Jul 8, 2022 · e7a1290 · e7a1290
1 parent 52ca009
commit e7a1290
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/jax_cosmo/angular_cl.py b/jax_cosmo/angular_cl.py
@@ -183,15 +183,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