Merge pull request #26 from ColibrITD-SAS/feat-capture-braket-hardprint

Catch hard print in Braket conversion

Author: hJaffaliColibritd

mpqp/qasm/open_qasm_2_and_3.py

@@ -5,11 +5,10 @@
 from enum import Enum
 
 from warnings import warn
-from textwrap import dedent
 from anytree import Node, PreOrderIter
 from typeguard import typechecked
 
-from mpqp.tools.errors import InstructionParsingError
+from mpqp.tools.errors import InstructionParsingError, OpenQASMTranslationWarning
 
 
 class Instr(Enum):
@@ -234,13 +233,13 @@ def add_std_lib():
         instructions_code += instr + ";\n"
     elif instr_name.lower() == "u":
         warn(
-            dedent(
-                """OpenQASMTranslationWarning: 
-                There is a phase e^(i(a+c)/2) difference between U(a,b,c) gate in 2.0 and 3.0.
-                We handled that for you by adding the extra phase at the right place. 
-                Be careful if you want to create a control gate from this circuit/gate, the
-                phase can become non-global."""
-            )
+            (
+                "\nThere is a phase e^(i(a+c)/2) difference between U(a,b,c) gate in 2.0 and 3.0.\n"
+                "We handled that for you by adding the extra phase at the right place.\n"
+                "Be careful if you want to create a control gate from this "
+                "circuit/gate, the phase can become non-global."
+            ),
+            OpenQASMTranslationWarning,
         )
         header_code += add_std_lib()
         instructions_code += "u3" + instr[1:] + ";\n"

mpqp/qasm/qasm_to_braket.py

@@ -1,10 +1,14 @@
 """File regrouping all features for translating QASM code to Amazon Braket objects."""
 
-from braket.ir.openqasm import Program
+import warnings
+from logging import Logger, getLogger
+
 from braket.circuits import Circuit
+from braket.ir.openqasm import Program
 from typeguard import typechecked
 
 from mpqp.qasm.open_qasm_2_and_3 import open_qasm_hard_includes
+from mpqp.tools.errors import UnsupportedBraketFeaturesWarning
 
 
 @typechecked
@@ -55,6 +59,25 @@ def qasm3_to_braket_Circuit(qasm3_str: str) -> Circuit:
     after_stdgates_included = open_qasm_hard_includes(qasm3_str, set())
     # NOTE : gphase is a already used in Braket and thus cannot be redefined as a native gate in OpenQASM.
     # We used ggphase instead
+    warning_message = (
+        "This program uses OpenQASM language features that may not be "
+        "supported on QPUs or on-demand simulators."
+    )
+
+    # handle the logger output
+    # capture their logger
+    braket_logger = getLogger()
+    # add logger handler
 
+    logger_out = []
     circuit = Circuit.from_ir(after_stdgates_included)
+
+    if warning_message in logger_out:
+        warnings.warn("\n" + warning_message, UnsupportedBraketFeaturesWarning)
+        del logger_out[logger_out.index(warning_message)]
+    # remove logger handler
+
+    for line in logger_out:
+        braket_logger.warning(line)
+
     return circuit

mpqp/tools/errors.py

@@ -1,32 +1,46 @@
 class InstructionParsingError(ValueError):
-    """Raised when an QASM instruction encountered by the parser is malformed"""
+    """Raised when an QASM instruction encountered by the parser is malformed."""
 
 
 class NumberQubitsError(ValueError):
-    """Raised when the number of qubits defining an instruction, a gate, or a measurement, is not coherent with the
-    related objets (circuit, matrix, observable, etc.)"""
+    """Raised when the number of qubits defining an instruction, a gate, or a
+    measurement, is not coherent with the related objects (circuit, matrix,
+    observable, etc...)."""
 
 
 class ResultAttributeError(AttributeError):
-    """Raised when one tries to access the attribute of the result that is incoherent with the associated job."""
+    """Raised when one tries to access the attribute of the result that is
+    incoherent with the associated job."""
 
 
 class DeviceJobIncompatibleError(ValueError):
-    """Raised when one tries to run a job with a JobType that is not suitable for the selected device
-    (for example SAMPLE job on a statevector simulator)"""
+    """Raised when one tries to run a job with a JobType that is not suitable
+    for the selected device (for example SAMPLE job on a statevector simulator)."""
 
 
 class RemoteExecutionError(ConnectionError):
-    """Raised when an error occurred during a remote connection, submission or execution"""
+    """Raised when an error occurred during a remote connection, submission or
+    execution."""
 
 
 class IBMRemoteExecutionError(RemoteExecutionError):
-    """Raised when an error occurred during the remote execution process of job(s) on an IBM device"""
+    """Raised when an error occurred during the remote execution process of
+    job(s) on an IBM device."""
 
 
 class QLMRemoteExecutionError(RemoteExecutionError):
-    """Raised when an error occurred during the remote execution process of job(s) on the remote QLM"""
+    """Raised when an error occurred during the remote execution process of
+    job(s) on the remote QLM."""
 
 
 class AWSBraketRemoteExecutionError(RemoteExecutionError):
-    """Raised when an error occurred during the remote execution process of job(s) on the remote Amazon Braket"""
+    """Raised when an error occurred during the remote execution process of
+    job(s) on the remote Amazon Braket."""
+
+
+class UnsupportedBraketFeaturesWarning(UserWarning):
+    """Warning for potential compatibility issues with Braket."""
+
+
+class OpenQASMTranslationWarning(UserWarning):
+    """Warning for potential translation error when exporting to OpenQASM."""