fixing bug

FlamTeam · Jan 10, 2024 · c8a37d2 · c8a37d2
1 parent 7acc825
commit c8a37d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flamedisx/lxe_blocks/s2_loss.py b/flamedisx/lxe_blocks/s2_loss.py
@@ -34,7 +34,7 @@ def _compute(self, data_tensor, ptensor,
 
     def _simulate(self, d):
         d['s2_raw_after_loss'] = stats.binom.rvs(
-            n=np.clip(d['s2_raw'], 1e-15, np.float32.max).astype(dtype=np.int32),
+            n=d['s2_raw'].astype(dtype=np.int32),
             p=np.nan_to_num(self.gimme_numpy('s2_survival_p')).clip(0., 1.))
 
     def _annotate(self, d):