Skip to content

Commit e6643e4

Browse files
committed
Explicitly cast to specific addresses in address impls
1 parent 509055c commit e6643e4

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/bloqade/analysis/address/impls.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ def unwrap(
190190
stmt: squin.wire.Unwrap,
191191
):
192192

193-
origin_qubit = frame.get(stmt.qubit)
193+
origin_qubit = frame.get_casted(stmt.qubit, AddressQubit)
194194

195195
return (AddressWire(origin_qubit=origin_qubit),)
196196

@@ -203,7 +203,10 @@ def apply(
203203
):
204204

205205
origin_qubits = tuple(
206-
[frame.get(input_elem).origin_qubit for input_elem in stmt.inputs]
206+
[
207+
frame.get_casted(input_elem, AddressWire).origin_qubit
208+
for input_elem in stmt.inputs
209+
]
207210
)
208211
new_address_wires = tuple(
209212
[AddressWire(origin_qubit=origin_qubit) for origin_qubit in origin_qubits]

0 commit comments

Comments
 (0)