negative zs bug

JaimeRZP · Nov 18, 2024 · 2c4f3ab · 2c4f3ab
1 parent a55115a
commit 2c4f3ab
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/tracers.jl b/src/tracers.jl
@@ -28,6 +28,9 @@ Returns:
 - `NumberCountsTracer::NumberCountsTracer` : Number counts tracer structure.
 """
 NumberCountsTracer(cosmo::Cosmology, z, nz; b=1.0) = begin
+    sel = @. (z > 0.0)
+    z = z[sel]
+    nz = nz[sel]
     nz_norm = integrate(z, nz, SimpsonEven())  
     chi = cosmo.chi(z)
     hz = Hmpc(cosmo, z)
@@ -58,6 +61,9 @@ end
 
 WeakLensingTracer(cosmo::Cosmology, z, nz;
     IA_params = [0.0, 0.0], m=0.0) = begin
+    sel = @. (z > 0.0)
+    z = z[sel]
+    nz = nz[sel]
     cosmo_type = cosmo.settings.cosmo_type
     res = length(z)
     nz_norm = integrate(z, nz, SimpsonEven())

diff --git a/test/test_output.npz b/test/test_output.npz