Remove example_test.py

Author: Liu Keyu

pyproject.toml

@@ -46,6 +46,7 @@ dependencies = [
     "numpy>=1.22,<2; sys_platform == 'darwin' and 'x86_64' in platform_machine and python_version < '3.13'",  # Restrict numpy v2 for macOS x86 since it is not supported anymore since torch v2.3.0
     "torch>=2.2.2,<2.3.0; sys_platform == 'darwin' and 'x86_64' in platform_machine and python_version < '3.13'",  # Restrict torch v2.3.0 for macOS x86 since it is not supported anymore.
     "typing-extensions>=4.1",  # for `assert_never`
+    "qiskit-ibm-transpiler>=0.2.0",
 ]
 
 classifiers = [

src/mqt/predictor/rl/helper.py

@@ -12,7 +12,7 @@
 
 import logging
 from pathlib import Path
-from typing import TYPE_CHECKING, Optional, Callable
+from typing import TYPE_CHECKING, Optional, Callable, List, Tuple, Dict, Any
 
 import numpy as np
 from qiskit import QuantumCircuit
@@ -34,7 +34,7 @@
 
 logger = logging.getLogger("mqt-predictor")
 
-def extract_cregs_and_measurements(qc: QuantumCircuit) -> tuple[list[ClassicalRegister], list[tuple[Instruction, list, list]]]:
+def extract_cregs_and_measurements(qc: QuantumCircuit) -> Tuple[List[ClassicalRegister], List[tuple[Instruction, List, List]]]:
     """
     Extracts classical registers and measurement operations from a quantum circuit.
 
@@ -77,9 +77,9 @@ def remove_cregs(qc: QuantumCircuit) -> QuantumCircuit:
 
 def add_cregs_and_measurements(
     qc: QuantumCircuit,
-    cregs: list[ClassicalRegister],
-    measurements: list[tuple[Instruction, list, list]],
-    qubit_map: Optional[dict] = None,
+    cregs: List[ClassicalRegister],
+    measurements: List[Tuple[Instruction, List, List]],
+    qubit_map: Optional[Dict] = None,
 ) -> QuantumCircuit:
     """
     Adds classical registers and measurement operations back to the quantum circuit.
@@ -155,9 +155,9 @@ def best_of_n_passmanager(
     action: Action,
     device: Target,
     qc: QuantumCircuit,
-    max_iteration: tuple[int, int] = (20, 20),
+    max_iteration: Tuple[int, int] = (20, 20),
     metric_fn: Optional[Callable[[QuantumCircuit], float]] = None,
-)-> tuple[QuantumCircuit, dict[str, any]]:
+)-> tuple[QuantumCircuit, Dict[str, Any]]:
     """
     Runs the given transpile_pass multiple times and keeps the best result.
 

src/mqt/predictor/rl/predictorenv.py

@@ -316,18 +316,18 @@ def apply_action(self, action_index: int) -> QuantumCircuit | None:
         raise ValueError(msg)
 
     def _apply_qiskit_action(self, action: Action, action_index: int) -> QuantumCircuit:
-        if action.get("stochastic", False):
+        if getattr(action, "stochastic", False):
             def metric_fn(circ: QuantumCircuit) -> float:
-                return circ.count_ops().get("swap", 0)
+                return float(circ.count_ops().get("swap", 0))
             # for stochastic actions, pass the layout/routing trials parameter
             max_iteration = self.max_iter
-            if "Sabre" in action["name"] and "AIRouting" not in action["name"]:
+            if "Sabre" in action.name and "AIRouting" not in action.name:
                 # Internal trials for Sabre
                 transpile_pass = action.transpile_pass(self.device, max_iteration)
                 pm = PassManager(transpile_pass)
                 altered_qc = pm.run(self.state)
                 pm_property_set = dict(pm.property_set)
-            elif "AIRouting" in action["name"]:
+            elif "AIRouting" in action.name:
                 # Run AIRouting in custom loop
                 altered_qc, pm_property_set = best_of_n_passmanager(
                     action,