revert changes in test_expval

PennyLaneAI · Jun 8, 2024 · 1c6acdc · 1c6acdc
1 parent 68dda84
commit 1c6acdc
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_expval.py b/tests/test_expval.py
@@ -192,8 +192,10 @@ def circuit():
 
             circ = qml.QNode(circuit, dev)
             circ_def = qml.QNode(circuit, dev_def)
-            rel_tol = 0 if device_name != "lightning.tensor" else 0.25
-            assert np.allclose(circ(), circ_def(), tol, rtol=rel_tol)
+            if device_name != "lightning.tensor":
+                assert np.allclose(circ(), circ_def(), tol)
+            else:
+                assert np.allclose(circ(), circ_def(), rtol=0.25)
 
 
 @pytest.mark.parametrize(