Fix: resolve pre-commit issues and add missing annotations

Fix: resolve pre-commit issues and add missing annotations

Fix: resolve pre-commit issues and add missing annotations

Fix: resolve pre-commit issues and add missing annotations

Author: Liu Keyu

pyproject.toml

@@ -46,7 +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",
+    "qiskit-ibm-transpiler[ai-local-mode]>=0.2.0"
 ]
 
 classifiers = [
@@ -164,7 +164,7 @@ implicit_reexport = true
 # recent versions of `gym` are typed, but stable-baselines3 pins a very old version of gym.
 # qiskit is not yet marked as typed, but is typed mostly.
 # the other libraries do not have type stubs.
-module = ["qiskit.*", "joblib.*", "sklearn.*", "matplotlib.*", "gymnasium.*", "mqt.bench.*", "sb3_contrib.*", "bqskit.*", "qiskit_ibm_runtime.*", "networkx.*", "stable_baselines3.*"]
+module = ["qiskit.*", "joblib.*", "sklearn.*", "matplotlib.*", "gymnasium.*", "mqt.bench.*", "sb3_contrib.*", "bqskit.*", "qiskit_ibm_runtime.*", "networkx.*", "stable_baselines3.*","qiskit_ibm_transpiler.*"]
 ignore_missing_imports = true
 
 [tool.ruff]

src/mqt/predictor/rl/helper.py

@@ -17,10 +17,10 @@
 import numpy as np
 from qiskit import QuantumCircuit
 from qiskit.converters import circuit_to_dag, dag_to_circuit
-from qiskit.circuit import ClassicalRegister, QuantumRegister, Instruction
+from qiskit.circuit import ClassicalRegister, QuantumRegister, Instruction, Qubit
 from qiskit.transpiler import PassManager, Target
 from qiskit.dagcircuit import DAGCircuit
-from qiskit_ibm_transpiler.ai.routing import AIRouting
+from qiskit_ibm_transpiler.ai.routing import AIRouting 
 
 from mqt.predictor.utils import calc_supermarq_features
 from mqt.predictor.rl.actions import Action
@@ -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[Any], List[Any]]]]:
     """
     Extracts classical registers and measurement operations from a quantum circuit.
 
@@ -78,8 +78,8 @@ 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,
+    measurements: List[Tuple[Instruction, List[Any], List[Any]]],
+    qubit_map: Optional[Dict[Qubit,Qubit]] = None,
 ) -> QuantumCircuit:
     """
     Adds classical registers and measurement operations back to the quantum circuit.