update tests

PennyLaneAI · Oct 1, 2024 · bc85e56 · bc85e56
1 parent 6bfd420
commit bc85e56
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/tests/lightning_qubit/test_adjoint_jacobian_class.py b/tests/lightning_qubit/test_adjoint_jacobian_class.py
@@ -420,10 +420,7 @@ def test_multiple_measurements(self, tol, lightning_sv):
         statevector = lightning_sv(num_wires=2)
         result_vjp = self.calculate_vjp(statevector, tape1, dy)
 
-        if device_name == "lightning.gpu":
-            statevector.reset_state(True)
-        else:
-            statevector.reset_state()
+        statevector.reset_state()
 
         result_jac = self.calculate_jacobian(statevector, tape2)
 
@@ -483,10 +480,7 @@ def test_hermitian_expectation(self, tol, lightning_sv):
                 qml.expval(qml.Hermitian(obs, wires=(0,)))
             tape.trainable_params = {0}
 
-            if device_name == "lightning.gpu":
-                statevector.reset_state(True)
-            else:
-                statevector.reset_state()
+            statevector.reset_state()
 
             vjp = self.calculate_vjp(statevector, tape, dy)