cirq-core 1.6.0.dev20250521011052__py3-none-any.whl → 1.6.0.dev20250522221502__py3-none-any.whl

cirq/protocols/control_key_protocol.py

@@ -16,7 +16,7 @@
 from __future__ import annotations
 
 from types import NotImplementedType
-from typing import Any, FrozenSet, TYPE_CHECKING, Union
+from typing import Any, TYPE_CHECKING
 
 from typing_extensions import Protocol
 
@@ -37,7 +37,7 @@ class SupportsControlKey(Protocol):
     """
 
     @doc_private
-    def _control_keys_(self) -> Union[FrozenSet[cirq.MeasurementKey], NotImplementedType, None]:
+    def _control_keys_(self) -> frozenset[cirq.MeasurementKey] | NotImplementedType | None:
         """Return the keys for controls referenced by the receiving object.
 
         Returns:
@@ -46,7 +46,7 @@ class SupportsControlKey(Protocol):
         """
 
 
-def control_keys(val: Any) -> FrozenSet[cirq.MeasurementKey]:
+def control_keys(val: Any) -> frozenset[cirq.MeasurementKey]:
     """Gets the keys that the value is classically controlled by.
 
     Args:
@@ -64,7 +64,7 @@ def control_keys(val: Any) -> FrozenSet[cirq.MeasurementKey]:
     return frozenset()
 
 
-def measurement_keys_touched(val: Any) -> FrozenSet[cirq.MeasurementKey]:
+def measurement_keys_touched(val: Any) -> frozenset[cirq.MeasurementKey]:
     """Returns all the measurement keys used by the value.
 
     This would be the case if the value is or contains a measurement gate, or

cirq/protocols/decompose_protocol_test.py

@@ -15,7 +15,6 @@
 from __future__ import annotations
 
 import itertools
-from typing import Optional
 from unittest import mock
 
 import pytest
@@ -231,7 +230,7 @@ def test_decompose_intercept() -> None:
 
     # Accepts a context, when provided.
     def _intercept_with_context(
-        op: cirq.Operation, context: Optional[cirq.DecompositionContext] = None
+        op: cirq.Operation, context: cirq.DecompositionContext | None = None
     ):
         assert context is not None
         if op.gate == cirq.SWAP:

cirq/protocols/has_stabilizer_effect_protocol.py

@@ -14,7 +14,7 @@
 
 from __future__ import annotations
 
-from typing import Any, Optional
+from typing import Any
 
 import cirq.protocols.decompose_protocol as decompose_protocol
 import cirq.protocols.has_unitary_protocol as has_unitary_protocol
@@ -52,7 +52,7 @@ def has_stabilizer_effect(val: Any) -> bool:
     return False
 
 
-def _strat_has_stabilizer_effect_from_has_stabilizer_effect(val: Any) -> Optional[bool]:
+def _strat_has_stabilizer_effect_from_has_stabilizer_effect(val: Any) -> bool | None:
     """Infer whether val has stabilizer effect via its `_has_stabilizer_effect_` method."""
     if hasattr(val, '_has_stabilizer_effect_'):
         result = val._has_stabilizer_effect_()
@@ -61,14 +61,14 @@ def _strat_has_stabilizer_effect_from_has_stabilizer_effect(val: Any) -> Optiona
     return None
 
 
-def _strat_has_stabilizer_effect_from_gate(val: Any) -> Optional[bool]:
+def _strat_has_stabilizer_effect_from_gate(val: Any) -> bool | None:
     """Infer whether val's gate has stabilizer effect via the _has_stabilizer_effect_ method."""
     if hasattr(val, 'gate'):
         return _strat_has_stabilizer_effect_from_has_stabilizer_effect(val.gate)
     return None
 
 
-def _strat_has_stabilizer_effect_from_unitary(val: Any) -> Optional[bool]:
+def _strat_has_stabilizer_effect_from_unitary(val: Any) -> bool | None:
     """Attempts to infer whether val has stabilizer effect from its unitary.
 
     Returns whether unitary of `val` normalizes the Pauli group. Works only for
@@ -103,7 +103,7 @@ def _strat_has_stabilizer_effect_from_unitary(val: Any) -> Optional[bool]:
     return True
 
 
-def _strat_has_stabilizer_effect_from_decompose(val: Any) -> Optional[bool]:
+def _strat_has_stabilizer_effect_from_decompose(val: Any) -> bool | None:
     decomposition, _, _ = decompose_protocol._try_decompose_into_operations_and_qubits(val)
     if decomposition is None:
         return None

cirq/protocols/has_unitary_protocol.py

@@ -14,7 +14,7 @@
 
 from __future__ import annotations
 
-from typing import Any, Optional, TypeVar
+from typing import Any, TypeVar
 
 import numpy as np
 from typing_extensions import Protocol
@@ -110,7 +110,7 @@ def has_unitary(val: Any, *, allow_decompose: bool = True) -> bool:
     return False
 
 
-def _strat_has_unitary_from_has_unitary(val: Any) -> Optional[bool]:
+def _strat_has_unitary_from_has_unitary(val: Any) -> bool | None:
     """Attempts to infer a value's unitary-ness via its _has_unitary_ method."""
     if hasattr(val, '_has_unitary_'):
         result = val._has_unitary_()
@@ -120,7 +120,7 @@ def _strat_has_unitary_from_has_unitary(val: Any) -> Optional[bool]:
     return None
 
 
-def _strat_has_unitary_from_unitary(val: Any) -> Optional[bool]:
+def _strat_has_unitary_from_unitary(val: Any) -> bool | None:
     """Attempts to infer a value's unitary-ness via its _unitary_ method."""
     getter = getattr(val, '_unitary_', None)
     if getter is None:
@@ -129,7 +129,7 @@ def _strat_has_unitary_from_unitary(val: Any) -> Optional[bool]:
     return result is not NotImplemented and result is not None
 
 
-def _strat_has_unitary_from_decompose(val: Any) -> Optional[bool]:
+def _strat_has_unitary_from_decompose(val: Any) -> bool | None:
     """Attempts to infer a value's unitary-ness via its _decompose_ method."""
     operations, _, _ = _try_decompose_into_operations_and_qubits(val)
     if operations is None:
@@ -137,7 +137,7 @@ def _strat_has_unitary_from_decompose(val: Any) -> Optional[bool]:
     return all(has_unitary(op) for op in operations)
 
 
-def _strat_has_unitary_from_apply_unitary(val: Any) -> Optional[bool]:
+def _strat_has_unitary_from_apply_unitary(val: Any) -> bool | None:
     """Attempts to infer a value's unitary-ness via its _apply_unitary_ method."""
     method = getattr(val, '_apply_unitary_', None)
     if method is None:

cirq/protocols/pauli_expansion_protocol.py

@@ -16,7 +16,7 @@
 
 from __future__ import annotations
 
-from typing import Any, TypeVar, Union
+from typing import Any, TypeVar
 
 from typing_extensions import Protocol
 
@@ -48,9 +48,9 @@ class SupportsPauliExpansion(Protocol):
 def pauli_expansion(
     val: Any,
     *,
-    default: Union[value.LinearDict[str], TDefault] = RaiseTypeErrorIfNotProvided,
+    default: value.LinearDict[str] | TDefault = RaiseTypeErrorIfNotProvided,
     atol: float = 1e-9,
-) -> Union[value.LinearDict[str], TDefault]:
+) -> value.LinearDict[str] | TDefault:
     """Returns coefficients of the expansion of val in the Pauli basis.
 
     Args:

cirq/protocols/pow_protocol.py

@@ -14,7 +14,7 @@
 
 from __future__ import annotations
 
-from typing import Any, Callable, Optional, overload, TYPE_CHECKING, TypeVar, Union
+from typing import Any, Callable, overload, TYPE_CHECKING, TypeVar
 
 if TYPE_CHECKING:
     import cirq
@@ -38,17 +38,17 @@ def pow(val: cirq.Operation, exponent: Any) -> cirq.Operation:
 
 
 @overload
-def pow(val: cirq.Gate, exponent: Any, default: TDefault) -> Union[TDefault, cirq.Gate]:
+def pow(val: cirq.Gate, exponent: Any, default: TDefault) -> TDefault | cirq.Gate:
     pass
 
 
 @overload
-def pow(val: cirq.Operation, exponent: Any, default: TDefault) -> Union[TDefault, cirq.Operation]:
+def pow(val: cirq.Operation, exponent: Any, default: TDefault) -> TDefault | cirq.Operation:
     pass
 
 
 @overload
-def pow(val: cirq.Circuit, exponent: int, default: TDefault) -> Union[TDefault, cirq.Circuit]:
+def pow(val: cirq.Circuit, exponent: int, default: TDefault) -> TDefault | cirq.Circuit:
     pass
 
 
@@ -84,7 +84,7 @@ def pow(val: Any, exponent: Any, default: Any = RaiseTypeErrorIfNotProvided) ->
         TypeError: `val` doesn't have a __pow__ method (or that method returned
             NotImplemented) and no `default` value was specified.
     """
-    raiser: Optional[Callable] = getattr(val, '__pow__', None)
+    raiser: Callable | None = getattr(val, '__pow__', None)
     result = NotImplemented if raiser is None else raiser(exponent)
     if result is not NotImplemented:
         return result

cirq/protocols/trace_distance_bound.py

@@ -14,7 +14,7 @@
 
 from __future__ import annotations
 
-from typing import Any, Optional, Sequence, TypeVar, Union
+from typing import Any, Sequence, TypeVar
 
 import numpy as np
 from typing_extensions import Protocol
@@ -81,7 +81,7 @@ def trace_distance_bound(val: Any) -> float:
     return 1.0
 
 
-def _strat_from_trace_distance_bound_method(val: Any) -> Optional[float]:
+def _strat_from_trace_distance_bound_method(val: Any) -> float | None:
     """Attempts to use a specialized method."""
     getter = getattr(val, '_trace_distance_bound_', None)
     result = NotImplemented if getter is None else getter()
@@ -95,7 +95,7 @@ def _strat_from_trace_distance_bound_method(val: Any) -> Optional[float]:
     return NotImplemented
 
 
-def _strat_distance_from_unitary(val: Any) -> Optional[float]:
+def _strat_distance_from_unitary(val: Any) -> float | None:
     """Attempts to compute a value's trace_distance_bound from its unitary."""
     u = unitary_protocol.unitary(val, default=None)
 
@@ -111,7 +111,7 @@ def _strat_distance_from_unitary(val: Any) -> Optional[float]:
     return trace_distance_from_angle_list(np.angle(np.linalg.eigvals(u)))
 
 
-def trace_distance_from_angle_list(angle_list: Union[Sequence[float], np.ndarray]) -> float:
+def trace_distance_from_angle_list(angle_list: Sequence[float] | np.ndarray) -> float:
     """Given a list of arguments of the eigenvalues of a unitary matrix,
     calculates the trace distance bound of the unitary effect.
 

cirq/protocols/unitary_protocol.py

@@ -15,7 +15,7 @@
 from __future__ import annotations
 
 from types import NotImplementedType
-from typing import Any, Optional, TypeVar, Union
+from typing import Any, TypeVar
 
 import numpy as np
 from typing_extensions import Protocol
@@ -39,7 +39,7 @@ class SupportsUnitary(Protocol):
     """An object that may be describable by a unitary matrix."""
 
     @doc_private
-    def _unitary_(self) -> Union[np.ndarray, NotImplementedType]:
+    def _unitary_(self) -> np.ndarray | NotImplementedType:
         """A unitary matrix describing this value, e.g. the matrix of a gate.
 
         This method is used by the global `cirq.unitary` method. If this method
@@ -78,8 +78,8 @@ class SupportsUnitary(Protocol):
 
 
 def unitary(
-    val: Any, default: Union[np.ndarray, TDefault] = RaiseTypeErrorIfNotProvided
-) -> Union[np.ndarray, TDefault]:
+    val: Any, default: np.ndarray | TDefault = RaiseTypeErrorIfNotProvided
+) -> np.ndarray | TDefault:
     """Returns a unitary matrix describing the given value.
 
     The matrix is determined by any one of the following techniques:
@@ -142,7 +142,7 @@ def unitary(
     )
 
 
-def _strat_unitary_from_unitary(val: Any) -> Optional[np.ndarray]:
+def _strat_unitary_from_unitary(val: Any) -> np.ndarray | None:
     """Attempts to compute a value's unitary via its _unitary_ method."""
     getter = getattr(val, '_unitary_', None)
     if getter is None:
@@ -150,7 +150,7 @@ def _strat_unitary_from_unitary(val: Any) -> Optional[np.ndarray]:
     return getter()
 
 
-def _strat_unitary_from_apply_unitary(val: Any) -> Optional[np.ndarray]:
+def _strat_unitary_from_apply_unitary(val: Any) -> np.ndarray | None:
     """Attempts to compute a value's unitary via its _apply_unitary_ method."""
     # Check for the magic method.
     method = getattr(val, '_apply_unitary_', None)
@@ -171,7 +171,7 @@ def _strat_unitary_from_apply_unitary(val: Any) -> Optional[np.ndarray]:
     return result.reshape((state_len, state_len))
 
 
-def _strat_unitary_from_decompose(val: Any) -> Optional[np.ndarray]:
+def _strat_unitary_from_decompose(val: Any) -> np.ndarray | None:
     """Attempts to compute a value's unitary via its _decompose_ method."""
     # Check if there's a decomposition.
     operations, qubits, val_qid_shape = _try_decompose_into_operations_and_qubits(val)

cirq/protocols/unitary_protocol_test.py

@@ -14,8 +14,6 @@
 
 from __future__ import annotations
 
-from typing import Optional
-
 import numpy as np
 import pytest
 
@@ -77,7 +75,7 @@ class FullyImplemented(cirq.Gate):
     def _has_unitary_(self) -> bool:
         return self.unitary_value
 
-    def _unitary_(self) -> Optional[np.ndarray]:
+    def _unitary_(self) -> np.ndarray | None:
         if not self.unitary_value:
             return None
         return m1

cirq/qis/entropy.py

@@ -17,7 +17,7 @@ from __future__ import annotations
 from collections.abc import Sequence
 from concurrent.futures import ThreadPoolExecutor
 from itertools import product
-from typing import Any, cast, Iterator, Optional
+from typing import Any, cast, Iterator
 
 import numpy as np
 import numpy.typing as npt
@@ -94,7 +94,7 @@ def _compute_bitstrings_contribution_to_purity(bitstrings: npt.NDArray[np.int8])
 def process_renyi_entropy_from_bitstrings(
     measured_bitstrings: npt.NDArray[np.int8],
     subsystem: tuple[int] | None = None,
-    pool: Optional[ThreadPoolExecutor] = None,
+    pool: ThreadPoolExecutor | None = None,
 ) -> float:
     """Compute the Rényi entropy of an array of bitstrings.
     Args:

cirq/sim/clifford/clifford_tableau_simulation_state.py

@@ -17,7 +17,7 @@ for Clifford Simulator."""
 
 from __future__ import annotations
 
-from typing import Optional, Sequence, TYPE_CHECKING
+from typing import Sequence, TYPE_CHECKING
 
 from cirq.qis import clifford_tableau
 from cirq.sim.clifford.stabilizer_simulation_state import StabilizerSimulationState
@@ -34,9 +34,9 @@ class CliffordTableauSimulationState(StabilizerSimulationState[clifford_tableau.
     def __init__(
         self,
         tableau: cirq.CliffordTableau,
-        prng: Optional[np.random.RandomState] = None,
-        qubits: Optional[Sequence[cirq.Qid]] = None,
-        classical_data: Optional[cirq.ClassicalDataStore] = None,
+        prng: np.random.RandomState | None = None,
+        qubits: Sequence[cirq.Qid] | None = None,
+        classical_data: cirq.ClassicalDataStore | None = None,
     ):
         """Inits CliffordTableauSimulationState.
 

cirq/sim/clifford/stabilizer_ch_form_simulation_state.py

@@ -14,7 +14,7 @@
 
 from __future__ import annotations
 
-from typing import Optional, Sequence, TYPE_CHECKING, Union
+from typing import Sequence, TYPE_CHECKING
 
 from cirq._compat import proper_repr
 from cirq.sim.clifford import stabilizer_state_ch_form
@@ -34,10 +34,10 @@ class StabilizerChFormSimulationState(
     def __init__(
         self,
         *,
-        prng: Optional[np.random.RandomState] = None,
-        qubits: Optional[Sequence[cirq.Qid]] = None,
-        initial_state: Union[int, cirq.StabilizerStateChForm] = 0,
-        classical_data: Optional[cirq.ClassicalDataStore] = None,
+        prng: np.random.RandomState | None = None,
+        qubits: Sequence[cirq.Qid] | None = None,
+        initial_state: int | cirq.StabilizerStateChForm = 0,
+        classical_data: cirq.ClassicalDataStore | None = None,
     ):
         """Initializes with the given state and the axes for the operation.
 

cirq/sim/clifford/stabilizer_simulation_state.py

@@ -16,7 +16,7 @@ from __future__ import annotations
 
 import abc
 from types import NotImplementedType
-from typing import Any, cast, Generic, Optional, Sequence, TYPE_CHECKING, TypeVar, Union
+from typing import Any, cast, Generic, Sequence, TYPE_CHECKING, TypeVar
 
 import sympy
 
@@ -44,9 +44,9 @@ class StabilizerSimulationState(
         self,
         *,
         state: TStabilizerState,
-        prng: Optional[np.random.RandomState] = None,
-        qubits: Optional[Sequence[cirq.Qid]] = None,
-        classical_data: Optional[cirq.ClassicalDataStore] = None,
+        prng: np.random.RandomState | None = None,
+        qubits: Sequence[cirq.Qid] | None = None,
+        classical_data: cirq.ClassicalDataStore | None = None,
     ):
         """Initializes the StabilizerSimulationState.
 
@@ -69,7 +69,7 @@ class StabilizerSimulationState(
 
     def _act_on_fallback_(
         self, action: Any, qubits: Sequence[cirq.Qid], allow_decompose: bool = True
-    ) -> Union[bool, NotImplementedType]:
+    ) -> bool | NotImplementedType:
         strats = [self._strat_apply_gate, self._strat_apply_mixture]
         if allow_decompose:
             strats.append(self._strat_decompose)