chore: cleaning the code and removing prints

Author: hJaffaliColibritd

mpqp/core/instruction/measurement/expectation_value.py

@@ -12,8 +12,6 @@ class to define your observable, and a :class:`ExpectationMeasure` to perform
 
 import numpy as np
 import numpy.typing as npt
-from typeguard import typechecked
-
 from mpqp.core.instruction.gates.native_gates import SWAP
 from mpqp.core.instruction.measurement.measure import Measure
 from mpqp.core.instruction.measurement.pauli_string import (
@@ -25,6 +23,7 @@ class to define your observable, and a :class:`ExpectationMeasure` to perform
 from mpqp.tools.errors import NumberQubitsError
 from mpqp.tools.generics import Matrix, OneOrMany
 from mpqp.tools.maths import is_diagonal, is_hermitian, is_power_of_two
+from typeguard import typechecked
 
 if TYPE_CHECKING:
     from braket.circuits.observables import Hermitian
@@ -113,7 +112,7 @@ def __init__(self, observable: Matrix | list[Real] | PauliString):
 
                     self._matrix = np.array(observable)
 
-            # correspond to if len(shape) == 1 or isinstance(observable, list)
+            # correspond to isinstance(observable, list)
             else:
                 self._is_diagonal = True
                 self._diag_elements = observable
@@ -359,17 +358,6 @@ def _check_targets_order(self):
         """Adjusted list of target qubits when they are not initially sorted and
         contiguous."""
 
-        print("Checking target order:")
-        print("Target size:", self.nb_qubits)
-        print(
-            "Observable sizes:",
-            (
-                [obs.nb_qubits for obs in self.observable]
-                if isinstance(self.observable, list)
-                else self.observable.nb_qubits
-            ),
-        )
-
     def get_pauli_grouping(
         self,
         method: Literal[

mpqp/execution/runner.py

@@ -23,9 +23,6 @@
 from typing import Iterable, Optional
 
 import numpy as np
-from sympy import Expr
-from typeguard import typechecked
-
 from mpqp.core.circuit import QCircuit
 from mpqp.core.instruction.breakpoint import Breakpoint
 from mpqp.core.instruction.measurement.basis_measure import BasisMeasure
@@ -52,6 +49,8 @@
 from mpqp.tools.display import state_vector_ket_shape
 from mpqp.tools.errors import DeviceJobIncompatibleError, RemoteExecutionError
 from mpqp.tools.generics import OneOrMany, find_index, flatten
+from sympy import Expr
+from typeguard import typechecked
 
 
 @typechecked
@@ -77,23 +76,16 @@ def adjust_measure(measure: ExpectationMeasure, circuit: QCircuit):
     Id_before = np.eye(2 ** measure.rearranged_targets[0])
     Id_after = np.eye(2 ** (circuit.nb_qubits - measure.rearranged_targets[-1] - 1))
 
-    # if isinstance(measure.observable, list):
     tweaked_observables = [
         Observable(np.kron(np.kron(Id_before, obs.matrix), Id_after))
         for obs in measure.observable
     ]
-    # else:
-    #     tweaked_observables = Observable(
-    #         np.kron(np.kron(Id_before, measure.observable.matrix), Id_after)
-    #     )
 
     tweaked_measure = ExpectationMeasure(
         tweaked_observables,
         list(range(circuit.nb_qubits)),
         measure.shots,
     )
-    print("Observable qubit size before tweaking:", measure.observable[0].nb_qubits)
-    print("Tweaked observable size:", tweaked_observables[0].nb_qubits)
     return tweaked_measure