Skip to content

Commit 201f506

Browse files
committed
Fix PyQrack target
1 parent 8a17bed commit 201f506

File tree

3 files changed

+6
-7
lines changed

3 files changed

+6
-7
lines changed

src/bloqade/pyqrack/target.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,8 @@ def run(
8787
"""
8888
fold = Fold(mt.dialects)
8989
fold(mt)
90-
return self._get_interp(mt).run(mt, args, kwargs)
90+
_, ret = self._get_interp(mt).run(mt, *args, **kwargs)
91+
return ret
9192

9293
def multi_run(
9394
self,

test/pyqrack/runtime/noise/qasm2/test_loss.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,10 @@
1-
from typing import Literal
21
from unittest.mock import Mock
32

43
from kirin import ir
5-
from kirin.dialects import ilist
64

75
from bloqade import qasm2
86
from bloqade.qasm2 import noise
9-
from bloqade.pyqrack import PyQrackQubit, PyQrackInterpreter, reg
7+
from bloqade.pyqrack import PyQrackInterpreter, reg
108
from bloqade.pyqrack.base import MockMemory
119

1210

@@ -34,9 +32,9 @@ def test_atom_loss(c: qasm2.CReg):
3432
input = reg.CRegister(1)
3533
memory = MockMemory()
3634

37-
result: ilist.IList[PyQrackQubit, Literal[2]] = PyQrackInterpreter(
35+
_, result = PyQrackInterpreter(
3836
qasm2.extended, memory=memory, rng_state=rng_state
39-
).run(test_atom_loss, (input,))
37+
).run(test_atom_loss, input)
4038

4139
assert result[0].state is reg.QubitState.Lost
4240
assert result[1].state is reg.QubitState.Active

test/pyqrack/runtime/noise/qasm2/test_pauli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
def run_mock(program: ir.Method, rng_state: Mock | None = None):
1212
PyQrackInterpreter(
1313
program.dialects, memory=(memory := MockMemory()), rng_state=rng_state
14-
).run(program, ())
14+
).run(program)
1515
assert isinstance(mock := memory.sim_reg, Mock)
1616
return mock
1717

0 commit comments

Comments
 (0)