From bc85e5608b7e091087f274da450ee28377e2aeba Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Luis=20Alfredo=20Nu=C3=B1ez=20Meneses?= Date: Tue, 1 Oct 2024 15:55:31 +0000 Subject: [PATCH] update tests --- tests/lightning_qubit/test_adjoint_jacobian_class.py | 10 ++-------- 1 file changed, 2 insertions(+), 8 deletions(-) diff --git a/tests/lightning_qubit/test_adjoint_jacobian_class.py b/tests/lightning_qubit/test_adjoint_jacobian_class.py index d5b9355bf..adfccffd2 100644 --- 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)