Fix mypy error

quantumlib · Jun 29, 2022 · 6dc35a7 · 6dc35a7
1 parent da9e55c
commit 6dc35a7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cirq-google/cirq_google/devices/serializable_device.py b/cirq-google/cirq_google/devices/serializable_device.py
@@ -103,8 +103,8 @@ def __init__(
         has_subcircuit_support: bool = cirq.FrozenCircuit in gate_definitions
 
         self._metadata = cirq.GridDeviceMetadata(
-            qubit_pairs=[
-                (cast(cirq.GridQubit, pair[0]), cast(cirq.GridQubit, pair[1]))
+            qubit_pairs=cast(List[Tuple[cirq.GridQubit, cirq.GridQubit]], [
+                (pair[0], pair[1])
                 for gate_defs in gate_definitions.values()
                 for gate_def in gate_defs
                 if gate_def.number_of_qubits == 2
@@ -113,7 +113,7 @@ def __init__(
                 and pair[0] < pair[1]
                 and isinstance(pair[0], cirq.GridQubit)
                 and isinstance(pair[1], cirq.GridQubit)
-            ],
+            ]),
             gateset=cirq.Gateset(
                 *(g for g in gate_definitions.keys() if issubclass(g, cirq.Gate)),
                 cirq.GlobalPhaseGate,