diff --git a/src/bloqade/pyqrack/noise/native.py b/src/bloqade/pyqrack/noise/native.py index 940038ca..e5002c33 100644 --- a/src/bloqade/pyqrack/noise/native.py +++ b/src/bloqade/pyqrack/noise/native.py @@ -93,8 +93,7 @@ def atom_loss_channel( for qarg in active_qubits: if interp.rng_state.uniform() <= stmt.prob: - sim_reg = qarg.ref.sim_reg - sim_reg.force_m(qarg.addr, 0) + qarg.ref.sim_reg.m(qarg.addr) qarg.drop() return () diff --git a/test/pyqrack/runtime/noise/native/test_pauli.py b/test/pyqrack/runtime/noise/native/test_pauli.py index 488ed21a..4e5fe833 100644 --- a/test/pyqrack/runtime/noise/native/test_pauli.py +++ b/test/pyqrack/runtime/noise/native/test_pauli.py @@ -79,7 +79,7 @@ def test_atom_loss(): rng_state.choice.side_effect = ["y"] rng_state.uniform.return_value = 0.5 sim_reg = run_mock(test_atom_loss, rng_state) - sim_reg.assert_has_calls([call.mcz([0], 1), call.force_m(0, 0), call.y(1)]) + sim_reg.assert_has_calls([call.mcz([0], 1), call.m(0), call.y(1)]) def test_cz_pauli_channel_true(): diff --git a/test/pyqrack/runtime/test_dyn_memory.py b/test/pyqrack/runtime/test_dyn_memory.py index 92967432..03e08a7a 100644 --- a/test/pyqrack/runtime/test_dyn_memory.py +++ b/test/pyqrack/runtime/test_dyn_memory.py @@ -25,7 +25,7 @@ def ghz(n: int): dynamic_qubits=True, ) - N = 50 + N = 20 result = target.multi_run(ghz, 100, N) result = Counter("".join(str(int(bit)) for bit in bits) for bits in result)