bring in return types

Author: johnzl-777

src/bloqade/annotate/_interface.py

@@ -6,16 +6,17 @@
 from bloqade.types import MeasurementResult
 
 from .stmts import SetDetector, SetObservable
+from .types import Detector, Observable
 
 
 @wraps(SetDetector)
 def set_detector(
     measurements: ilist.IList[MeasurementResult, Any] | list[MeasurementResult],
     coordinates: tuple[float | int, ...],
-) -> None: ...
+) -> Detector: ...
 
 
 @wraps(SetObservable)
 def set_observable(
     measurements: ilist.IList[MeasurementResult, Any] | list[MeasurementResult],
-) -> None: ...
+) -> Observable: ...

src/bloqade/annotate/stmts.py

@@ -3,6 +3,7 @@
 from kirin.dialects import ilist
 
 from bloqade.types import MeasurementResultType
+from bloqade.annotate.types import DetectorType, ObservableType
 
 from ._dialect import dialect
 
@@ -20,8 +21,9 @@ class SetDetector(ConsumesMeasurementResults):
     coordinates: ir.SSAValue = info.argument(
         type=kirin_types.Tuple[kirin_types.Int | kirin_types.Float]
     )
+    result: ir.ResultValue = info.result(DetectorType)
 
 
 @statement(dialect=dialect)
 class SetObservable(ConsumesMeasurementResults):
-    pass
+    result: ir.ResultValue = info.result(ObservableType)

src/bloqade/annotate/types.py

@@ -0,0 +1,13 @@
+from kirin import types
+
+
+class Detector:
+    pass
+
+
+class Observable:
+    pass
+
+
+DetectorType = types.PyClass(Detector)
+ObservableType = types.PyClass(Observable)