diff --git a/pennylane_cirq/qsim_device.py b/pennylane_cirq/qsim_device.py index dd7b009..a481716 100644 --- 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 index 87a7ae3..f9ebb7c 100644 --- 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 index 6dc6d64..19a03cf 100644 --- 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")