minor import fix

AgnostiqHQ · Jan 24, 2024 · 4cd7626 · 4cd7626
1 parent 25f1c93
commit 4cd7626
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/covalent/_shared_files/qresult_utils.py b/covalent/_shared_files/qresult_utils.py
@@ -23,7 +23,7 @@
 from pennylane.tape import QuantumTape
 
 from .._workflow.qdevice import QEDevice
-from .utils import get_original_shots
+from .qelectron_utils import get_original_shots
 
 
 def re_execute(

diff --git a/covalent/_workflow/qnode.py b/covalent/_workflow/qnode.py
@@ -25,9 +25,9 @@
 
 from .._results_manager.qresult import QNodeFutureResult
 from .._shared_files import logger
+from .._shared_files.qelectron_utils import get_original_shots
 from .._shared_files.qinfo import QElectronInfo, QNodeSpecs
 from .._shared_files.qresult_utils import re_execute
-from .._shared_files.utils import get_original_shots
 from ..executor.qbase import BaseQExecutor
 from .qdevice import QEDevice
 

diff --git a/tests/qelectron_tests/pennylane_tests/conftest.py b/tests/qelectron_tests/pennylane_tests/conftest.py
@@ -33,7 +33,7 @@
 import pytest
 
 import covalent as ct
-from covalent._shared_files.utils import get_original_shots
+from covalent._shared_files.qelectron_utils import get_original_shots
 from covalent.quantum.qcluster.simulator import SIMULATOR_DEVICES
 
 SKIP_RETURN_TYPES = ["qml.apply", "qml.vn_entropy", "qml.mutual_info"]