Add __repr__ method to ParameterProperties

probml · Oct 15, 2024 · 2882457 · 2882457
1 parent 1cd9a96
commit 2882457
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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.