fix: Remove deprecated import path for QubitDevice (#281)

amazon-braket · Nov 14, 2024 · 278b2f6 · 278b2f6
1 parent 4105263
commit 278b2f6
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/src/braket/pennylane_plugin/ahs_device.py b/src/braket/pennylane_plugin/ahs_device.py
@@ -39,8 +39,8 @@
 
 import numpy as np
 import pennylane as qml
-from pennylane import QubitDevice
 from pennylane._version import __version__
+from pennylane.devices import QubitDevice
 from pennylane.measurements import MeasurementProcess, SampleMeasurement
 from pennylane.ops import CompositeOp, Hamiltonian
 from pennylane.pulse import ParametrizedEvolution

diff --git a/src/braket/pennylane_plugin/braket_device.py b/src/braket/pennylane_plugin/braket_device.py
@@ -44,8 +44,9 @@
 
 import numpy as onp
 import pennylane as qml
-from pennylane import QuantumFunctionError, QubitDevice
+from pennylane import QuantumFunctionError
 from pennylane import numpy as np
+from pennylane.devices import QubitDevice
 from pennylane.gradients import param_shift
 from pennylane.measurements import (
     Counts,

diff --git a/test/unit_tests/test_braket_device.py b/test/unit_tests/test_braket_device.py
@@ -53,8 +53,9 @@
     OQC_PULSE_PROPERTIES_ALL_FRAMES,
     RESULT,
 )
-from pennylane import QuantumFunctionError, QubitDevice
+from pennylane import QuantumFunctionError
 from pennylane import numpy as np
+from pennylane.devices import QubitDevice
 from pennylane.pulse import ParametrizedEvolution
 from pennylane.tape import QuantumScript, QuantumTape