[FIX] Fixed mypy errors.

Author: MartinBeseda

qbench/batching.py

@@ -84,7 +84,9 @@ def execute_in_batches(
     #     for batch in batches
     # )
 
-    result_gen = (BatchJob(sampler.run(batch.circuits, shots=shots), batch.keys) for batch in batches)
+    result_gen = (
+        BatchJob(sampler.run(batch.circuits, shots=shots), batch.keys) for batch in batches
+    )
 
     # Declare the variable 'result'
     result: Iterable[BatchJob]

qbench/fourier/_components/__init__.py

@@ -15,4 +15,5 @@ def discrimination_probability_upper_bound(
     """
     return 0.5 + 0.25 * np.abs(1 - np.exp(1j * phi))
 
+
 __all__ = ["discrimination_probability_upper_bound"]

qbench/fourier_certification/experiment_runner.py

@@ -2,14 +2,14 @@
 
 from collections import Counter, defaultdict
 from logging import getLogger
-from typing import Dict, Iterable, List, Optional, Tuple, Union, cast, Any, Generator
+from typing import Dict, Iterable, List, Optional, Tuple, Union, cast
 
 import numpy as np
 import pandas as pd
 from mthree import M3Mitigation
 from qiskit import QiskitError, QuantumCircuit, transpile
 from qiskit.providers import JobV1
-from qiskit_ibm_runtime import RuntimeJobV2, RuntimeJob
+from qiskit_ibm_runtime import RuntimeJobV2
 from tqdm import tqdm
 
 from qbench.batching import BatchJob, execute_in_batches
@@ -147,7 +147,7 @@ def _extract_result_from_job(
     return ResultForCircuit.parse_obj(result)
 
 
-CircuitKey = Tuple[int, int, str, float]
+CircuitKey = Tuple[int, int, str, float, float]
 
 
 def _collect_circuits_and_keys(
@@ -207,6 +207,7 @@ def _iter_batches(batches: Iterable[BatchJob]) -> Iterable[Tuple[int, CircuitKey
         for i, key in enumerate(tqdm(batch.keys, desc="Circuit", leave=False))
     )
 
+
 # def _iter_batches(batches: Iterable[BatchJob]) -> Generator[
 #     tuple[int, tuple[int, Any], JobV1 | RuntimeJob | RuntimeJobV2], None, None]:
 #     """Iterate batches in a flat manner.