More numpy types (#5683)

12 errors left

Part of #3767

Author: vtomole

cirq-core/cirq/circuits/circuit.py

@@ -30,6 +30,7 @@
     Any,
     Callable,
     Mapping,
+    MutableSequence,
     cast,
     Dict,
     FrozenSet,
@@ -1462,7 +1463,7 @@ def concat_ragged(
 
         # Allocate a buffer large enough to append and prepend all the circuits.
         pad_len = sum(len(c) for c in circuits) - n_acc
-        buffer = np.zeros(shape=pad_len * 2 + n_acc, dtype=object)
+        buffer: MutableSequence['cirq.Moment'] = [cirq.Moment()] * (pad_len * 2 + n_acc)
 
         # Put the initial circuit in the center of the buffer.
         offset = pad_len
@@ -1601,7 +1602,11 @@ def _overlap_collision_time(
 
 
 def _concat_ragged_helper(
-    c1_offset: int, n1: int, buf: np.ndarray, c2: Sequence['cirq.Moment'], align: 'cirq.Alignment'
+    c1_offset: int,
+    n1: int,
+    buf: MutableSequence['cirq.Moment'],
+    c2: Sequence['cirq.Moment'],
+    align: 'cirq.Alignment',
 ) -> Tuple[int, int]:
     n2 = len(c2)
     shift = _overlap_collision_time(buf[c1_offset : c1_offset + n1], c2, align)
@@ -2369,7 +2374,7 @@ def _resolve_parameters_(
         return Circuit(resolved_moments)
 
     @property
-    def moments(self):
+    def moments(self) -> Sequence['cirq.Moment']:
         return self._moments
 
     def with_noise(self, noise: 'cirq.NOISE_MODEL_LIKE') -> 'cirq.Circuit':

cirq-core/cirq/circuits/circuit_operation.py

@@ -589,7 +589,7 @@ def repeat(
                 # As CircuitOperation is immutable, this can safely return the original.
                 return self
 
-            expected_repetition_id_length = abs(repetitions)
+            expected_repetition_id_length: int = np.abs(repetitions)
 
             if repetition_ids is None:
                 if self.use_repetition_ids:

cirq-core/cirq/contrib/quimb/state_vector.py

@@ -76,7 +76,7 @@ def circuit_to_tensors(
 
     for moment in circuit.moments:
         for op in moment.operations:
-            assert op.gate._has_unitary_()
+            assert cirq.has_unitary(op.gate)
             start_inds = [f'i{qubit_frontier[q]}_q{q}' for q in op.qubits]
             for q in op.qubits:
                 qubit_frontier[q] += 1

cirq-core/cirq/ops/pauli_string.py

@@ -744,7 +744,8 @@ def _expectation_from_density_matrix_no_validation(
 
         while any(result.shape):
             result = np.trace(result, axis1=0, axis2=len(result.shape) // 2)
-        return result * self.coefficient
+
+        return float(result * self.coefficient)
 
     def zip_items(
         self, other: 'cirq.PauliString[TKey]'

cirq-core/cirq/protocols/circuit_diagram_info_protocol.py

@@ -274,7 +274,7 @@ def format_radians(self, radians: Union[sympy.Basic, int, float]) -> str:
             return '0'
         if radians == -np.pi:
             return '-' + unit
-        if self.precision is not None:
+        if self.precision is not None and not isinstance(radians, sympy.Basic):
             quantity = self.format_real(radians / np.pi)
             return quantity + unit
         return repr(radians)

cirq-core/cirq/protocols/trace_distance_bound.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from typing import Any, TypeVar, Optional, Sequence
+from typing import Any, TypeVar, Optional, Sequence, Union
 
 import numpy as np
 from typing_extensions import Protocol
@@ -109,7 +109,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: Sequence[float]) -> float:
+def trace_distance_from_angle_list(angle_list: Union[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-core/cirq/qis/clifford_tableau.py

@@ -19,7 +19,7 @@
 from cirq import protocols
 from cirq._compat import proper_repr
 from cirq.qis import quantum_state_representation
-from cirq.value import big_endian_int_to_digits, linear_dict
+from cirq.value import big_endian_int_to_digits, linear_dict, random_state
 
 if TYPE_CHECKING:
     import cirq
@@ -509,7 +509,7 @@ def destabilizers(self) -> List['cirq.DensePauliString']:
         generators above generate the full Pauli group on n qubits."""
         return [self._row_to_dense_pauli(i) for i in range(self.n)]
 
-    def _measure(self, q, prng: np.random.RandomState = np.random) -> int:
+    def _measure(self, q, prng: np.random.RandomState) -> int:
         """Performs a projective measurement on the q'th qubit.
 
         Returns: the result (0 or 1) of the measurement.
@@ -651,6 +651,6 @@ def apply_global_phase(self, coefficient: linear_dict.Scalar):
         pass
 
     def measure(
-        self, axes: Sequence[int], seed: Optional['cirq.RANDOM_STATE_OR_SEED_LIKE'] = None
+        self, axes: Sequence[int], seed: 'cirq.RANDOM_STATE_OR_SEED_LIKE' = None
     ) -> List[int]:
-        return [self._measure(axis, seed) for axis in axes]
+        return [self._measure(axis, random_state.parse_random_state(seed)) for axis in axes]

cirq-core/cirq/sim/clifford/stabilizer_state_ch_form.py

@@ -17,7 +17,7 @@
 
 import cirq
 from cirq import protocols, qis, value
-from cirq.value import big_endian_int_to_digits
+from cirq.value import big_endian_int_to_digits, random_state
 
 
 @value.value_equality
@@ -388,7 +388,7 @@ def apply_global_phase(self, coefficient: value.Scalar):
     def measure(
         self, axes: Sequence[int], seed: 'cirq.RANDOM_STATE_OR_SEED_LIKE' = None
     ) -> List[int]:
-        return [self._measure(axis, seed) for axis in axes]
+        return [self._measure(axis, random_state.parse_random_state(seed)) for axis in axes]
 
 
 def _phase(exponent, global_shift):

cirq-core/cirq/sim/density_matrix_utils.py

@@ -68,7 +68,6 @@ def sample_density_matrix(
         qid_shape = (2,) * num_qubits
     else:
         _validate_density_matrix_qid_shape(density_matrix, qid_shape)
-        num_qubits = len(qid_shape)
     meas_shape = _indices_shape(qid_shape, indices)
 
     if repetitions == 0 or len(indices) == 0:
@@ -139,16 +138,16 @@ def measure_density_matrix(
         qid_shape = (2,) * num_qubits
     else:
         _validate_density_matrix_qid_shape(density_matrix, qid_shape)
-        num_qubits = len(qid_shape)
     meas_shape = _indices_shape(qid_shape, indices)
 
-    arrout: np.ndarray = (
-        np.copy(density_matrix)
-        if out is None
-        else density_matrix
-        if out is density_matrix
-        else (np.copyto(dst=out, src=density_matrix), out)[-1]
-    )
+    arrout: np.ndarray
+    if out is None:
+        arrout = np.copy(density_matrix)
+    elif out is density_matrix:
+        arrout = density_matrix
+    else:
+        np.copyto(dst=out, src=density_matrix)
+        arrout = out
 
     if len(indices) == 0:
         return ([], arrout)

cirq-core/cirq/sim/state_vector_simulation_state.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """Objects and methods for acting efficiently on a state vector."""
-
 from typing import Any, Callable, List, Optional, Sequence, Tuple, Type, TYPE_CHECKING, Union
 
 import numpy as np
@@ -225,13 +224,12 @@ def prepare_into_buffer(k: int):
             e.reshape(shape * 2).astype(self._state_vector.dtype) for e in kraus_operators
         ]
         p = prng.random()
-        weight = None
-        fallback_weight = 0
+        fallback_weight = 0.0
         fallback_weight_index = 0
-        index = None
+
         for index in range(len(kraus_tensors)):
             prepare_into_buffer(index)
-            weight = np.linalg.norm(self._buffer) ** 2
+            weight = float(np.linalg.norm(self._buffer) ** 2)
 
             if weight > fallback_weight:
                 fallback_weight_index = index