Add __repr__ method to ParameterProperties

Author: gileshd

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.