diff --git a/bayesflow/data_adapters/transforms/constrain.py b/bayesflow/data_adapters/transforms/constrain.py index ea44f5e4..3643d82c 100644 --- a/bayesflow/data_adapters/transforms/constrain.py +++ b/bayesflow/data_adapters/transforms/constrain.py @@ -29,7 +29,7 @@ def __init__( raise ValueError("The lower bound must be strictly less than the upper bound.") match method: - case "default" | "sigmoid" | "logit": + case "default" | "sigmoid" | "expit" | "logit": def constrain(x): return (upper - lower) * sigmoid(x) + lower