docs: impl repr for kinds

CQCL · Jun 28, 2024 · 5b22902 · 5b22902
1 parent ab198c5
commit 5b22902
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/hugr-py/src/hugr/ops.py b/hugr-py/src/hugr/ops.py
@@ -49,7 +49,7 @@ def port_kind(self, port: InPort | OutPort) -> tys.Kind:
         Example:
             >>> op = Const(val.TRUE)
             >>> op.port_kind(OutPort(Node(0), 0))
-            ConstKind(ty=Bool)
+            ConstKind(Bool)
         """
         ...  # pragma: no cover
 

diff --git a/hugr-py/src/hugr/tys.py b/hugr-py/src/hugr/tys.py
@@ -400,22 +400,34 @@ def __repr__(self) -> str:
 class ValueKind:
     """Dataflow value edges."""
 
+    #: Type of the value.
     ty: Type
 
+    def __repr__(self) -> str:
+        return f"ValueKind({self.ty})"
+
 
 @dataclass(frozen=True)
 class ConstKind:
     """Static constant value edges."""
 
+    #: Type of the constant.
     ty: Type
 
+    def __repr__(self) -> str:
+        return f"ConstKind({self.ty})"
+
 
 @dataclass(frozen=True)
 class FunctionKind:
     """Statically defined function edges."""
 
+    #: Type of the function.
     ty: PolyFuncType
 
+    def __repr__(self) -> str:
+        return f"FunctionKind({self.ty})"
+
 
 @dataclass(frozen=True)
 class CFKind: