Replace qml.QubitDevice with qml.devices.QubitDevice (#195)

* replace qml.QubitDevice with qml.devices.QubitDevice * Update order of imports to make CodeFactor happy
PennyLaneAI · Sep 17, 2024 · 4668f36 · 4668f36
1 parent 725ef8f
commit 4668f36
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/pennylane_cirq/qsim_device.py b/pennylane_cirq/qsim_device.py
@@ -15,8 +15,8 @@
 """
 This module provides the ``QSimDevice`` and ``QSimhDevice`` from Cirq.
 """
-import cirq
 import numpy as np
+import cirq
 import pennylane as qml
 
 try:
@@ -151,7 +151,7 @@ def capabilities(self):  # pylint: disable=missing-function-docstring
         return capabilities
 
     def expval(self, observable, shot_range=None, bin_size=None):
-        return qml.QubitDevice.expval(self, observable, shot_range, bin_size)
+        return qml.devices.QubitDevice.expval(self, observable, shot_range, bin_size)
 
     def apply(self, operations, **kwargs):
         # pylint: disable=missing-function-docstring

diff --git a/pennylane_cirq/simulator_device.py b/pennylane_cirq/simulator_device.py
@@ -31,8 +31,8 @@
 
 ----
 """
-import cirq
 import numpy as np
+import cirq
 import pennylane as qml
 
 from .cirq_device import CirqDevice
@@ -266,7 +266,7 @@ def expval(self, observable, shot_range=None, bin_size=None):
         try:
             return super().expval(observable, shot_range, bin_size)
         except ValueError:
-            return qml.QubitDevice.expval(self, observable, shot_range, bin_size)
+            return qml.devices.QubitDevice.expval(self, observable, shot_range, bin_size)
 
     def _apply_basis_state(self, basis_state_operation):
         super()._apply_basis_state(basis_state_operation)

diff --git a/tests/test_mixed_simulator_device.py b/tests/test_mixed_simulator_device.py
@@ -701,7 +701,7 @@ def circuit():
                     qml.apply(gate)
             return qml.expval(qml.PauliZ(0))
 
-        spy = mocker.spy(qml.QubitDevice, "expval")
+        spy = mocker.spy(qml.devices.QubitDevice, "expval")
 
         if use_super:
             mock_simulate = mocker.patch("cirq.DensityMatrixSimulator.simulate_expectation_values")