diff --git a/dynamax/parameters.py b/dynamax/parameters.py index c829c586..7e7a98f3 100644 --- a/dynamax/parameters.py +++ b/dynamax/parameters.py @@ -46,6 +46,9 @@ def tree_flatten(self): def tree_unflatten(cls, aux_data, children): return cls(*aux_data) + def __repr__(self): + return f"ParameterProperties(trainable={self.trainable}, constrainer={self.constrainer})" + def to_unconstrained(params: ParameterSet, props: PropertySet) -> ParameterSet: """Convert the constrained parameters to unconstrained form.