fix poisson scale factors

pfackeldey · Oct 2, 2023 · cf381b0 · cf381b0
1 parent 5931ba4
commit cf381b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dilax/parameter.py b/src/dilax/parameter.py
@@ -162,7 +162,7 @@ def scale_factor(self, parameter: Parameter, sumw: jax.Array) -> jax.Array:
         gauss_cdf = jnp.broadcast_to(
             self.constraint.cdf(parameter.value), self.lamb.shape
         )
-        return Poisson(self.lamb).inv_cdf(gauss_cdf)  # type: ignore[arg-type]
+        return Poisson(self.lamb).inv_cdf(gauss_cdf) / sumw  # type: ignore[arg-type]
 
 
 class ModifierBase(eqx.Module):