fix: more progress on delayed imports

Author: Henri-ColibrITD

mpqp/execution/connection/aws_connection.py

@@ -1,7 +1,6 @@
 import os
 from typing import TYPE_CHECKING, Any
 
-import pkg_resources
 from termcolor import colored
 from typeguard import typechecked
 
@@ -113,9 +112,6 @@ def get_braket_device(device: AWSDevice, is_noisy: bool = False) -> "BraketDevic
          ResultType(name='Probability', observables=None, minShots=10, maxShots=100000)]
 
     """
-    import boto3
-    from botocore.exceptions import NoRegionError
-    from braket.aws import AwsDevice, AwsSession
     from braket.devices import LocalSimulator
 
     if not device.is_remote():
@@ -124,6 +120,11 @@ def get_braket_device(device: AWSDevice, is_noisy: bool = False) -> "BraketDevic
         else:
             return LocalSimulator()
 
+    import boto3
+    import pkg_resources
+    from botocore.exceptions import NoRegionError
+    from braket.aws import AwsDevice, AwsSession
+
     try:
         braket_client = boto3.client("braket", region_name=device.get_region())
         aws_session = AwsSession(braket_client=braket_client)

mpqp/execution/connection/ibm_connection.py

@@ -1,13 +1,11 @@
 from getpass import getpass
 from typing import TYPE_CHECKING
 
-from qiskit_ibm_provider import IBMProvider
-from qiskit_ibm_provider.accounts import AccountNotFoundError
-from qiskit_ibm_provider.api.exceptions import RequestsApiError
 from termcolor import colored
 from typeguard import typechecked
 
 if TYPE_CHECKING:
+    from qiskit_ibm_provider import IBMProvider
     from qiskit.providers.backend import BackendV1
     from qiskit_ibm_runtime import QiskitRuntimeService
 
@@ -29,6 +27,7 @@ def config_ibm_account(token: str):
     Raises:
         IBMRemoteExecutionError
     """
+    from qiskit_ibm_provider import IBMProvider
 
     try:
         IBMProvider.save_account(token=token, overwrite=True)
@@ -77,6 +76,9 @@ def test_connection() -> bool:
     Returns:
         ``False`` if login failed.
     """
+    from qiskit_ibm_provider import IBMProvider
+    from qiskit_ibm_provider.api.exceptions import RequestsApiError
+
     try:
         IBMProvider()
     except RequestsApiError as err:
@@ -88,7 +90,7 @@ def test_connection() -> bool:
     return True
 
 
-def get_IBMProvider() -> IBMProvider:
+def get_IBMProvider() -> "IBMProvider":
     """Returns the IBMProvider needed to get one or several backends for
     execution.
 
@@ -108,6 +110,10 @@ def get_IBMProvider() -> IBMProvider:
          <IBMBackend('ibm_osaka')>]
 
     """
+    from qiskit_ibm_provider import IBMProvider
+    from qiskit_ibm_provider.accounts import AccountNotFoundError
+    from qiskit_ibm_provider.api.exceptions import RequestsApiError
+
     global Ibm_Provider
     if Ibm_Provider is None:
         if get_env_variable("IBM_CONFIGURED") == "False":

mpqp/execution/job.py

@@ -24,11 +24,6 @@
 if TYPE_CHECKING:
     from enum import Enum
 
-from braket.aws import AwsQuantumTask
-from qat.comm.qlmaas.ttypes import JobStatus as QLM_JobStatus
-from qat.comm.qlmaas.ttypes import QLMServiceException
-from qiskit.providers import JobStatus as IBM_JobStatus
-
 from mpqp.core.instruction.measurement import BasisMeasure, ExpectationMeasure, Measure
 
 from ..core.circuit import QCircuit
@@ -171,6 +166,9 @@ def get_qlm_job_status(job_id: str) -> JobStatus:
     Args:
         job_id: Id of the job for which we want to retrieve the status.
     """
+    from qat.comm.qlmaas.ttypes import JobStatus as QLM_JobStatus
+    from qat.comm.qlmaas.ttypes import QLMServiceException
+
     try:
         qlm_status = get_QLMaaSConnection().get_status(job_id)
     except QLMServiceException as e:
@@ -203,6 +201,8 @@ def get_ibm_job_status(job_id: str) -> JobStatus:
     Args:
         job_id: Id of the job for which we want to retrieve the status.
     """
+    from qiskit.providers import JobStatus as IBM_JobStatus
+
     # test with QiskitRuntimeService
     if job_id in [e.job_id() for e in get_QiskitRuntimeService().jobs()]:
         ibm_job = get_QiskitRuntimeService().job(job_id)
@@ -237,6 +237,8 @@ def get_aws_job_status(job_id: str) -> JobStatus:
     Args:
         job_id: Id of the job for which we want to retrieve the status.
     """
+    from braket.aws import AwsQuantumTask
+
     task = AwsQuantumTask(job_id)
     state = task.state()
     if state == "FAILED":

mpqp/execution/providers/ibm.py

@@ -3,21 +3,14 @@
 import math
 from typing import TYPE_CHECKING, Optional
 
-from qiskit import QuantumCircuit
-from qiskit.compiler import transpile
-from qiskit.primitives import BackendEstimator
-from qiskit.primitives import Estimator as Qiskit_Estimator
-from qiskit.primitives import EstimatorResult, SamplerResult
-from qiskit.providers import BackendV1, BackendV2
-from qiskit.providers import JobStatus as IBM_JobStatus
-from qiskit.quantum_info import Operator
-from qiskit.result import Result as QiskitResult
-from qiskit_aer import Aer, AerSimulator
-from qiskit_ibm_provider.job import IBMJob
-from qiskit_ibm_runtime import Estimator as Runtime_Estimator
-from qiskit_ibm_runtime import RuntimeJob
-from qiskit_ibm_runtime import Sampler as Runtime_Sampler
-from qiskit_ibm_runtime import Session
+if TYPE_CHECKING:
+    from qiskit import QuantumCircuit
+    from qiskit.primitives import EstimatorResult, SamplerResult
+    from qiskit.providers import BackendV1, BackendV2
+    from qiskit.result import Result as QiskitResult
+    from qiskit_ibm_provider.job import IBMJob
+    from qiskit_ibm_runtime import RuntimeJob
+
 from typeguard import typechecked
 
 from mpqp.core.circuit import QCircuit
@@ -71,6 +64,10 @@ def compute_expectation_value(
         This function is not meant to be used directly, please use
         :func:``run<mpqp.execution.runner.run>`` instead.
     """
+    from qiskit.primitives import BackendEstimator
+    from qiskit.primitives import Estimator as Qiskit_Estimator
+    from qiskit.quantum_info import Operator
+
     if not isinstance(job.measure, ExpectationMeasure):
         raise ValueError(
             "Cannot compute expectation value if measure used in job is not of "
@@ -159,6 +156,10 @@ def run_aer(job: Job):
         This function is not meant to be used directly, please use
         :func:``run<mpqp.execution.runner.run>`` instead.
     """
+    from qiskit import QuantumCircuit
+    from qiskit.compiler import transpile
+    from qiskit_aer import Aer, AerSimulator
+
     qiskit_circuit = (
         job.circuit.without_measurements().to_other_language(Language.QISKIT)
         if (job.job_type == JobType.STATE_VECTOR)
@@ -225,6 +226,12 @@ def submit_ibmq(job: Job) -> tuple[str, RuntimeJob | IBMJob]:
         This function is not meant to be used directly, please use
         :func:``run<mpqp.execution.runner.run>`` instead.
     """
+    from qiskit import QuantumCircuit
+    from qiskit.quantum_info import Operator
+    from qiskit_ibm_runtime import Estimator as Runtime_Estimator
+    from qiskit_ibm_runtime import Sampler as Runtime_Sampler
+    from qiskit_ibm_runtime import Session
+
     if job.job_type == JobType.STATE_VECTOR:
         raise DeviceJobIncompatibleError(
             "State vector cannot be computed using IBM remote simulators and"
@@ -317,6 +324,9 @@ def extract_result(
     Returns:
         The ``qiskit`` result converted to our format.
     """
+    from qiskit.primitives import EstimatorResult, SamplerResult
+    from qiskit.result import Result as QiskitResult
+    from qiskit_ibm_runtime import RuntimeJob
 
     if job is not None and (
         isinstance(result, EstimatorResult) != (job.job_type == JobType.OBSERVABLE)
@@ -436,9 +446,13 @@ def get_result_from_ibm_job_id(job_id: str) -> Result:
     Returns:
         The result converted to our format.
     """
+    from qiskit.providers import BackendV1, BackendV2
+    from qiskit.providers import JobStatus as IBM_JobStatus
+
     # search for job id in the connector given in parameter first
     # if not found, try with IBMProvider, then QiskitRuntimeService
     # if not found, raise an error
+
     connector = get_IBMProvider()
     ibm_job = (
         connector.retrieve_job(job_id)

mpqp/execution/result.py

@@ -28,7 +28,6 @@
 
 import numpy as np
 import numpy.typing as npt
-from matplotlib import pyplot as plt
 from typeguard import typechecked
 
 from mpqp.execution.devices import AvailableDevice
@@ -412,6 +411,8 @@ def plot(self, show: bool = True):
             show: ``plt.show()`` is only executed if ``show``, useful to batch
                 plots.
         """
+        from matplotlib import pyplot as plt
+
         if show:
             plt.figure()
 
@@ -532,6 +533,8 @@ def plot(self, show: bool = True):
             show: ``plt.show()`` is only executed if ``show``, useful to batch
                 plots.
         """
+        from matplotlib import pyplot as plt
+
         n_cols = math.ceil((len(self.results) + 1) // 2)
         n_rows = math.ceil(len(self.results) / n_cols)