update adj unit tests

PennyLaneAI · Oct 27, 2023 · f590e47 · f590e47
1 parent 2f50387
commit f590e47
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/mpitests/test_adjoint_jacobian.py b/mpitests/test_adjoint_jacobian.py
@@ -1222,13 +1222,10 @@ def circuit_compare(params, excitations):
                 qml.SingleExcitation(params[i], wires=excitation)
         return qml.expval(H)
 
-    jac_func = qml.jacobian(circuit)
-    jac_func_comp = qml.jacobian(circuit_compare)
-
     params = qml.numpy.array([0.0] * len(doubles), requires_grad=True)
 
-    jacs = jac_func(params, excitations=doubles)
-    jacs_comp = jac_func_comp(params, excitations=doubles)
+    jacs = qml.jacobian(circuit)(params, excitations=doubles)
+    jacs_comp = qml.jacobian(circuit_compare)(params, excitations=doubles)
 
     comm.Barrier()