make constructor for Some simpler

CQCL · Dec 18, 2024 · 1e9eee2 · 1e9eee2
1 parent 59dbaf6
commit 1e9eee2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/hugr-py/src/hugr/ops.py b/hugr-py/src/hugr/ops.py
@@ -558,10 +558,16 @@ def __repr__(self) -> str:
 
 @dataclass
 class Some(Tag):
-    """Tag operation for the `Some` variant of an Option type."""
+    """Tag operation for the `Some` variant of an Option type.
 
-    def __init__(self, option_type: tys.Option) -> None:
-        super().__init__(1, option_type)
+    Example:
+        # construct a Some variant holding a row of Bool and Unit types
+        >>> Some(tys.Bool, tys.Unit)
+        Some
+    """
+
+    def __init__(self, *some_tys: tys.Type) -> None:
+        super().__init__(1, tys.Option(*some_tys))
 
     def __repr__(self) -> str:
         return "Some"

diff --git a/hugr-py/tests/test_hugr_build.py b/hugr-py/tests/test_hugr_build.py
@@ -336,7 +336,7 @@ def test_option() -> None:
     dfg = Dfg(tys.Bool)
     b = dfg.inputs()[0]
 
-    dfg.add_op(ops.Some(tys.Option(tys.Bool)), b)
+    dfg.add_op(ops.Some(tys.Bool), b)
 
     dfg.set_outputs(b)