fix failing test

PennyLaneAI · Aug 26, 2024 · ae42aca · ae42aca
1 parent addad41
commit ae42aca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_gradients.py b/tests/test_gradients.py
@@ -61,7 +61,7 @@ def test_gradient_with_custom_operator(qvm, compiler):
 
     def func(x, y):
         """Reference QNode"""
-        qml.BasisState(np.array([1, 1], requires_grad=False), wires=0)
+        qml.BasisState(np.array([1, 1], requires_grad=False), wires=[0, 1])
         qml.RY(x, wires=0)
         qml.RX(y, wires=1)
         qml.PSWAP(np.array(0.432, requires_grad=False), wires=[0, 1])