diff --git a/test/unit_tests/braket/circuits/test_circuit.py b/test/unit_tests/braket/circuits/test_circuit.py index ffa83957d..ca9bf77bb 100644 --- a/test/unit_tests/braket/circuits/test_circuit.py +++ b/test/unit_tests/braket/circuits/test_circuit.py @@ -1175,7 +1175,7 @@ def h(target): @circuit.subroutine() def h_nested(target): for qubit in target: - yield h(target) + yield h([qubit]) circ = Circuit().add(h_nested, [0, 1]) expected = Circuit([Instruction(gates.H(), j) for i in range(2) for j in range(2)]) @@ -2182,9 +2182,7 @@ def foo( "OPENQASM 3.0;", "bit[1] b;", "qubit[1] q;", - "#pragma braket noise " - "kraus([[0.9486833im, 0], [0, 0.9486833im]], [[0, 0.31622777], " - "[0.31622777, 0]]) q[0]", + "#pragma braket noise kraus([[0.9486833im, 0], [0, 0.9486833im]], [[0, 0.31622777], [0.31622777, 0]]) q[0]", "b[0] = measure q[0];", ]), inputs={},