use exceptions

Author: andrijapau

pennylane_aqt/device.py

@@ -23,7 +23,7 @@
 from time import sleep
 
 import numpy as np
-from pennylane import DeviceError
+from pennylane.exceptions import DeviceError
 from pennylane.devices import QubitDevice
 from pennylane.ops import Adjoint
 

tests/test_device.py

@@ -110,7 +110,7 @@ def test_retry_delay(self):
         dev.retry_delay = 1.0
         assert dev.retry_delay == 1.0
 
-        with pytest.raises(qml.DeviceError, match="needs to be positive"):
+        with pytest.raises(qml.exceptions.DeviceError, match="needs to be positive"):
             dev.retry_delay = -5
 
     def test_set_api_configs(self):
@@ -304,7 +304,9 @@ def test_apply_basisstate_not_first_exception(self):
 
         dev = AQTDevice(3, api_key=SOME_API_KEY)
 
-        with pytest.raises(qml.DeviceError, match="only supported at the beginning of a circuit"):
+        with pytest.raises(
+            qml.exceptions.DeviceError, match="only supported at the beginning of a circuit"
+        ):
             dev.apply([qml.RX(0.5, wires=1), qml.BasisState(np.array([1, 1, 1]), wires=[0, 1, 2])])
 
     def test_apply_statevector_not_first_exception(self):
@@ -314,7 +316,9 @@ def test_apply_statevector_not_first_exception(self):
         dev = AQTDevice(2, api_key=SOME_API_KEY)
 
         state = np.ones(8) / np.sqrt(8)
-        with pytest.raises(qml.DeviceError, match="only supported at the beginning of a circuit"):
+        with pytest.raises(
+            qml.exceptions.DeviceError, match="only supported at the beginning of a circuit"
+        ):
             dev.apply([qml.RX(0.5, wires=1), qml.StatePrep(state, wires=[0, 1, 2])])
 
     def test_apply_raises_for_error(self, monkeypatch):
@@ -409,7 +413,7 @@ def test_unsupported_operation_exception(self):
 
         dev = AQTDevice(1, api_key=SOME_API_KEY)
 
-        with pytest.raises(qml.DeviceError, match="is not supported on AQT devices"):
+        with pytest.raises(qml.exceptions.DeviceError, match="is not supported on AQT devices"):
             dev._append_op_to_queue("BAD_GATE", 0.5, [0])