feat(py): Allow setting the width of IntVal constants

CQCL · Aug 12, 2024 · 9940a0a · 9940a0a
1 parent 07ccb17
commit 9940a0a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/hugr-py/src/hugr/std/int.py b/hugr-py/src/hugr/std/int.py
@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, ClassVar
 
 from typing_extensions import Self
@@ -73,9 +73,10 @@ class IntVal(val.ExtensionValue):
     """Custom value for an integer."""
 
     v: int
+    width: int = field(default=5)
 
     def to_value(self) -> val.Extension:
-        return val.Extension("int", INT_T, self.v)
+        return val.Extension("int", int_t(self.width), self.v)
 
 
 OPS_EXTENSION: tys.ExtensionId = "arithmetic.int"