fixing issues with terminal measure

Author: weinbe58

src/bloqade/gemini/dialects/logical/_interface.py

@@ -1,4 +1,4 @@
-from typing import TypeVar
+from typing import Any, TypeVar
 
 from kirin import lowering
 from kirin.dialects import ilist
@@ -8,13 +8,12 @@
 from .stmts import TerminalLogicalMeasurement
 
 Len = TypeVar("Len")
-CodeN = TypeVar("CodeN")
 
 
 @lowering.wraps(TerminalLogicalMeasurement)
 def terminal_measure(
     qubits: ilist.IList[Qubit, Len],
-) -> ilist.IList[ilist.IList[MeasurementResult, CodeN], Len]:
+) -> ilist.IList[ilist.IList[MeasurementResult, Any], Len]:
     """Perform measurements on a list of logical qubits.
 
     Measurements are returned as a nested list where each member list

src/bloqade/gemini/dialects/logical/stmts.py

@@ -27,7 +27,6 @@ class Initialize(ir.Statement):
 
 
 Len = types.TypeVar("Len")
-CodeN = types.TypeVar("CodeN")
 
 
 @statement(dialect=dialect)
@@ -48,5 +47,5 @@ class TerminalLogicalMeasurement(ir.Statement):
     traits = frozenset({lowering.FromPythonCall()})
     qubits: ir.SSAValue = info.argument(ilist.IListType[QubitType, Len])
     result: ir.ResultValue = info.result(
-        ilist.IListType[ilist.IListType[MeasurementResultType, CodeN], Len]
+        ilist.IListType[ilist.IListType[MeasurementResultType, types.Any], Len]
     )