fix DeviceError errors

Author: Shiro-Raven

tests/lightning_tensor/test_tensornet_class.py

@@ -21,7 +21,6 @@
 import pennylane as qml
 import pytest
 from conftest import LightningDevice, device_name  # tested device
-from pennylane import DeviceError
 from pennylane.wires import Wires
 
 if device_name != "lightning.tensor":
@@ -88,6 +87,7 @@ def test_errors_apply_operation_state_preparation(operation, par):
     tensornet = LightningTensorNet(wires, bondDims)
 
     with pytest.raises(
-        DeviceError, match="lightning.tensor does not support initialization with a state vector."
+        qml.DeviceError,
+        match="lightning.tensor does not support initialization with a state vector.",
     ):
         tensornet.apply_operations([operation(np.array(par), Wires(range(wires)))])

tests/test_measurements.py

@@ -393,7 +393,7 @@ def circuit():
             qml.RX(0.52, wires=0)
             return qml.expval(qml.RX(0.742, wires=[0]))
 
-        with pytest.raises(qml._device.DeviceError, match="Observable RX.*not supported"):
+        with pytest.raises(qml.DeviceError, match="Observable RX.*not supported"):
             circuit()
 
     def test_observable_return_type_is_expectation(self, dev):
@@ -494,7 +494,7 @@ def circuit():
             qml.RX(0.52, wires=0)
             return qml.var(qml.RX(0.742, wires=[0]))
 
-        with pytest.raises(qml._device.DeviceError, match="Observable RX.*not supported"):
+        with pytest.raises(qml.DeviceError, match="Observable RX.*not supported"):
             circuit()
 
     def test_observable_return_type_is_variance(self, dev):
@@ -523,7 +523,7 @@ def circuit():
             qml.RX(0.52, wires=0)
             return qml.var(qml.RX(0.742, wires=[0]))
 
-        with pytest.raises(qml._device.DeviceError, match="Observable RX.*not supported"):
+        with pytest.raises(qml.DeviceError, match="Observable RX.*not supported"):
             circuit()
 
 

tests/test_native_mcm.py

@@ -20,7 +20,6 @@
 import pytest
 from conftest import LightningDevice, device_name, validate_measurements
 from flaky import flaky
-from pennylane._device import DeviceError
 
 if device_name not in ("lightning.qubit", "lightning.kokkos"):
     pytest.skip("Native MCM not supported. Skipping.", allow_module_level=True)
@@ -86,7 +85,7 @@ def func(x, y):
 
     if device_name == "lightning.qubit":
         with pytest.raises(
-            DeviceError,
+            qml.DeviceError,
             match=f"not accepted with finite shots on lightning.qubit",
         ):
             func(*params)

tests/test_templates.py

@@ -176,7 +176,7 @@ def circuit(feature_vector):
 
         X = np.arange(1, n_qubits + 1)
 
-        with pytest.raises(qml._device.DeviceError, match="not supported"):
+        with pytest.raises(qml.DeviceError, match="not supported"):
             _ = qml.QNode(circuit, dev, diff_method=None)(X)
 
 
@@ -242,7 +242,7 @@ def circuit(weights):
         shapes = qml.CVNeuralNetLayers.shape(n_layers=2, n_wires=n_qubits)
         weights = [np.random.random(shape) for shape in shapes]
 
-        with pytest.raises(qml._device.DeviceError, match="not supported"):
+        with pytest.raises(qml.DeviceError, match="not supported"):
             _ = qml.QNode(circuit, dev, diff_method=None)(weights)