cirq-core 1.6.0.dev20250520054601__py3-none-any.whl → 1.6.0.dev20250520183459__py3-none-any.whl

cirq/protocols/kraus_protocol.py

@@ -18,7 +18,7 @@ from __future__ import annotations
 
 import warnings
 from types import NotImplementedType
-from typing import Any, Sequence, Tuple, TypeVar, Union
+from typing import Any, Sequence, TypeVar
 
 import numpy as np
 from typing_extensions import Protocol
@@ -33,7 +33,7 @@ from cirq.protocols.unitary_protocol import unitary
 # Sequence[np.ndarray] to ensure the method has the correct type signature in
 # that case. It is checked for using `is`, so it won't have a false positive
 # if the user provides a different (np.array([]),) value.
-RaiseTypeErrorIfNotProvided: Tuple[np.ndarray] = (np.array([]),)
+RaiseTypeErrorIfNotProvided: tuple[np.ndarray] = (np.array([]),)
 
 
 TDefault = TypeVar('TDefault')
@@ -43,7 +43,7 @@ class SupportsKraus(Protocol):
     """An object that may be describable as a quantum channel."""
 
     @doc_private
-    def _kraus_(self) -> Union[Sequence[np.ndarray], NotImplementedType]:
+    def _kraus_(self) -> Sequence[np.ndarray] | NotImplementedType:
         r"""A list of Kraus matrices describing the quantum channel.
 
         These matrices are the terms in the operator sum representation of a
@@ -96,7 +96,7 @@ class SupportsKraus(Protocol):
 
 def kraus(
     val: Any, default: Any = RaiseTypeErrorIfNotProvided
-) -> Union[Tuple[np.ndarray, ...], TDefault]:
+) -> tuple[np.ndarray, ...] | TDefault:
     r"""Returns a list of matrices describing the channel for the given value.
 
     These matrices are the terms in the operator sum representation of

cirq/protocols/kraus_protocol_test.py

@@ -16,14 +16,14 @@
 
 from __future__ import annotations
 
-from typing import Iterable, List, Sequence, Tuple
+from typing import Iterable, Sequence
 
 import numpy as np
 import pytest
 
 import cirq
 
-LOCAL_DEFAULT: List[np.ndarray] = [np.array([])]
+LOCAL_DEFAULT: list[np.ndarray] = [np.array([])]
 
 
 def test_kraus_no_methods() -> None:
@@ -104,7 +104,7 @@ def test_kraus_fallback_to_mixture() -> None:
     m = ((0.3, cirq.unitary(cirq.X)), (0.4, cirq.unitary(cirq.Y)), (0.3, cirq.unitary(cirq.Z)))
 
     class ReturnsMixture:
-        def _mixture_(self) -> Iterable[Tuple[float, np.ndarray]]:
+        def _mixture_(self) -> Iterable[tuple[float, np.ndarray]]:
             return m
 
     c = (

cirq/protocols/measurement_key_protocol.py

@@ -17,7 +17,7 @@
 from __future__ import annotations
 
 from types import NotImplementedType
-from typing import Any, FrozenSet, Mapping, Optional, Tuple, TYPE_CHECKING, Union
+from typing import Any, Mapping, TYPE_CHECKING
 
 from typing_extensions import Protocol
 
@@ -72,9 +72,7 @@ class SupportsMeasurementKey(Protocol):
         """
 
     @doc_private
-    def _measurement_key_objs_(
-        self,
-    ) -> Union[FrozenSet[cirq.MeasurementKey], NotImplementedType, None]:
+    def _measurement_key_objs_(self) -> frozenset[cirq.MeasurementKey] | NotImplementedType | None:
         """Return the key objects for measurements performed by the receiving object.
 
         When a measurement occurs, either on hardware, or in a simulation,
@@ -92,7 +90,7 @@ class SupportsMeasurementKey(Protocol):
         """
 
     @doc_private
-    def _measurement_key_names_(self) -> Union[FrozenSet[str], NotImplementedType, None]:
+    def _measurement_key_names_(self) -> frozenset[str] | NotImplementedType | None:
         """Return the string keys for measurements performed by the receiving object.
 
         When a measurement occurs, either on hardware, or in a simulation,
@@ -178,7 +176,7 @@ def measurement_key_name(val: Any, default: Any = RaiseTypeErrorIfNotProvided):
 
 def _measurement_key_objs_from_magic_methods(
     val: Any,
-) -> Union[FrozenSet[cirq.MeasurementKey], NotImplementedType, None]:
+) -> frozenset[cirq.MeasurementKey] | NotImplementedType | None:
     """Uses the measurement key related magic methods to get the `MeasurementKey`s for this
     object."""
 
@@ -196,7 +194,7 @@ def _measurement_key_objs_from_magic_methods(
 
 def _measurement_key_names_from_magic_methods(
     val: Any,
-) -> Union[FrozenSet[str], NotImplementedType, None]:
+) -> frozenset[str] | NotImplementedType | None:
     """Uses the measurement key related magic methods to get the key strings for this object."""
 
     getter = getattr(val, '_measurement_key_names_', None)
@@ -212,7 +210,7 @@ def _measurement_key_names_from_magic_methods(
     return result
 
 
-def measurement_key_objs(val: Any) -> FrozenSet[cirq.MeasurementKey]:
+def measurement_key_objs(val: Any) -> frozenset[cirq.MeasurementKey]:
     """Gets the measurement key objects of measurements within the given value.
 
     Args:
@@ -231,7 +229,7 @@ def measurement_key_objs(val: Any) -> FrozenSet[cirq.MeasurementKey]:
     return frozenset()
 
 
-def measurement_key_names(val: Any) -> FrozenSet[str]:
+def measurement_key_names(val: Any) -> frozenset[str]:
     """Gets the measurement key strings of measurements within the given value.
 
     Args:
@@ -256,7 +254,7 @@ def measurement_key_names(val: Any) -> FrozenSet[str]:
     return frozenset()
 
 
-def _is_measurement_from_magic_method(val: Any) -> Optional[bool]:
+def _is_measurement_from_magic_method(val: Any) -> bool | None:
     """Uses `is_measurement` magic method to determine if this object is a measurement."""
     getter = getattr(val, '_is_measurement_', None)
     return NotImplemented if getter is None else getter()
@@ -292,7 +290,7 @@ def with_measurement_key_mapping(val: Any, key_map: Mapping[str, str]):
     return NotImplemented if getter is None else getter(key_map)
 
 
-def with_key_path(val: Any, path: Tuple[str, ...]):
+def with_key_path(val: Any, path: tuple[str, ...]):
     """Adds the path to the target's measurement keys.
 
     The path usually refers to an identifier or a list of identifiers from a subcircuit that
@@ -303,7 +301,7 @@ def with_key_path(val: Any, path: Tuple[str, ...]):
     return NotImplemented if getter is None else getter(path)
 
 
-def with_key_path_prefix(val: Any, prefix: Tuple[str, ...]):
+def with_key_path_prefix(val: Any, prefix: tuple[str, ...]):
     """Prefixes the path to the target's measurement keys.
 
     The path usually refers to an identifier or a list of identifiers from a subcircuit that
@@ -319,7 +317,7 @@ def with_key_path_prefix(val: Any, prefix: Tuple[str, ...]):
 
 
 def with_rescoped_keys(
-    val: Any, path: Tuple[str, ...], bindable_keys: Optional[FrozenSet[cirq.MeasurementKey]] = None
+    val: Any, path: tuple[str, ...], bindable_keys: frozenset[cirq.MeasurementKey] | None = None
 ):
     """Rescopes any measurement and control keys to the provided path, given the existing keys.
 

cirq/protocols/mixture_protocol.py

@@ -17,7 +17,7 @@
 from __future__ import annotations
 
 from types import NotImplementedType
-from typing import Any, Sequence, Tuple, Union
+from typing import Any, Sequence
 
 import numpy as np
 from typing_extensions import Protocol
@@ -29,14 +29,14 @@ from cirq.protocols.unitary_protocol import unitary
 
 # This is a special indicator value used by the inverse method to determine
 # whether or not the caller provided a 'default' argument.
-RaiseTypeErrorIfNotProvided: Sequence[Tuple[float, Any]] = ((0.0, []),)
+RaiseTypeErrorIfNotProvided: Sequence[tuple[float, Any]] = ((0.0, []),)
 
 
 class SupportsMixture(Protocol):
     """An object that decomposes into a probability distribution of unitaries."""
 
     @doc_private
-    def _mixture_(self) -> Union[Sequence[Tuple[float, Any]], NotImplementedType]:
+    def _mixture_(self) -> Sequence[tuple[float, Any]] | NotImplementedType:
         """Decompose into a probability distribution of unitaries.
 
         This method is used by the global `cirq.mixture` method.
@@ -67,7 +67,7 @@ class SupportsMixture(Protocol):
 
 def mixture(
     val: Any, default: Any = RaiseTypeErrorIfNotProvided
-) -> Sequence[Tuple[float, np.ndarray]]:
+) -> Sequence[tuple[float, np.ndarray]]:
     """Return a sequence of tuples representing a probabilistic unitary.
 
     A mixture is described by an iterable of tuples of the form

cirq/protocols/qasm.py

@@ -16,7 +16,7 @@ from __future__ import annotations
 
 import string
 from types import NotImplementedType
-from typing import Any, Dict, Iterable, Optional, Tuple, TYPE_CHECKING, TypeVar, Union
+from typing import Any, Iterable, TYPE_CHECKING, TypeVar
 
 from typing_extensions import Protocol
 
@@ -38,9 +38,9 @@ class QasmArgs(string.Formatter):
         self,
         precision: int = 10,
         version: str = '2.0',
-        qubit_id_map: Optional[Dict[cirq.Qid, str]] = None,
-        meas_key_id_map: Optional[Dict[str, str]] = None,
-        meas_key_bitcount: Optional[Dict[str, int]] = None,
+        qubit_id_map: dict[cirq.Qid, str] | None = None,
+        meas_key_id_map: dict[str, str] | None = None,
+        meas_key_bitcount: dict[str, int] | None = None,
     ) -> None:
         """Inits QasmArgs.
 
@@ -99,7 +99,7 @@ class SupportsQasm(Protocol):
     """
 
     @doc_private
-    def _qasm_(self) -> Union[None, NotImplementedType, str]:
+    def _qasm_(self) -> None | NotImplementedType | str:
         pass
 
 
@@ -112,7 +112,7 @@ class SupportsQasmWithArgs(Protocol):
     """
 
     @doc_private
-    def _qasm_(self, args: QasmArgs) -> Union[None, NotImplementedType, str]:
+    def _qasm_(self, args: QasmArgs) -> None | NotImplementedType | str:
         pass
 
 
@@ -125,9 +125,7 @@ class SupportsQasmWithArgsAndQubits(Protocol):
     """
 
     @doc_private
-    def _qasm_(
-        self, qubits: Tuple[cirq.Qid], args: QasmArgs
-    ) -> Union[None, NotImplementedType, str]:
+    def _qasm_(self, qubits: tuple[cirq.Qid], args: QasmArgs) -> None | NotImplementedType | str:
         pass
 
 
@@ -135,10 +133,10 @@ class SupportsQasmWithArgsAndQubits(Protocol):
 def qasm(
     val: Any,
     *,
-    args: Optional[QasmArgs] = None,
-    qubits: Optional[Iterable[cirq.Qid]] = None,
+    args: QasmArgs | None = None,
+    qubits: Iterable[cirq.Qid] | None = None,
     default: TDefault = RaiseTypeErrorIfNotProvided,
-) -> Union[str, TDefault]:
+) -> str | TDefault:
     """Returns QASM code for the given value, if possible.
 
     Different values require different sets of arguments. The general rule of
@@ -171,7 +169,7 @@ def qasm(
     method = getattr(val, '_qasm_', None)
     result = NotImplemented
     if method is not None:
-        kwargs: Dict[str, Any] = {}
+        kwargs: dict[str, Any] = {}
         if args is not None:
             kwargs['args'] = args
         # pylint: disable=not-callable

cirq/protocols/qid_shape_protocol.py

@@ -15,7 +15,7 @@
 from __future__ import annotations
 
 from types import NotImplementedType
-from typing import Any, Sequence, Tuple, TypeVar, Union
+from typing import Any, Sequence, TypeVar
 
 from typing_extensions import Protocol
 
@@ -24,7 +24,7 @@ from cirq._doc import doc_private, document
 
 # This is a special indicator value used by the methods to determine whether or
 # not the caller provided a 'default' argument. It must be of type
-# Tuple[int, ...] to ensure the method has the correct type signature in that
+# tuple[int, ...] to ensure the method has the correct type signature in that
 # case. It is checked for using `is`, so it won't have a false positive if the
 # user provides a different (0,) value.
 RaiseTypeErrorIfNotProvided: Any = (0,)
@@ -39,7 +39,7 @@ class SupportsExplicitQidShape(Protocol):
     number qubits/qudits/qids, each with a specific number of quantum levels."""
 
     @doc_private
-    def _qid_shape_(self) -> Union[Tuple[int, ...], NotImplementedType]:
+    def _qid_shape_(self) -> tuple[int, ...] | NotImplementedType:
         """A tuple specifying the number of quantum levels of each qid this
         object operates on, e.g. (2, 2, 2) for a three-qubit gate.
 
@@ -67,7 +67,7 @@ class SupportsExplicitNumQubits(Protocol):
     number of qubits."""
 
     @document
-    def _num_qubits_(self) -> Union[int, NotImplementedType]:
+    def _num_qubits_(self) -> int | NotImplementedType:
         """The number of qubits, qudits, or qids this object operates on.
 
         This method is used by the global `cirq.num_qubits` method (and by
@@ -82,7 +82,7 @@ class SupportsExplicitNumQubits(Protocol):
 
 def qid_shape(
     val: Any, default: TDefault = RaiseTypeErrorIfNotProvided
-) -> Union[Tuple[int, ...], TDefault]:
+) -> tuple[int, ...] | TDefault:
     """Returns a tuple describing the number of quantum levels of each
     qubit/qudit/qid `val` operates on.
 
@@ -136,9 +136,7 @@ def qid_shape(
     raise TypeError(f"object of type '{type(val)}' has no _num_qubits_ or _qid_shape_ methods.")
 
 
-def num_qubits(
-    val: Any, default: TDefault = RaiseTypeErrorIfNotProvidedInt
-) -> Union[int, TDefault]:
+def num_qubits(val: Any, default: TDefault = RaiseTypeErrorIfNotProvidedInt) -> int | TDefault:
     """Returns the number of qubits, qudits, or qids `val` operates on.
 
     Args:

cirq/qis/clifford_tableau.py

@@ -15,7 +15,7 @@
 from __future__ import annotations
 
 import abc
-from typing import Any, Dict, List, Optional, Sequence, TYPE_CHECKING
+from typing import Any, Sequence, TYPE_CHECKING
 
 import numpy as np
 
@@ -147,9 +147,9 @@ class CliffordTableau(StabilizerState):
         self,
         num_qubits,
         initial_state: int = 0,
-        rs: Optional[np.ndarray] = None,
-        xs: Optional[np.ndarray] = None,
-        zs: Optional[np.ndarray] = None,
+        rs: np.ndarray | None = None,
+        xs: np.ndarray | None = None,
+        zs: np.ndarray | None = None,
     ):
         """Initializes CliffordTableau
         Args:
@@ -166,7 +166,7 @@ class CliffordTableau(StabilizerState):
         self._xs = self._reconstruct_xs(xs)
         self._zs = self._reconstruct_zs(zs)
 
-    def _reconstruct_rs(self, rs: Optional[np.ndarray]) -> np.ndarray:
+    def _reconstruct_rs(self, rs: np.ndarray | None) -> np.ndarray:
         if rs is None:
             new_rs = np.zeros(2 * self.n + 1, dtype=bool)
             for i, val in enumerate(
@@ -185,7 +185,7 @@ class CliffordTableau(StabilizerState):
                 )
         return new_rs
 
-    def _reconstruct_xs(self, xs: Optional[np.ndarray]) -> np.ndarray:
+    def _reconstruct_xs(self, xs: np.ndarray | None) -> np.ndarray:
         if xs is None:
             new_xs = np.zeros((2 * self.n + 1, self.n), dtype=bool)
             for i in range(self.n):
@@ -207,7 +207,7 @@ class CliffordTableau(StabilizerState):
                 )
         return new_xs
 
-    def _reconstruct_zs(self, zs: Optional[np.ndarray]) -> np.ndarray:
+    def _reconstruct_zs(self, zs: np.ndarray | None) -> np.ndarray:
         if zs is None:
             new_zs = np.zeros((2 * self.n + 1, self.n), dtype=bool)
             for i in range(self.n):
@@ -260,7 +260,7 @@ class CliffordTableau(StabilizerState):
         """Returns the 2n * 2n matrix representation of the Clifford tableau."""
         return np.concatenate([self.xs, self.zs], axis=1)
 
-    def _json_dict_(self) -> Dict[str, Any]:
+    def _json_dict_(self) -> dict[str, Any]:
         return protocols.obj_to_dict_helper(self, ['n', 'rs', 'xs', 'zs'])
 
     @classmethod
@@ -511,12 +511,12 @@ class CliffordTableau(StabilizerState):
                 pauli_mask += "I"
         return DensePauliString(pauli_mask, coefficient=coefficient)
 
-    def stabilizers(self) -> List[cirq.DensePauliString]:
+    def stabilizers(self) -> list[cirq.DensePauliString]:
         """Returns the stabilizer generators of the state. These
         are n operators {S_1,S_2,...,S_n} such that S_i |psi> = |psi>"""
         return [self._row_to_dense_pauli(i) for i in range(self.n, 2 * self.n)]
 
-    def destabilizers(self) -> List[cirq.DensePauliString]:
+    def destabilizers(self) -> list[cirq.DensePauliString]:
         """Returns the destabilizer generators of the state. These
         are n operators {S_1,S_2,...,S_n} such that along with the stabilizer
         generators above generate the full Pauli group on n qubits."""
@@ -665,14 +665,14 @@ class CliffordTableau(StabilizerState):
 
     def measure(
         self, axes: Sequence[int], seed: cirq.RANDOM_STATE_OR_SEED_LIKE = None
-    ) -> List[int]:
+    ) -> list[int]:
         return [self._measure(axis, random_state.parse_random_state(seed)) for axis in axes]
 
     @cached_method
     def __hash__(self) -> int:
         return hash(self.matrix().tobytes() + self.rs.tobytes())
 
-    def __getstate__(self) -> Dict[str, Any]:
+    def __getstate__(self) -> dict[str, Any]:
         # clear cached hash value when pickling, see #6674
         state = self.__dict__
         hash_attr = _method_cache_name(self.__hash__)

cirq/qis/measures.py

@@ -16,7 +16,7 @@
 
 from __future__ import annotations
 
-from typing import Optional, Tuple, TYPE_CHECKING
+from typing import TYPE_CHECKING
 
 import numpy as np
 
@@ -44,7 +44,7 @@ def _sqrt_positive_semidefinite_matrix(mat: np.ndarray) -> np.ndarray:
     return vecs @ (np.sqrt(np.abs(eigs)) * vecs).T.conj()
 
 
-def _validate_int_state(state: int, qid_shape: Optional[Tuple[int, ...]]) -> None:
+def _validate_int_state(state: int, qid_shape: tuple[int, ...] | None) -> None:
     if state < 0:
         raise ValueError(
             'Invalid state: A state specified as an integer must be non-negative, '
@@ -60,7 +60,7 @@ def _validate_int_state(state: int, qid_shape: Optional[Tuple[int, ...]]) -> Non
             )
 
 
-def _validate_product_state(state: cirq.ProductState, qid_shape: Optional[Tuple[int, ...]]) -> None:
+def _validate_product_state(state: cirq.ProductState, qid_shape: tuple[int, ...] | None) -> None:
     if qid_shape is not None and qid_shape != (2,) * len(state):
         raise ValueError(
             'Invalid state for given qid shape: '
@@ -72,7 +72,7 @@ def _validate_product_state(state: cirq.ProductState, qid_shape: Optional[Tuple[
 def fidelity(
     state1: cirq.QUANTUM_STATE_LIKE,
     state2: cirq.QUANTUM_STATE_LIKE,
-    qid_shape: Optional[Tuple[int, ...]] = None,
+    qid_shape: tuple[int, ...] | None = None,
     validate: bool = True,
     atol: float = 1e-7,
 ) -> float:
@@ -156,10 +156,10 @@ def fidelity(
 def _numpy_arrays_to_state_vectors_or_density_matrices(
     state1: np.ndarray,
     state2: np.ndarray,
-    qid_shape: Optional[Tuple[int, ...]],
+    qid_shape: tuple[int, ...] | None,
     validate: bool,
     atol: float,
-) -> Tuple[np.ndarray, np.ndarray]:
+) -> tuple[np.ndarray, np.ndarray]:
     if state1.ndim > 2 or (state1.ndim == 2 and state1.shape[0] != state1.shape[1]):
         # State tensor, convert to state vector
         state1 = state1.reshape(-1)
@@ -256,7 +256,7 @@ def _fidelity_state_vectors_or_density_matrices(state1: np.ndarray, state2: np.n
 
 def von_neumann_entropy(
     state: cirq.QUANTUM_STATE_LIKE,
-    qid_shape: Optional[Tuple[int, ...]] = None,
+    qid_shape: tuple[int, ...] | None = None,
     validate: bool = True,
     atol: float = 1e-7,
 ) -> float:

cirq/qis/quantum_state_representation.py

@@ -15,7 +15,7 @@
 from __future__ import annotations
 
 import abc
-from typing import List, Sequence, Tuple, TYPE_CHECKING
+from typing import Sequence, TYPE_CHECKING
 
 import numpy as np
 from typing_extensions import Self
@@ -41,7 +41,7 @@ class QuantumStateRepresentation(metaclass=abc.ABCMeta):
     @abc.abstractmethod
     def measure(
         self, axes: Sequence[int], seed: cirq.RANDOM_STATE_OR_SEED_LIKE = None
-    ) -> List[int]:
+    ) -> list[int]:
         """Measures the state.
 
         Args:
@@ -74,7 +74,7 @@ class QuantumStateRepresentation(metaclass=abc.ABCMeta):
         """Joins two state spaces together."""
         raise NotImplementedError()
 
-    def factor(self, axes: Sequence[int], *, validate=True, atol=1e-07) -> Tuple[Self, Self]:
+    def factor(self, axes: Sequence[int], *, validate=True, atol=1e-07) -> tuple[Self, Self]:
         """Splits two state spaces after a measurement or reset."""
         raise NotImplementedError()