diff --git a/model/src/pyrenew/transformation/builtin.py b/model/src/pyrenew/transformation/builtin.py index 166643d3..81b3f3c9 100644 --- a/model/src/pyrenew/transformation/builtin.py +++ b/model/src/pyrenew/transformation/builtin.py @@ -30,7 +30,7 @@ def ScaledLogitTransform( return nt.ComposeTransform( [ nt.AffineTransform( - 0.0, 1.0 / x_max, domain=constraints.interval(0, x_max) + 0.0, 1.0 / x_max, domain=constraints.interval(0.0, 1.0 * x_max) ), nt.SigmoidTransform().inv, ]