Merge pull request #59 from P-1884/Truncated-redshift-distribution-fix

Bug fix: Truncated Redshift distribution
swagnercarena · Jul 9, 2024 · 3cebbd4 · 3cebbd4
2 parents 3a51d30 + 66dc8c6
commit 3cebbd4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/paltas/Sampling/distributions.py b/paltas/Sampling/distributions.py
@@ -356,9 +356,10 @@ def __call__(self):
 		clip = (z_lens - self.z_source_mean) / self.z_source_std
 		# number of std. devs away to stop (negative)
 		if(clip > self.z_source_min):
-			self.z_source_min = clip
+			z_source_min = clip
+		else: z_source_min = self.z_source_min
 		# define truncnorm dist for source redshift
-		z_source = truncnorm(self.z_source_min,np.inf,self.z_source_mean,
+		z_source = truncnorm(z_source_min,np.inf,self.z_source_mean,
 			self.z_source_std).rvs()
 
 		return z_lens,z_source