change type annotation from float to TParamVal for theta and phi

cirq-core/cirq/ops/diagonal_gate.py

@@ -144,7 +144,7 @@ def _value_equality_values_(self) -> Any:
         return tuple(self._diag_angles_radians)
 
     def _decompose_for_basis(
-        self, index: int, bit_flip: int, theta: float, qubits: Sequence['cirq.Qid']
+        self, index: int, bit_flip: int, theta: value.TParamVal, qubits: Sequence['cirq.Qid']
     ) -> Iterator[Union['cirq.ZPowGate', 'cirq.CXPowGate']]:
         if index == 0:
             return []

cirq-core/cirq/ops/fsim_gate.py

@@ -78,7 +78,7 @@ class FSimGate(gate_features.TwoQubitGate, gate_features.InterchangeableQubitsGa
         FSimGate(θ, φ) = ISWAP**(-2θ/π) CZPowGate(exponent=-φ/π)
     """
 
-    def __init__(self, theta: float, phi: float) -> None:
+    def __init__(self, theta: value.TParamVal, phi: value.TParamVal) -> None:
         """
         Args:
             theta: Swap angle on the ``|01⟩`` ``|10⟩`` subspace, in radians.

cirq-google/cirq_google/calibration/workflow_test.py

@@ -1181,7 +1181,7 @@ def test_run_floquet_characterization_for_moments():
     itertools.product([0.1, 0.7], [-0.3, 0.1, 0.5], [-0.3, 0.2, 0.4], [-0.6, 0.1, 0.6], [0.2, 0.6]),
 )
 def test_fsim_phase_corrections(
-    theta: float, zeta: float, chi: float, gamma: float, phi: float
+    theta: cirq.value.TParamVal, zeta: float, chi: float, gamma: float, phi: cirq.value.TParamVal
 ) -> None:
     a, b = cirq.LineQubit.range(2)
 
@@ -1209,7 +1209,7 @@ def test_fsim_phase_corrections(
     ),
 )
 def test_phase_corrected_fsim_operations_with_phase_exponent(
-    theta: float, zeta: float, chi: float, gamma: float, phi: float
+    theta: cirq.value.TParamVal, zeta: float, chi: float, gamma: float, phi: cirq.value.TParamVal
 ) -> None:
     a, b = cirq.LineQubit.range(2)
 

cirq-google/cirq_google/common_serializers.py

@@ -513,7 +513,7 @@ def _add_phase_match(op: cirq.Operation, proto: v2.program_pb2.Operation):
 # FSim serializer
 # Only allows iswap, sqrt_iswap and their inverses, iswap, CZ, identity, and sycamore
 # Note that not all combinations may not be available on all processors
-def _can_serialize_limited_fsim(theta: float, phi: float):
+def _can_serialize_limited_fsim(theta: cirq.value.TParamVal, phi: cirq.value.TParamVal):
     # Symbols for LIMITED_FSIM are allowed, but may fail server-side
     # if an incorrect run context is specified
     if _near_mod_2pi(phi, 0) or isinstance(phi, sympy.Symbol):

cirq-google/cirq_google/optimizers/convert_to_sycamore_gates.py

@@ -256,7 +256,7 @@ def decompose_phased_iswap_into_syc(
 
 
 def decompose_phased_iswap_into_syc_precomputed(
-    theta: float, a: cirq.Qid, b: cirq.Qid
+    theta: cirq.value.TParamVal, a: cirq.Qid, b: cirq.Qid
 ) -> cirq.OP_TREE:
     """Decompose PhasedISwap into sycamore gates using precomputed coefficients.
 
@@ -466,7 +466,7 @@ def _phased_x_z_ops(mat: np.ndarray, q: cirq.Qid) -> Iterator[cirq.Operation]:
         yield gate(q)
 
 
-def rzz(theta: float, q0: cirq.Qid, q1: cirq.Qid) -> cirq.OP_TREE:
+def rzz(theta: cirq.value.TParamVal, q0: cirq.Qid, q1: cirq.Qid) -> cirq.OP_TREE:
     """Generate exp(-1j * theta * zz) from Sycamore gates.
 
     Args:
@@ -495,7 +495,7 @@ def rzz(theta: float, q0: cirq.Qid, q1: cirq.Qid) -> cirq.OP_TREE:
     yield create_corrected_circuit(target_unitary, program, q0, q1)
 
 
-def cphase(theta: float, q0: cirq.Qid, q1: cirq.Qid) -> cirq.OP_TREE:
+def cphase(theta: cirq.value.TParamVal, q0: cirq.Qid, q1: cirq.Qid) -> cirq.OP_TREE:
     """
     Implement a cphase using the Ising gate generated from 2 Sycamore gates
 
@@ -515,7 +515,7 @@ def cphase(theta: float, q0: cirq.Qid, q1: cirq.Qid) -> cirq.OP_TREE:
     yield cirq.rz(theta / 2).on(q1)
 
 
-def swap_rzz(theta: float, q0: cirq.Qid, q1: cirq.Qid) -> cirq.OP_TREE:
+def swap_rzz(theta: cirq.value.TParamVal, q0: cirq.Qid, q1: cirq.Qid) -> cirq.OP_TREE:
     """
     An implementation of SWAP * EXP(1j theta ZZ) using three sycamore gates.