cirq-core 1.5.0.dev20250409225226__py3-none-any.whl → 1.6.0__py3-none-any.whl

cirq/contrib/quimb/density_matrix_test.py

@@ -1,11 +1,14 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
+
+from __future__ import annotations
+
 import numpy as np
 
 import cirq
 import cirq.contrib.quimb as ccq
 
 
-def test_tensor_density_matrix_1():
+def test_tensor_density_matrix_1() -> None:
     q = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.YPowGate(exponent=0.25).on(q[0]))
 
@@ -14,7 +17,7 @@ def test_tensor_density_matrix_1():
     np.testing.assert_allclose(rho1, rho2, atol=1e-15)
 
 
-def test_tensor_density_matrix_optional_qubits():
+def test_tensor_density_matrix_optional_qubits() -> None:
     q = cirq.LineQubit.range(2)
     c = cirq.Circuit(cirq.YPowGate(exponent=0.25).on(q[0]))
 
@@ -23,7 +26,7 @@ def test_tensor_density_matrix_optional_qubits():
     np.testing.assert_allclose(rho1, rho2, atol=1e-15)
 
 
-def test_tensor_density_matrix_noise_1():
+def test_tensor_density_matrix_noise_1() -> None:
     q = cirq.LineQubit.range(2)
     c = cirq.Circuit(
         cirq.YPowGate(exponent=0.25).on(q[0]),
@@ -36,7 +39,7 @@ def test_tensor_density_matrix_noise_1():
     np.testing.assert_allclose(rho1, rho2, atol=1e-15)
 
 
-def test_tensor_density_matrix_2():
+def test_tensor_density_matrix_2() -> None:
     q = cirq.LineQubit.range(2)
     rs = np.random.RandomState(52)
     for _ in range(10):
@@ -47,7 +50,7 @@ def test_tensor_density_matrix_2():
         np.testing.assert_allclose(rho1, rho2, atol=1e-8)
 
 
-def test_tensor_density_matrix_3():
+def test_tensor_density_matrix_3() -> None:
     qubits = cirq.LineQubit.range(10)
     circuit = cirq.testing.random_circuit(qubits=qubits, n_moments=10, op_density=0.8)
     rho1 = cirq.final_density_matrix(circuit, dtype=np.complex128)
@@ -55,7 +58,7 @@ def test_tensor_density_matrix_3():
     np.testing.assert_allclose(rho1, rho2, atol=1e-8)
 
 
-def test_tensor_density_matrix_4():
+def test_tensor_density_matrix_4() -> None:
     qubits = cirq.LineQubit.range(4)
     circuit = cirq.testing.random_circuit(qubits=qubits, n_moments=100, op_density=0.8)
     circuit = cirq.drop_empty_moments(circuit)
@@ -66,7 +69,7 @@ def test_tensor_density_matrix_4():
     np.testing.assert_allclose(rho1, rho2, atol=1e-8)
 
 
-def test_tensor_density_matrix_gridqubit():
+def test_tensor_density_matrix_gridqubit() -> None:
     qubits = cirq.GridQubit.rect(2, 2)
     circuit = cirq.testing.random_circuit(qubits=qubits, n_moments=10, op_density=0.8)
     circuit = cirq.drop_empty_moments(circuit)

cirq/contrib/quimb/grid_circuits.py

@@ -12,12 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Iterator
+from __future__ import annotations
 
-import networkx as nx
+from typing import Iterator, TYPE_CHECKING
 
 import cirq
 
+if TYPE_CHECKING:
+    import networkx as nx
+
 
 def get_grid_moments(
     problem_graph: nx.Graph, two_qubit_gate=cirq.ZZPowGate

cirq/contrib/quimb/grid_circuits_test.py

@@ -1,4 +1,7 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
+
+from __future__ import annotations
+
 import networkx as nx
 import numpy as np
 import pytest

cirq/contrib/quimb/mps_simulator.py

@@ -21,7 +21,7 @@ from __future__ import annotations
 
 import dataclasses
 import math
-from typing import Any, Dict, List, Optional, Sequence, Set, Tuple, TYPE_CHECKING, Union
+from typing import Any, Sequence, TYPE_CHECKING
 
 import numpy as np
 import quimb.tensor as qtn
@@ -43,7 +43,7 @@ class MPSOptions:
     # How to split the tensor. Refer to the Quimb documentation for the exact meaning.
     method: str = 'svds'
     # If integer, the maxmimum number of singular values to keep, regardless of ``cutoff``.
-    max_bond: Optional[int] = None
+    max_bond: int | None = None
     # Method with which to apply the cutoff threshold. Refer to the Quimb documentation.
     cutoff_mode: str = 'rsum2'
     # The threshold below which to discard singular values. Refer to the Quimb documentation.
@@ -63,7 +63,7 @@ class MPSSimulator(
         noise: cirq.NOISE_MODEL_LIKE = None,
         seed: cirq.RANDOM_STATE_OR_SEED_LIKE = None,
         simulation_options: MPSOptions = MPSOptions(),
-        grouping: Optional[Dict[cirq.Qid, int]] = None,
+        grouping: dict[cirq.Qid, int] | None = None,
     ):
         """Creates instance of `MPSSimulator`.
 
@@ -86,7 +86,7 @@ class MPSSimulator(
 
     def _create_partial_simulation_state(
         self,
-        initial_state: Union[int, MPSState],
+        initial_state: int | MPSState,
         qubits: Sequence[cirq.Qid],
         classical_data: cirq.ClassicalDataStore,
     ) -> MPSState:
@@ -122,7 +122,7 @@ class MPSSimulator(
     def _create_simulator_trial_result(
         self,
         params: cirq.ParamResolver,
-        measurements: Dict[str, np.ndarray],
+        measurements: dict[str, np.ndarray],
         final_simulator_state: cirq.SimulationStateBase[MPSState],
     ) -> MPSTrialResult:
         """Creates a single trial results with the measurements.
@@ -147,7 +147,7 @@ class MPSTrialResult(simulator_base.SimulationTrialResultBase['MPSState']):
     def __init__(
         self,
         params: cirq.ParamResolver,
-        measurements: Dict[str, np.ndarray],
+        measurements: dict[str, np.ndarray],
         final_simulator_state: cirq.SimulationStateBase[MPSState],
     ) -> None:
         super().__init__(
@@ -212,11 +212,11 @@ class _MPSHandler(qis.QuantumStateRepresentation):
 
     def __init__(
         self,
-        qid_shape: Tuple[int, ...],
-        grouping: Dict[int, int],
-        M: List[qtn.Tensor],
+        qid_shape: tuple[int, ...],
+        grouping: dict[int, int],
+        M: list[qtn.Tensor],
         format_i: str,
-        estimated_gate_error_list: List[float],
+        estimated_gate_error_list: list[float],
         simulation_options: MPSOptions = MPSOptions(),
     ):
         """Creates an MPSQuantumState
@@ -241,8 +241,8 @@ class _MPSHandler(qis.QuantumStateRepresentation):
     def create(
         cls,
         *,
-        qid_shape: Tuple[int, ...],
-        grouping: Dict[int, int],
+        qid_shape: tuple[int, ...],
+        grouping: dict[int, int],
         initial_state: int = 0,
         simulation_options: MPSOptions = MPSOptions(),
     ):
@@ -342,7 +342,7 @@ class _MPSHandler(qis.QuantumStateRepresentation):
         sorted_ind = tuple(sorted(state_vector.inds))
         return state_vector.fuse({'i': sorted_ind}).data
 
-    def partial_trace(self, keep_axes: Set[int]) -> np.ndarray:
+    def partial_trace(self, keep_axes: set[int]) -> np.ndarray:
         """Traces out all qubits except keep_axes.
 
         Args:
@@ -484,8 +484,8 @@ class _MPSHandler(qis.QuantumStateRepresentation):
 
     def _measure(
         self, axes: Sequence[int], prng: np.random.RandomState, collapse_state_vector=True
-    ) -> List[int]:
-        results: List[int] = []
+    ) -> list[int]:
+        results: list[int] = []
 
         if collapse_state_vector:
             state = self
@@ -523,7 +523,7 @@ class _MPSHandler(qis.QuantumStateRepresentation):
 
     def measure(
         self, axes: Sequence[int], seed: cirq.RANDOM_STATE_OR_SEED_LIKE = None
-    ) -> List[int]:
+    ) -> list[int]:
         """Measures the MPS.
 
         Args:
@@ -547,7 +547,7 @@ class _MPSHandler(qis.QuantumStateRepresentation):
             The samples in order.
         """
 
-        measurements: List[List[int]] = []
+        measurements: list[list[int]] = []
         prng = value.parse_random_state(seed)
 
         for _ in range(repetitions):
@@ -566,9 +566,9 @@ class MPSState(SimulationState[_MPSHandler]):
         qubits: Sequence[cirq.Qid],
         prng: np.random.RandomState,
         simulation_options: MPSOptions = MPSOptions(),
-        grouping: Optional[Dict[cirq.Qid, int]] = None,
+        grouping: dict[cirq.Qid, int] | None = None,
         initial_state: int = 0,
-        classical_data: Optional[cirq.ClassicalDataStore] = None,
+        classical_data: cirq.ClassicalDataStore | None = None,
     ):
         """Creates and MPSState
 
@@ -622,7 +622,7 @@ class MPSState(SimulationState[_MPSHandler]):
         """
         return self._state.state_vector()
 
-    def partial_trace(self, keep_qubits: Set[cirq.Qid]) -> np.ndarray:
+    def partial_trace(self, keep_qubits: set[cirq.Qid]) -> np.ndarray:
         """Traces out all qubits except keep_qubits.
 
         Args:

cirq/contrib/quimb/mps_simulator_test.py

@@ -1,4 +1,7 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
+
+from __future__ import annotations
+
 import itertools
 import math
 

cirq/contrib/quimb/state_vector.py

@@ -1,13 +1,18 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
+
+from __future__ import annotations
+
 import warnings
-from typing import cast, Dict, List, Optional, Sequence, Tuple, Union
+from typing import cast, Sequence, TYPE_CHECKING
 
-import numpy as np
 import quimb
 import quimb.tensor as qtn
 
 import cirq
 
+if TYPE_CHECKING:
+    import numpy as np
+
 
 def _get_quimb_version():
     """Returns the quimb version and parsed (major,minor) numbers if possible.
@@ -25,10 +30,8 @@ QUIMB_VERSION = _get_quimb_version()
 
 
 def circuit_to_tensors(
-    circuit: cirq.Circuit,
-    qubits: Optional[Sequence[cirq.Qid]] = None,
-    initial_state: Union[int, None] = 0,
-) -> Tuple[List[qtn.Tensor], Dict['cirq.Qid', int], None]:
+    circuit: cirq.Circuit, qubits: Sequence[cirq.Qid] | None = None, initial_state: int | None = 0
+) -> tuple[list[qtn.Tensor], dict[cirq.Qid, int], None]:
     """Given a circuit, construct a tensor network representation.
 
     Indices are named "i{i}_q{x}" where i is a time index and x is a
@@ -62,7 +65,7 @@ def circuit_to_tensors(
 
     qubit_frontier = {q: 0 for q in qubits}
     positions = None
-    tensors: List[qtn.Tensor] = []
+    tensors: list[qtn.Tensor] = []
 
     if initial_state == 0:
         for q in qubits:
@@ -89,7 +92,7 @@ def circuit_to_tensors(
 
 
 def tensor_state_vector(
-    circuit: cirq.Circuit, qubits: Optional[Sequence[cirq.Qid]] = None
+    circuit: cirq.Circuit, qubits: Sequence[cirq.Qid] | None = None
 ) -> np.ndarray:
     """Given a circuit contract a tensor network into a final state vector."""
     if qubits is None:
@@ -102,9 +105,7 @@ def tensor_state_vector(
     return tn.to_dense(f_inds)
 
 
-def tensor_unitary(
-    circuit: cirq.Circuit, qubits: Optional[Sequence[cirq.Qid]] = None
-) -> np.ndarray:
+def tensor_unitary(circuit: cirq.Circuit, qubits: Sequence[cirq.Qid] | None = None) -> np.ndarray:
     """Given a circuit contract a tensor network into a dense unitary
     of the circuit."""
     if qubits is None:

cirq/contrib/quimb/state_vector_test.py

@@ -1,4 +1,7 @@
 # pylint: disable=wrong-or-nonexistent-copyright-notice
+
+from __future__ import annotations
+
 import functools
 import operator
 

cirq/contrib/quirk/export_to_quirk.py

@@ -12,9 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import json
 import urllib.parse
-from typing import Any, cast, Iterable, List, Tuple
+from typing import Any, cast, Iterable
 
 from cirq import circuits, devices, ops, protocols
 from cirq.contrib.quirk.linearize_circuit import linearize_circuit_qubits
@@ -40,7 +42,7 @@ def _try_convert_to_quirk_gate(op: ops.Operation, prefer_unknown_gate_to_failure
 
 def _to_quirk_cols(
     op: ops.Operation, prefer_unknown_gate_to_failure: bool
-) -> Iterable[Tuple[List[Any], bool]]:
+) -> Iterable[tuple[list[Any], bool]]:
     gate = _try_convert_to_quirk_gate(op, prefer_unknown_gate_to_failure)
     qubits = cast(Iterable[devices.LineQubit], op.qubits)
 
@@ -76,7 +78,7 @@ def circuit_to_quirk_url(
     circuit = circuit.copy()
     linearize_circuit_qubits(circuit)
 
-    cols: List[List[Any]] = []
+    cols: list[list[Any]] = []
     for moment in circuit:
         can_merges = []
         for op in moment.operations:

cirq/contrib/quirk/export_to_quirk_test.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import pytest
 import sympy
 
@@ -19,7 +21,7 @@ import cirq
 from cirq.contrib.quirk.export_to_quirk import circuit_to_quirk_url
 
 
-def assert_links_to(circuit: cirq.Circuit, expected: str, **kwargs):
+def assert_links_to(circuit: cirq.Circuit, expected: str, **kwargs) -> None:
     actual = circuit_to_quirk_url(circuit, **kwargs)
     actual = actual.replace('\n', '').replace(' ', '').strip()
     expected = expected.replace('],[', '],\n[').strip()
@@ -27,7 +29,7 @@ def assert_links_to(circuit: cirq.Circuit, expected: str, **kwargs):
     assert actual == expected
 
 
-def test_x_z_same_col():
+def test_x_z_same_col() -> None:
     a = cirq.NamedQubit('a')
     b = cirq.NamedQubit('b')
     circuit = cirq.Circuit(cirq.X(a), cirq.Z(b))
@@ -44,7 +46,7 @@ def test_x_z_same_col():
     )
 
 
-def test_x_cnot_split_cols():
+def test_x_cnot_split_cols() -> None:
     a = cirq.NamedQubit('a')
     b = cirq.NamedQubit('b')
     c = cirq.NamedQubit('c')
@@ -58,7 +60,7 @@ def test_x_cnot_split_cols():
     )
 
 
-def test_cz_cnot_split_cols():
+def test_cz_cnot_split_cols() -> None:
     a = cirq.NamedQubit('a')
     b = cirq.NamedQubit('b')
     c = cirq.NamedQubit('c')
@@ -72,7 +74,7 @@ def test_cz_cnot_split_cols():
     )
 
 
-def test_various_known_gate_types():
+def test_various_known_gate_types() -> None:
     a = cirq.NamedQubit('a')
     b = cirq.NamedQubit('b')
     circuit = cirq.Circuit(
@@ -116,7 +118,7 @@ def test_various_known_gate_types():
     )
 
 
-def test_parameterized_gates():
+def test_parameterized_gates() -> None:
     a = cirq.LineQubit(0)
     s = sympy.Symbol('s')
     t = sympy.Symbol('t')
@@ -182,7 +184,7 @@ class MysteryGate(cirq.testing.SingleQubitGate):
         return True
 
 
-def test_various_unknown_gate_types():
+def test_various_unknown_gate_types() -> None:
     a = cirq.NamedQubit('a')
     b = cirq.NamedQubit('b')
     circuit = cirq.Circuit(
@@ -225,7 +227,7 @@ def test_various_unknown_gate_types():
     )
 
 
-def test_formulaic_exponent_export():
+def test_formulaic_exponent_export() -> None:
     a = cirq.LineQubit(0)
     t = sympy.Symbol('t')
     assert_links_to(
@@ -241,7 +243,7 @@ def test_formulaic_exponent_export():
     )
 
 
-def test_formulaic_rotation_xyz_export():
+def test_formulaic_rotation_xyz_export() -> None:
     a = cirq.LineQubit(0)
     t = sympy.Symbol('t')
     assert_links_to(
@@ -262,9 +264,9 @@ def test_formulaic_rotation_xyz_export():
         _ = circuit_to_quirk_url(cirq.Circuit(cirq.rx(sympy.FallingFactorial(t, t)).on(a)))
 
 
-def test_unrecognized_single_qubit_gate_with_matrix():
+def test_unrecognized_single_qubit_gate_with_matrix() -> None:
     a = cirq.NamedQubit('a')
-    circuit = cirq.Circuit(cirq.PhasedXPowGate(phase_exponent=0).on(a) ** 0.2731)
+    circuit = cirq.Circuit(cirq.PhasedXPowGate(exponent=0.2731, phase_exponent=0).on(a))
     assert_links_to(
         circuit,
         """
@@ -279,7 +281,7 @@ def test_unrecognized_single_qubit_gate_with_matrix():
     )
 
 
-def test_unknown_gate():
+def test_unknown_gate() -> None:
     class UnknownGate(cirq.testing.SingleQubitGate):
         pass
 
@@ -299,7 +301,7 @@ def test_unknown_gate():
     )
 
 
-def test_controlled_gate():
+def test_controlled_gate() -> None:
     a, b, c, d = cirq.LineQubit.range(4)
     circuit = cirq.Circuit(cirq.ControlledGate(cirq.ControlledGate(cirq.CZ)).on(a, d, c, b))
     assert_links_to(
@@ -334,7 +336,7 @@ def test_controlled_gate():
     )
 
 
-def test_toffoli():
+def test_toffoli() -> None:
     a, b, c, d = cirq.LineQubit.range(4)
 
     # Raw.
@@ -372,7 +374,7 @@ def test_toffoli():
     )
 
 
-def test_fredkin():
+def test_fredkin() -> None:
     a, b, c = cirq.LineQubit.range(3)
     circuit = cirq.Circuit(cirq.FREDKIN(a, b, c))
     assert_links_to(
@@ -398,7 +400,7 @@ def test_fredkin():
     )
 
 
-def test_ccz():
+def test_ccz() -> None:
     a, b, c, d = cirq.LineQubit.range(4)
 
     # Raw.

cirq/contrib/quirk/linearize_circuit.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from typing import Callable
 
 from cirq import circuits, devices, ops

cirq/contrib/quirk/quirk_gate.py

@@ -12,7 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, Callable, cast, Dict, Optional, Union
+from __future__ import annotations
+
+from typing import Any, Callable, cast
 
 import numpy as np
 import sympy
@@ -46,7 +48,7 @@ class QuirkOp:
         self.keys = keys
         self.can_merge = can_merge
 
-    def controlled(self, control_count: int = 1) -> 'QuirkOp':
+    def controlled(self, control_count: int = 1) -> QuirkOp:
         return QuirkOp(*['•'] * control_count, *self.keys, can_merge=False)
 
 
@@ -68,7 +70,7 @@ def _is_supported_formula(formula: sympy.Basic) -> bool:
     return False
 
 
-def _val_to_quirk_formula(t: Union[float, sympy.Basic]) -> str:
+def _val_to_quirk_formula(t: float | sympy.Basic) -> str:
     if isinstance(t, sympy.Basic):
         if not set(t.free_symbols) <= {sympy.Symbol('t')}:
             raise ValueError(f'Symbol other than "t": {t!r}.')
@@ -79,7 +81,7 @@ def _val_to_quirk_formula(t: Union[float, sympy.Basic]) -> str:
     return f'{float(t):.4f}'
 
 
-def angle_to_exponent_key(t: Union[float, sympy.Basic]) -> Optional[str]:
+def angle_to_exponent_key(t: float | sympy.Basic) -> str | None:
     if isinstance(t, sympy.Basic):
         if t == sympy.Symbol('t'):
             return '^t'
@@ -107,11 +109,10 @@ def angle_to_exponent_key(t: Union[float, sympy.Basic]) -> Optional[str]:
     return None
 
 
-def single_qubit_matrix_gate(matrix: Optional[np.ndarray]) -> Optional[QuirkOp]:
+def single_qubit_matrix_gate(matrix: np.ndarray | None) -> QuirkOp | None:
     if matrix is None or matrix.shape[0] != 2:
         return None
 
-    # pylint: disable=consider-using-f-string
     matrix = matrix.round(6)
     matrix_repr = '{{%s+%si,%s+%si},{%s+%si,%s+%si}}' % (
         np.real(matrix[0, 0]),
@@ -135,7 +136,7 @@ def single_qubit_matrix_gate(matrix: Optional[np.ndarray]) -> Optional[QuirkOp]:
     return QuirkOp({'id': '?', 'matrix': matrix_repr})
 
 
-def known_quirk_op_for_operation(op: ops.Operation) -> Optional[QuirkOp]:
+def known_quirk_op_for_operation(op: ops.Operation) -> QuirkOp | None:
     if isinstance(op, ops.GateOperation):
         return _gate_to_quirk_op(op.gate)
     if isinstance(op, ops.ControlledOperation):
@@ -143,7 +144,7 @@ def known_quirk_op_for_operation(op: ops.Operation) -> Optional[QuirkOp]:
     return None
 
 
-def _gate_to_quirk_op(gate: ops.Gate) -> Optional[QuirkOp]:
+def _gate_to_quirk_op(gate: ops.Gate) -> QuirkOp | None:
     for gate_type, func in _known_gate_conversions.items():
         if isinstance(gate, gate_type):
             return func(gate)
@@ -176,45 +177,45 @@ def z_to_quirk_op(gate: ops.ZPowGate) -> QuirkOp:
     return xyz_to_quirk_op('z', gate)
 
 
-def cz_to_quirk_op(gate: ops.CZPowGate) -> Optional[QuirkOp]:
+def cz_to_quirk_op(gate: ops.CZPowGate) -> QuirkOp | None:
     return z_to_quirk_op(ops.Z**gate.exponent).controlled()
 
 
-def cnot_to_quirk_op(gate: ops.CXPowGate) -> Optional[QuirkOp]:
+def cnot_to_quirk_op(gate: ops.CXPowGate) -> QuirkOp | None:
     return x_to_quirk_op(ops.X**gate.exponent).controlled()
 
 
-def h_to_quirk_op(gate: ops.HPowGate) -> Optional[QuirkOp]:
+def h_to_quirk_op(gate: ops.HPowGate) -> QuirkOp | None:
     if gate.exponent == 1:
         return QuirkOp('H')
     return None
 
 
-def swap_to_quirk_op(gate: ops.SwapPowGate) -> Optional[QuirkOp]:
+def swap_to_quirk_op(gate: ops.SwapPowGate) -> QuirkOp | None:
     if gate.exponent == 1:
         return QuirkOp('Swap', 'Swap', can_merge=False)
     return None
 
 
-def cswap_to_quirk_op(gate: ops.CSwapGate) -> Optional[QuirkOp]:
+def cswap_to_quirk_op(gate: ops.CSwapGate) -> QuirkOp | None:
     return QuirkOp('•', 'Swap', 'Swap', can_merge=False)
 
 
-def ccx_to_quirk_op(gate: ops.CCXPowGate) -> Optional[QuirkOp]:
+def ccx_to_quirk_op(gate: ops.CCXPowGate) -> QuirkOp | None:
     e = angle_to_exponent_key(gate.exponent)
     if e is None:
         return None
     return QuirkOp('•', '•', 'X' + e, can_merge=False)
 
 
-def ccz_to_quirk_op(gate: ops.CCZPowGate) -> Optional[QuirkOp]:
+def ccz_to_quirk_op(gate: ops.CCZPowGate) -> QuirkOp | None:
     e = angle_to_exponent_key(gate.exponent)
     if e is None:
         return None
     return QuirkOp('•', '•', 'Z' + e, can_merge=False)
 
 
-def controlled_unwrap(op: ops.ControlledOperation) -> Optional[QuirkOp]:
+def controlled_unwrap(op: ops.ControlledOperation) -> QuirkOp | None:
     sub = known_quirk_op_for_operation(op.sub_operation)
     if sub is None:
         return None
@@ -222,7 +223,7 @@ def controlled_unwrap(op: ops.ControlledOperation) -> Optional[QuirkOp]:
 
 
 _known_gate_conversions = cast(
-    Dict[type, Callable[[ops.Gate], Optional[QuirkOp]]],
+    dict[type, Callable[[ops.Gate], QuirkOp | None]],
     {
         ops.CCXPowGate: ccx_to_quirk_op,
         ops.CCZPowGate: ccz_to_quirk_op,

cirq/contrib/routing/device.py

@@ -12,8 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import itertools
-from typing import Dict, Iterable, Tuple
+from typing import Iterable
 
 import networkx as nx
 
@@ -45,7 +47,7 @@ def _manhattan_distance(qubit1: cirq.GridQubit, qubit2: cirq.GridQubit) -> int:
     return abs(qubit1.row - qubit2.row) + abs(qubit1.col - qubit2.col)
 
 
-def nx_qubit_layout(graph: nx.Graph) -> Dict[cirq.Qid, Tuple[float, float]]:
+def nx_qubit_layout(graph: nx.Graph) -> dict[cirq.Qid, tuple[float, float]]:
     """Return a layout for a graph for nodes which are qubits.
 
     This can be used in place of nx.spring_layout or other networkx layouts.
@@ -62,7 +64,7 @@ def nx_qubit_layout(graph: nx.Graph) -> Dict[cirq.Qid, Tuple[float, float]]:
     >>> nx.draw_networkx(g, pos=pos)
 
     """
-    pos: Dict[cirq.Qid, Tuple[float, float]] = {}
+    pos: dict[cirq.Qid, tuple[float, float]] = {}
 
     _node_to_i_cache = None
     for node in graph.nodes:

cirq/contrib/routing/device_test.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import networkx as nx
 import pytest