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

cirq/ops/pauli_sum_exponential_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 numpy as np
 import pytest
 import sympy
@@ -22,12 +24,12 @@ import cirq.testing
 q0, q1, q2, q3 = cirq.LineQubit.range(4)
 
 
-def test_raises_for_non_commuting_paulis():
+def test_raises_for_non_commuting_paulis() -> None:
     with pytest.raises(ValueError, match='commuting'):
         cirq.PauliSumExponential(cirq.X(q0) + cirq.Z(q0), np.pi / 2)
 
 
-def test_raises_for_non_hermitian_pauli():
+def test_raises_for_non_hermitian_pauli() -> None:
     with pytest.raises(ValueError, match='hermitian'):
         cirq.PauliSumExponential(cirq.X(q0) + 1j * cirq.Z(q1), np.pi / 2)
 
@@ -46,7 +48,7 @@ def test_raises_for_non_hermitian_pauli():
         ),
     ),
 )
-def test_pauli_sum_exponential_qubits(psum_exp, expected_qubits):
+def test_pauli_sum_exponential_qubits(psum_exp, expected_qubits) -> None:
     assert psum_exp.qubits == expected_qubits
 
 
@@ -67,7 +69,7 @@ def test_pauli_sum_exponential_qubits(psum_exp, expected_qubits):
         ),
     ),
 )
-def test_pauli_sum_exponential_with_qubits(psum_exp, expected_psum_exp):
+def test_pauli_sum_exponential_with_qubits(psum_exp, expected_psum_exp) -> None:
     assert psum_exp.with_qubits(*expected_psum_exp.qubits) == expected_psum_exp
 
 
@@ -79,7 +81,7 @@ def test_pauli_sum_exponential_with_qubits(psum_exp, expected_psum_exp):
         (cirq.X(q0) * cirq.Y(q1) + cirq.Y(q1) * cirq.Z(q3), np.pi),
     ),
 )
-def test_with_parameters_resolved_by(psum, exp):
+def test_with_parameters_resolved_by(psum, exp) -> None:
     psum_exp = cirq.PauliSumExponential(psum, sympy.Symbol("theta"))
     resolver = cirq.ParamResolver({"theta": exp})
     actual = cirq.resolve_parameters(psum_exp, resolver)
@@ -87,7 +89,7 @@ def test_with_parameters_resolved_by(psum, exp):
     assert actual == expected
 
 
-def test_pauli_sum_exponential_parameterized_matrix_raises():
+def test_pauli_sum_exponential_parameterized_matrix_raises() -> None:
     with pytest.raises(ValueError, match='parameterized'):
         cirq.PauliSumExponential(cirq.X(q0) + cirq.Z(q1), sympy.Symbol("theta")).matrix()
 
@@ -102,7 +104,7 @@ def test_pauli_sum_exponential_parameterized_matrix_raises():
         ),
     ),
 )
-def test_pauli_sum_exponential_has_correct_unitary(psum_exp, expected_unitary):
+def test_pauli_sum_exponential_has_correct_unitary(psum_exp, expected_unitary) -> None:
     assert cirq.has_unitary(psum_exp)
     assert np.allclose(cirq.unitary(psum_exp), expected_unitary)
 
@@ -127,7 +129,7 @@ def test_pauli_sum_exponential_has_correct_unitary(psum_exp, expected_unitary):
         ),
     ),
 )
-def test_pauli_sum_exponential_pow(psum_exp, power, expected_psum):
+def test_pauli_sum_exponential_pow(psum_exp, power, expected_psum) -> None:
     assert psum_exp**power == expected_psum
 
 
@@ -138,7 +140,7 @@ def test_pauli_sum_exponential_pow(psum_exp, power, expected_psum):
         (cirq.PauliSumExponential(2j * cirq.X(q0) + 3j * cirq.Z(q1), np.pi / 2)),
     ),
 )
-def test_pauli_sum_exponential_repr(psum_exp):
+def test_pauli_sum_exponential_repr(psum_exp) -> None:
     cirq.testing.assert_equivalent_repr(psum_exp)
 
 
@@ -156,5 +158,5 @@ def test_pauli_sum_exponential_repr(psum_exp):
         ),
     ),
 )
-def test_pauli_sum_exponential_formatting(psum_exp, expected_str):
+def test_pauli_sum_exponential_formatting(psum_exp, expected_str) -> None:
     assert str(psum_exp) == expected_str

cirq/ops/permutation_gate.py

@@ -12,7 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, Dict, Iterator, Sequence, Tuple, TYPE_CHECKING
+from __future__ import annotations
+
+from typing import Any, Iterator, Sequence, TYPE_CHECKING
 
 from cirq import protocols, value
 from cirq.ops import raw_types, swap_gates
@@ -61,7 +63,7 @@ class QubitPermutationGate(raw_types.Gate):
         self._permutation = tuple(permutation)
 
     @property
-    def permutation(self) -> Tuple[int, ...]:
+    def permutation(self) -> tuple[int, ...]:
         return self._permutation
 
     def _value_equality_values_(self):
@@ -73,7 +75,7 @@ class QubitPermutationGate(raw_types.Gate):
     def _has_unitary_(self):
         return True
 
-    def _decompose_(self, qubits: Sequence['cirq.Qid']) -> Iterator['cirq.OP_TREE']:
+    def _decompose_(self, qubits: Sequence[cirq.Qid]) -> Iterator[cirq.OP_TREE]:
         permutation = [p for p in self.permutation]
 
         for i in range(len(permutation)):
@@ -90,7 +92,7 @@ class QubitPermutationGate(raw_types.Gate):
             for idx in cycle[1:]:
                 yield swap_gates.SWAP(qubits[cycle[0]], qubits[idx])
 
-    def _apply_unitary_(self, args: 'cirq.ApplyUnitaryArgs'):
+    def _apply_unitary_(self, args: cirq.ApplyUnitaryArgs):
         # Compute the permutation index list.
         permuted_axes = list(range(len(args.target_tensor.shape)))
         for i in range(len(args.axes)):
@@ -104,13 +106,13 @@ class QubitPermutationGate(raw_types.Gate):
         args.available_buffer[...] = args.target_tensor.transpose(permuted_axes)
         return args.available_buffer
 
-    def _circuit_diagram_info_(self, args: 'cirq.CircuitDiagramInfoArgs') -> Tuple[str, ...]:
+    def _circuit_diagram_info_(self, args: cirq.CircuitDiagramInfoArgs) -> tuple[str, ...]:
         return tuple(f'[{i}>{self.permutation[i]}]' for i in range(len(self.permutation)))
 
     def __repr__(self) -> str:
         return f'cirq.QubitPermutationGate(permutation={self.permutation!r})'
 
-    def _json_dict_(self) -> Dict[str, Any]:
+    def _json_dict_(self) -> dict[str, Any]:
         return protocols.obj_to_dict_helper(self, attribute_names=['permutation'])
 
     @classmethod

cirq/ops/permutation_gate_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 numpy as np
 import pytest
 
@@ -19,7 +21,7 @@ import cirq
 from cirq.ops import QubitPermutationGate
 
 
-def test_permutation_gate_equality():
+def test_permutation_gate_equality() -> None:
     eq = cirq.testing.EqualsTester()
     eq.make_equality_group(
         lambda: QubitPermutationGate([0, 1]), lambda: QubitPermutationGate((0, 1))
@@ -27,7 +29,7 @@ def test_permutation_gate_equality():
     eq.add_equality_group(QubitPermutationGate([1, 0]), QubitPermutationGate((1, 0)))
 
 
-def test_permutation_gate_repr():
+def test_permutation_gate_repr() -> None:
     cirq.testing.assert_equivalent_repr(QubitPermutationGate([0, 1]))
 
 
@@ -35,26 +37,26 @@ rs = np.random.RandomState(seed=1234)
 
 
 @pytest.mark.parametrize('permutation', [rs.permutation(i) for i in range(3, 7)])
-def test_permutation_gate_consistent_protocols(permutation):
+def test_permutation_gate_consistent_protocols(permutation) -> None:
     gate = QubitPermutationGate(list(permutation))
     cirq.testing.assert_implements_consistent_protocols(gate)
 
 
-def test_permutation_gate_invalid_indices():
+def test_permutation_gate_invalid_indices() -> None:
     with pytest.raises(ValueError, match="Invalid indices"):
         QubitPermutationGate([1, 0, 2, 4])
     with pytest.raises(ValueError, match="Invalid indices"):
         QubitPermutationGate([-1])
 
 
-def test_permutation_gate_invalid_permutation():
+def test_permutation_gate_invalid_permutation() -> None:
     with pytest.raises(ValueError, match="Invalid permutation"):
         QubitPermutationGate([1, 1])
     with pytest.raises(ValueError, match="Invalid permutation"):
         QubitPermutationGate([])
 
 
-def test_permutation_gate_diagram():
+def test_permutation_gate_diagram() -> None:
     q = cirq.LineQubit.range(6)
     cirq.testing.assert_has_diagram(
         cirq.Circuit(cirq.X(q[0]), cirq.X(q[5]), QubitPermutationGate([3, 2, 1, 0]).on(*q[1:5])),
@@ -74,7 +76,7 @@ def test_permutation_gate_diagram():
     )
 
 
-def test_permutation_gate_json_dict():
+def test_permutation_gate_json_dict() -> None:
     assert cirq.QubitPermutationGate([0, 1, 2])._json_dict_() == {'permutation': (0, 1, 2)}
 
 
@@ -96,7 +98,7 @@ def test_permutation_gate_json_dict():
         ],
     ],
 )
-def test_permutation_gate_maps(maps, permutation):
+def test_permutation_gate_maps(maps, permutation) -> None:
     qs = cirq.LineQubit.range(len(permutation))
     permutationOp = cirq.QubitPermutationGate(permutation).on(*qs)
     circuit = cirq.Circuit(permutationOp)

cirq/ops/phased_iswap_gate.py

@@ -13,7 +13,9 @@
 # limitations under the License.
 """ISWAPPowGate conjugated by tensor product Rz(phi) and Rz(-phi)."""
 
-from typing import AbstractSet, Any, cast, Dict, Iterator, List, Optional, Sequence, Tuple, Union
+from __future__ import annotations
+
+from typing import AbstractSet, Any, cast, Iterator, Sequence
 
 import numpy as np
 import sympy
@@ -60,8 +62,8 @@ class PhasedISwapPowGate(eigen_gate.EigenGate):
     def __init__(
         self,
         *,
-        phase_exponent: Union[float, sympy.Expr] = 0.25,
-        exponent: Union[float, sympy.Expr] = 1.0,
+        phase_exponent: float | sympy.Expr = 0.25,
+        exponent: float | sympy.Expr = 1.0,
         global_shift: float = 0.0,
     ):
         """Inits PhasedISwapPowGate.
@@ -79,13 +81,13 @@ class PhasedISwapPowGate(eigen_gate.EigenGate):
         super().__init__(exponent=exponent, global_shift=global_shift)
 
     @property
-    def phase_exponent(self) -> Union[float, sympy.Expr]:
+    def phase_exponent(self) -> float | sympy.Expr:
         return self._phase_exponent
 
     def _num_qubits_(self) -> int:
         return 2
 
-    def _json_dict_(self) -> Dict[str, Any]:
+    def _json_dict_(self) -> dict[str, Any]:
         return {
             'phase_exponent': self._phase_exponent,
             'exponent': self._exponent,
@@ -112,32 +114,32 @@ class PhasedISwapPowGate(eigen_gate.EigenGate):
         )
 
     def _resolve_parameters_(
-        self, resolver: 'cirq.ParamResolver', recursive: bool
-    ) -> 'PhasedISwapPowGate':
+        self, resolver: cirq.ParamResolver, recursive: bool
+    ) -> PhasedISwapPowGate:
         return self.__class__(
             phase_exponent=protocols.resolve_parameters(self.phase_exponent, resolver, recursive),
             exponent=protocols.resolve_parameters(self.exponent, resolver, recursive),
         )
 
-    def _with_exponent(self, exponent: value.type_alias.TParamVal) -> 'PhasedISwapPowGate':
+    def _with_exponent(self, exponent: value.type_alias.TParamVal) -> PhasedISwapPowGate:
         return PhasedISwapPowGate(
             phase_exponent=self.phase_exponent, exponent=exponent, global_shift=self.global_shift
         )
 
-    def _eigen_shifts(self) -> List[float]:
+    def _eigen_shifts(self) -> list[float]:
         return [0.0, +0.5, -0.5]
 
-    def _eigen_components(self) -> List[Tuple[float, np.ndarray]]:
+    def _eigen_components(self) -> list[tuple[float, np.ndarray]]:
         phase = np.exp(1j * np.pi * self.phase_exponent)
         phase_matrix = np.diag([1, phase, phase.conjugate(), 1])
         inverse_phase_matrix = np.conjugate(phase_matrix)
-        eigen_components: List[Tuple[float, np.ndarray]] = []
+        eigen_components: list[tuple[float, np.ndarray]] = []
         for eigenvalue, projector in self._iswap._eigen_components():
             new_projector = phase_matrix @ projector @ inverse_phase_matrix
             eigen_components.append((eigenvalue, new_projector))
         return eigen_components
 
-    def _apply_unitary_(self, args: 'protocols.ApplyUnitaryArgs') -> Optional[np.ndarray]:
+    def _apply_unitary_(self, args: protocols.ApplyUnitaryArgs) -> np.ndarray | None:
         if protocols.is_parameterized(self):
             return NotImplemented
 
@@ -156,7 +158,7 @@ class PhasedISwapPowGate(eigen_gate.EigenGate):
         )
         return args.available_buffer
 
-    def _decompose_(self, qubits: Sequence['cirq.Qid']) -> Iterator['cirq.OP_TREE']:
+    def _decompose_(self, qubits: Sequence[cirq.Qid]) -> Iterator[cirq.OP_TREE]:
         if len(qubits) != 2:
             raise ValueError(f'Expected two qubits, got {len(qubits)}')
         a, b = qubits
@@ -189,9 +191,7 @@ class PhasedISwapPowGate(eigen_gate.EigenGate):
             }
         )
 
-    def _circuit_diagram_info_(
-        self, args: 'cirq.CircuitDiagramInfoArgs'
-    ) -> 'cirq.CircuitDiagramInfo':
+    def _circuit_diagram_info_(self, args: cirq.CircuitDiagramInfoArgs) -> cirq.CircuitDiagramInfo:
         s = f'PhISwap({args.format_real(self._phase_exponent)})'
         return protocols.CircuitDiagramInfo(
             wire_symbols=(s, s), exponent=self._diagram_exponent(args)

cirq/ops/phased_iswap_gate_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 numpy as np
 import pytest
 import scipy
@@ -22,7 +24,7 @@ import cirq
 np.set_printoptions(linewidth=300)
 
 
-def test_phased_iswap_init():
+def test_phased_iswap_init() -> None:
     p = -0.25
     t = 0.75
     s = 0.5
@@ -32,7 +34,7 @@ def test_phased_iswap_init():
     assert gate.global_shift == s
 
 
-def test_phased_iswap_equality():
+def test_phased_iswap_equality() -> None:
     eq = cirq.testing.EqualsTester()
     eq.add_equality_group(cirq.PhasedISwapPowGate(phase_exponent=0, exponent=0.4))
     eq.add_equality_group(cirq.ISWAP**0.4)
@@ -40,7 +42,7 @@ def test_phased_iswap_equality():
     eq.add_equality_group(cirq.ISwapPowGate(global_shift=0.3) ** 0.4)
 
 
-def test_repr():
+def test_repr() -> None:
     p = -0.25
     t = 0.75
     s = 0.3
@@ -48,7 +50,7 @@ def test_repr():
     cirq.testing.assert_equivalent_repr(gate)
 
 
-def test_phased_iswap_unitary():
+def test_phased_iswap_unitary() -> None:
     p = 0.3
     t = 0.4
     actual = cirq.unitary(cirq.PhasedISwapPowGate(phase_exponent=p, exponent=t))
@@ -64,7 +66,7 @@ def test_phased_iswap_unitary():
     assert np.allclose(actual, expected)
 
 
-def test_phased_iswap_equivalent_circuit():
+def test_phased_iswap_equivalent_circuit() -> None:
     p = 0.7
     t = -0.4
     gate = cirq.PhasedISwapPowGate(phase_exponent=p, exponent=t)
@@ -81,7 +83,7 @@ def test_phased_iswap_equivalent_circuit():
     assert np.allclose(cirq.unitary(gate), cirq.unitary(equivalent_circuit))
 
 
-def test_phased_iswap_str():
+def test_phased_iswap_str() -> None:
     assert str(cirq.PhasedISwapPowGate(exponent=1)) == 'PhasedISWAP'
     assert str(cirq.PhasedISwapPowGate(exponent=0.5)) == 'PhasedISWAP**0.5'
     assert (
@@ -90,7 +92,7 @@ def test_phased_iswap_str():
     )
 
 
-def test_phased_iswap_pow():
+def test_phased_iswap_pow() -> None:
     gate1 = cirq.PhasedISwapPowGate(phase_exponent=0.1, exponent=0.25)
     gate2 = cirq.PhasedISwapPowGate(phase_exponent=0.1, exponent=0.5)
     assert gate1**2 == gate2
@@ -108,7 +110,7 @@ def test_phased_iswap_pow():
     assert np.allclose(u1 @ u1, u2)
 
 
-def test_decompose_invalid_qubits():
+def test_decompose_invalid_qubits() -> None:
     qs = cirq.LineQubit.range(3)
     with pytest.raises(ValueError):
         cirq.protocols.decompose_once_with_qubits(cirq.PhasedISwapPowGate(), qs)
@@ -134,7 +136,7 @@ def test_decompose_invalid_qubits():
         (sympy.Symbol('t'), sympy.Symbol('p'), 1),
     ],
 )
-def test_phased_iswap_has_consistent_protocols(phase_exponent, exponent, global_shift):
+def test_phased_iswap_has_consistent_protocols(phase_exponent, exponent, global_shift) -> None:
     cirq.testing.assert_implements_consistent_protocols(
         cirq.PhasedISwapPowGate(
             phase_exponent=phase_exponent, exponent=exponent, global_shift=global_shift
@@ -142,7 +144,7 @@ def test_phased_iswap_has_consistent_protocols(phase_exponent, exponent, global_
     )
 
 
-def test_diagram():
+def test_diagram() -> None:
     q0, q1 = cirq.LineQubit.range(2)
     c = cirq.Circuit(
         cirq.PhasedISwapPowGate(phase_exponent=sympy.Symbol('p'), exponent=sympy.Symbol('t')).on(
@@ -165,7 +167,7 @@ def test_diagram():
 
 
 @pytest.mark.parametrize('angle_rads', (-np.pi, -np.pi / 3, -0.1, np.pi / 5))
-def test_givens_rotation_unitary(angle_rads):
+def test_givens_rotation_unitary(angle_rads) -> None:
     actual = cirq.unitary(cirq.givens(angle_rads))
     c = np.cos(angle_rads)
     s = np.sin(angle_rads)
@@ -179,7 +181,7 @@ def test_givens_rotation_unitary(angle_rads):
 
 
 @pytest.mark.parametrize('angle_rads', (-2 * np.pi / 3, -0.2, 0.4, np.pi / 4))
-def test_givens_rotation_hamiltonian(angle_rads):
+def test_givens_rotation_hamiltonian(angle_rads) -> None:
     actual = cirq.unitary(cirq.givens(angle_rads))
     x = np.array([[0, 1], [1, 0]])
     y = np.array([[0, -1j], [1j, 0]])
@@ -189,7 +191,7 @@ def test_givens_rotation_hamiltonian(angle_rads):
     assert np.allclose(actual, expected)
 
 
-def test_givens_rotation_equivalent_circuit():
+def test_givens_rotation_equivalent_circuit() -> None:
     angle_rads = 3 * np.pi / 7
     t = 2 * angle_rads / np.pi
     gate = cirq.givens(angle_rads)
@@ -201,11 +203,11 @@ def test_givens_rotation_equivalent_circuit():
 
 
 @pytest.mark.parametrize('angle_rads', (-np.pi / 5, 0.4, 2, np.pi))
-def test_givens_rotation_has_consistent_protocols(angle_rads):
+def test_givens_rotation_has_consistent_protocols(angle_rads) -> None:
     cirq.testing.assert_implements_consistent_protocols(cirq.givens(angle_rads))
 
 
-def test_approx_eq():
+def test_approx_eq() -> None:
     gate0 = cirq.PhasedISwapPowGate(phase_exponent=0)
     gate1 = cirq.PhasedISwapPowGate(phase_exponent=1e-12)
     gate2 = cirq.PhasedISwapPowGate(phase_exponent=2e-12)

cirq/ops/phased_x_gate.py

@@ -11,12 +11,15 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
 """An `XPowGate` conjugated by `ZPowGate`s."""
 
+from __future__ import annotations
+
 import math
 import numbers
 from types import NotImplementedType
-from typing import AbstractSet, Any, cast, Dict, Optional, Sequence, Tuple, Union
+from typing import AbstractSet, Any, cast, Sequence
 
 import numpy as np
 import sympy
@@ -48,8 +51,8 @@ class PhasedXPowGate(raw_types.Gate):
     def __init__(
         self,
         *,
-        phase_exponent: Union[float, sympy.Expr],
-        exponent: Union[float, sympy.Expr] = 1.0,
+        phase_exponent: float | sympy.Expr,
+        exponent: float | sympy.Expr = 1.0,
         global_shift: float = 0.0,
     ) -> None:
         """Inits PhasedXPowGate.
@@ -64,7 +67,7 @@ class PhasedXPowGate(raw_types.Gate):
         self._exponent = exponent
         self._global_shift = global_shift
 
-    def _qasm_(self, args: 'cirq.QasmArgs', qubits: Tuple['cirq.Qid', ...]) -> Optional[str]:
+    def _qasm_(self, args: cirq.QasmArgs, qubits: tuple[cirq.Qid, ...]) -> str | None:
         if cirq.is_parameterized(self):
             return None
 
@@ -92,7 +95,7 @@ class PhasedXPowGate(raw_types.Gate):
             qubits[0],
         )
 
-    def _decompose_(self, qubits: Sequence['cirq.Qid']) -> 'cirq.OP_TREE':
+    def _decompose_(self, qubits: Sequence[cirq.Qid]) -> cirq.OP_TREE:
         assert len(qubits) == 1
         q = qubits[0]
         z = cirq.Z(q) ** self._phase_exponent
@@ -100,12 +103,12 @@ class PhasedXPowGate(raw_types.Gate):
         return z**-1, x, z
 
     @property
-    def exponent(self) -> Union[float, sympy.Expr]:
+    def exponent(self) -> float | sympy.Expr:
         """The exponent on the central X gate conjugated by the Z gates."""
         return self._exponent
 
     @property
-    def phase_exponent(self) -> Union[float, sympy.Expr]:
+    def phase_exponent(self) -> float | sympy.Expr:
         """The exponent on the Z gates conjugating the X gate."""
         return self._phase_exponent
 
@@ -113,7 +116,7 @@ class PhasedXPowGate(raw_types.Gate):
     def global_shift(self) -> float:
         return self._global_shift
 
-    def __pow__(self, exponent: Union[float, sympy.Expr]) -> 'PhasedXPowGate':
+    def __pow__(self, exponent: float | sympy.Expr) -> PhasedXPowGate:
         new_exponent = protocols.mul(self._exponent, exponent, NotImplemented)
         if new_exponent is NotImplemented:
             return NotImplemented  # pragma: no cover
@@ -123,7 +126,7 @@ class PhasedXPowGate(raw_types.Gate):
             global_shift=self._global_shift,
         )
 
-    def _trace_distance_bound_(self) -> Optional[float]:
+    def _trace_distance_bound_(self) -> float | None:
         if self._is_parameterized_():
             return None
         return abs(np.sin(self._exponent * 0.5 * np.pi))
@@ -131,7 +134,7 @@ class PhasedXPowGate(raw_types.Gate):
     def _has_unitary_(self):
         return not self._is_parameterized_()
 
-    def _unitary_(self) -> Optional[Union[np.ndarray, NotImplementedType]]:
+    def _unitary_(self) -> np.ndarray | NotImplementedType | None:
         """See `cirq.SupportsUnitary`."""
         if self._is_parameterized_():
             return None
@@ -169,9 +172,7 @@ class PhasedXPowGate(raw_types.Gate):
             self._phase_exponent
         )
 
-    def _resolve_parameters_(
-        self, resolver: 'cirq.ParamResolver', recursive: bool
-    ) -> 'PhasedXPowGate':
+    def _resolve_parameters_(self, resolver: cirq.ParamResolver, recursive: bool) -> PhasedXPowGate:
         """See `cirq.SupportsParameterization`."""
         phase_exponent = resolver.value_of(self._phase_exponent, recursive)
         exponent = resolver.value_of(self._exponent, recursive)
@@ -198,9 +199,7 @@ class PhasedXPowGate(raw_types.Gate):
             global_shift=self._global_shift,
         )
 
-    def _circuit_diagram_info_(
-        self, args: 'cirq.CircuitDiagramInfoArgs'
-    ) -> 'cirq.CircuitDiagramInfo':
+    def _circuit_diagram_info_(self, args: cirq.CircuitDiagramInfoArgs) -> cirq.CircuitDiagramInfo:
         """See `cirq.SupportsCircuitDiagramInfo`."""
 
         return protocols.CircuitDiagramInfo(
@@ -244,5 +243,5 @@ class PhasedXPowGate(raw_types.Gate):
     def _value_equality_values_(self):
         return self.phase_exponent, self._canonical_exponent, self._global_shift
 
-    def _json_dict_(self) -> Dict[str, Any]:
+    def _json_dict_(self) -> dict[str, Any]:
         return protocols.obj_to_dict_helper(self, ['phase_exponent', 'exponent', 'global_shift'])