cirq-core 1.7.0.dev20250818230756__py3-none-any.whl → 1.7.0.dev20250820205919__py3-none-any.whl

cirq/_version.py

@@ -28,4 +28,4 @@ if sys.version_info < (3, 11, 0):  # pragma: no cover
         'of Cirq (e.g. "python -m pip install cirq==1.5.0")'
     )
 
-__version__ = "1.7.0.dev20250818230756"
+__version__ = "1.7.0.dev20250820205919"

cirq/_version_test.py

@@ -3,4 +3,4 @@ import cirq
 
 
 def test_version() -> None:
-    assert cirq.__version__ == "1.7.0.dev20250818230756"
+    assert cirq.__version__ == "1.7.0.dev20250820205919"

cirq/experiments/z_phase_calibration_test.py

@@ -88,7 +88,7 @@ class _TestSimulator(cirq.Simulator):
 @pytest.mark.parametrize(
     ['angles', 'error', 'characterization_flags'], _create_tests(n=10, with_options=True)
 )
-def test_calibrate_z_phases(angles, error, characterization_flags):
+def test_calibrate_z_phases(angles, error, characterization_flags) -> None:
 
     original_gate = cirq.PhasedFSimGate(**{k: v for k, v in zip(_ANGLES, angles)})
     actual_gate = cirq.PhasedFSimGate(**{k: v + e for k, v, e in zip(_ANGLES, angles, error)})
@@ -127,7 +127,7 @@ def test_calibrate_z_phases(angles, error, characterization_flags):
 
 
 @pytest.mark.parametrize(['angles', 'error'], _create_tests(n=3))
-def test_calibrate_z_phases_no_options(angles, error):
+def test_calibrate_z_phases_no_options(angles, error) -> None:
 
     original_gate = cirq.PhasedFSimGate(**{k: v for k, v in zip(_ANGLES, angles)})
     actual_gate = cirq.PhasedFSimGate(**{k: v + e for k, v, e in zip(_ANGLES, angles, error)})
@@ -162,7 +162,7 @@ def test_calibrate_z_phases_no_options(angles, error):
 
 
 @pytest.mark.parametrize(['angles', 'error'], _create_tests(n=3))
-def test_calibrate_z_phases_workflow_no_options(angles, error):
+def test_calibrate_z_phases_workflow_no_options(angles, error) -> None:
 
     original_gate = cirq.PhasedFSimGate(**{k: v for k, v in zip(_ANGLES, angles)})
     actual_gate = cirq.PhasedFSimGate(**{k: v + e for k, v, e in zip(_ANGLES, angles, error)})
@@ -189,7 +189,7 @@ def test_calibrate_z_phases_workflow_no_options(angles, error):
         assert 'theta' not in params
 
 
-def test_plot_z_phase_calibration_result():
+def test_plot_z_phase_calibration_result() -> None:
     df = pd.DataFrame()
     qs = cirq.q(0, 0), cirq.q(0, 1), cirq.q(0, 2)
     df.index = [qs[:2], qs[-2:]]
@@ -211,13 +211,14 @@ def test_plot_z_phase_calibration_result():
 
 
 @pytest.mark.parametrize('angles', 2 * np.pi * np.random.random((10, 10)))
-def test_transform_circuit(angles):
+def test_transform_circuit(angles) -> None:
     theta, phi = angles[:2]
     old_zs = angles[2:6]
     new_zs = angles[6:]
     gate = cirq.PhasedFSimGate.from_fsim_rz(theta, phi, old_zs[:2], old_zs[2:])
     fsim = cirq.PhasedFSimGate.from_fsim_rz(theta, phi, new_zs[:2], new_zs[2:])
     c = cirq.Circuit(gate(cirq.q(0), cirq.q(1)))
+    replacement_map: dict[tuple[cirq.Qid, cirq.Qid], cirq.PhasedFSimGate]
     replacement_map = {(cirq.q(1), cirq.q(0)): fsim}
 
     new_circuit = CalibrationTransformer(gate, replacement_map)(c)
@@ -230,11 +231,11 @@ def test_transform_circuit(angles):
     np.testing.assert_allclose(cirq.unitary(circuit_with_replacement_gate), cirq.unitary(c))
 
 
-def test_transform_circuit_invalid_gate_raises():
+def test_transform_circuit_invalid_gate_raises() -> None:
     with pytest.raises(ValueError, match="is not equivalent to a PhasedFSimGate"):
         _ = CalibrationTransformer(cirq.XX, {})
 
 
-def test_transform_circuit_uncalibrated_gates_pass():
+def test_transform_circuit_uncalibrated_gates_pass() -> None:
     c = cirq.Circuit(cirq.CZ(cirq.q(0), cirq.q(1)), cirq.measure(cirq.q(0)))
     assert c == CalibrationTransformer(cirq.CZ, {})(c)

cirq/interop/quirk/cells/arithmetic_cells.py

@@ -205,7 +205,7 @@ class ArithmeticCell(Cell):
         )
 
     @property
-    def operation(self):
+    def operation(self) -> _QuirkArithmeticCallable:
         return ARITHMETIC_OP_TABLE[self.identifier]
 
     def with_input(self, letter: str, register: Sequence[cirq.Qid] | int) -> ArithmeticCell:

cirq/interop/quirk/cells/cell_test.py

@@ -20,7 +20,7 @@ import cirq
 from cirq.interop.quirk.cells.cell import Cell, ExplicitOperationsCell
 
 
-def test_cell_defaults():
+def test_cell_defaults() -> None:
     class BasicCell(Cell):
         def with_line_qubits_mapped_to(self, qubits):
             raise NotImplementedError()
@@ -32,12 +32,12 @@ def test_cell_defaults():
     assert c.operations() == ()
     assert c.basis_change() == ()
     assert c.controlled_by(cirq.LineQubit(0)) is c
-    x = []
+    x: list[Cell | None] = []
     c.modify_column(x)
     assert x == []
 
 
-def test_cell_replace_utils():
+def test_cell_replace_utils() -> None:
     a, b, c = cirq.NamedQubit.range(3, prefix='q')
     assert Cell._replace_qubit(cirq.LineQubit(1), [a, b, c]) == b
     with pytest.raises(ValueError, match='only map from line qubits'):
@@ -48,7 +48,7 @@ def test_cell_replace_utils():
         _ = Cell._replace_qubit(cirq.LineQubit(999), [a, b, c])
 
 
-def test_explicit_operations_cell_equality():
+def test_explicit_operations_cell_equality() -> None:
     a = cirq.LineQubit(0)
     eq = cirq.testing.EqualsTester()
     eq.add_equality_group(ExplicitOperationsCell([], []), ExplicitOperationsCell([]))
@@ -56,7 +56,7 @@ def test_explicit_operations_cell_equality():
     eq.add_equality_group(ExplicitOperationsCell([], [cirq.Y(a)]))
 
 
-def test_explicit_operations_cell():
+def test_explicit_operations_cell() -> None:
     a, b = cirq.LineQubit.range(2)
     v = ExplicitOperationsCell([cirq.X(a)], [cirq.S(a)])
     assert v.operations() == (cirq.X(a),)

cirq/interop/quirk/cells/composite_cell_test.py

@@ -14,6 +14,8 @@
 
 from __future__ import annotations
 
+from typing import Iterable
+
 import pytest
 
 import cirq
@@ -22,7 +24,7 @@ from cirq.interop.quirk.cells.composite_cell import _iterator_to_iterable
 from cirq.interop.quirk.cells.testing import assert_url_to_circuit_returns
 
 
-def test_iterator_to_iterable():
+def test_iterator_to_iterable() -> None:
     k = 0
 
     def counter():
@@ -32,7 +34,7 @@ def test_iterator_to_iterable():
 
     # Normal iterator usage.
     k = 0
-    generator = (counter() for _ in range(10))
+    generator: Iterable[int] = (counter() for _ in range(10))
     assert k == 0
     assert list(generator) == list(range(10))
     assert k == 10
@@ -73,7 +75,7 @@ def test_iterator_to_iterable():
     assert k == 10
 
 
-def test_custom_circuit_gate():
+def test_custom_circuit_gate() -> None:
     a, b, c, d, e = cirq.LineQubit.range(5)
 
     # Without name.

cirq/interop/quirk/cells/control_cells.py

@@ -53,7 +53,7 @@ class ControlCell(Cell):
             ),
         )
 
-    def modify_column(self, column: list[Cell | None]):
+    def modify_column(self, column: list[Cell | None]) -> None:
         for i in range(len(column)):
             gate = column[i]
             if gate is not None:
@@ -97,7 +97,7 @@ class ParityControlCell(Cell):
             ),
         )
 
-    def modify_column(self, column: list[Cell | None]):
+    def modify_column(self, column: list[Cell | None]) -> None:
         for i in range(len(column)):
             gate = column[i]
             if gate is self:

cirq/interop/quirk/cells/input_cells.py

@@ -14,7 +14,7 @@
 
 from __future__ import annotations
 
-from typing import Iterable, Iterator, TYPE_CHECKING
+from typing import Callable, Iterable, Iterator, TYPE_CHECKING
 
 from cirq.interop.quirk.cells.cell import Cell, CELL_SIZES, CellMaker
 
@@ -35,7 +35,7 @@ class InputCell(Cell):
     def with_line_qubits_mapped_to(self, qubits: list[cirq.Qid]) -> Cell:
         return InputCell(qubits=Cell._replace_qubits(self.qubits, qubits), letter=self.letter)
 
-    def modify_column(self, column: list[Cell | None]):
+    def modify_column(self, column: list[Cell | None]) -> None:
         for i in range(len(column)):
             cell = column[i]
             if cell is not None:
@@ -55,7 +55,7 @@ class SetDefaultInputCell(Cell):
     def with_line_qubits_mapped_to(self, qubits: list[cirq.Qid]) -> Cell:
         return self
 
-    def persistent_modifiers(self):
+    def persistent_modifiers(self) -> dict[str, Callable[[Cell], Cell]]:
         return {f'set_default_{self.letter}': lambda cell: cell.with_input(self.letter, self.value)}
 
 

cirq/interop/quirk/cells/input_rotation_cells.py

@@ -77,7 +77,7 @@ class InputRotationCell(Cell):
             )
         return self
 
-    def controlled_by(self, qubit: cirq.Qid):
+    def controlled_by(self, qubit: cirq.Qid) -> InputRotationCell:
         return InputRotationCell(
             self.identifier,
             self.register,
@@ -118,7 +118,7 @@ class QuirkInputRotationOperation(ops.Operation):
     def qubits(self) -> tuple[cirq.Qid, ...]:
         return tuple(self.base_operation.qubits) + self.register
 
-    def with_qubits(self, *new_qubits):
+    def with_qubits(self, *new_qubits) -> QuirkInputRotationOperation:
         k = len(self.base_operation.qubits)
         new_op_qubits = new_qubits[:k]
         new_register = new_qubits[k:]

cirq/interop/quirk/cells/swap_cell.py

@@ -38,7 +38,7 @@ class SwapCell(Cell):
             controls=Cell._replace_qubits(self._controls, qubits),
         )
 
-    def modify_column(self, column: list[Cell | None]):
+    def modify_column(self, column: list[Cell | None]) -> None:
         # Swallow other swap cells.
         for i in range(len(column)):
             gate = column[i]
@@ -52,7 +52,7 @@ class SwapCell(Cell):
             raise ValueError('Wrong number of swap gates in a column.')
         return ops.SWAP(*self._qubits).controlled_by(*self._controls)
 
-    def controlled_by(self, qubit: cirq.Qid):
+    def controlled_by(self, qubit: cirq.Qid) -> SwapCell:
         return SwapCell(self._qubits, self._controls + [qubit])
 
     def _value_equality_values_(self) -> Any:

cirq/interop/quirk/cells/testing.py

@@ -28,7 +28,7 @@ def assert_url_to_circuit_returns(
     diagram: str | None = None,
     output_amplitudes_from_quirk: list[dict[str, float]] | None = None,
     maps: dict[int, int] | None = None,
-):
+) -> None:
     """Assert that `quirk_url_to_circuit` functions correctly.
 
     Args:

cirq/interop/quirk/url_to_circuit.py

@@ -39,7 +39,7 @@ def quirk_url_to_circuit(
     quirk_url: str,
     *,
     qubits: Sequence[cirq.Qid] | None = None,
-    extra_cell_makers: dict[str, cirq.Gate] | Iterable[cirq.interop.quirk.cells.CellMaker] = (),
+    extra_cell_makers: Mapping[str, cirq.Gate] | Iterable[cirq.interop.quirk.cells.CellMaker] = (),
     max_operation_count: int = 10**6,
 ) -> cirq.Circuit:
     """Parses a Cirq circuit out of a Quirk URL.
@@ -140,7 +140,7 @@ def quirk_json_to_circuit(
     data: dict,
     *,
     qubits: Sequence[cirq.Qid] | None = None,
-    extra_cell_makers: dict[str, cirq.Gate] | Iterable[cirq.interop.quirk.cells.CellMaker] = (),
+    extra_cell_makers: Mapping[str, cirq.Gate] | Iterable[cirq.interop.quirk.cells.CellMaker] = (),
     quirk_url: str | None = None,
     max_operation_count: int = 10**6,
 ) -> cirq.Circuit:

cirq/interop/quirk/url_to_circuit_test.py

@@ -25,7 +25,7 @@ from cirq import quirk_json_to_circuit, quirk_url_to_circuit
 from cirq.interop.quirk.cells.testing import assert_url_to_circuit_returns
 
 
-def test_parse_simple_cases():
+def test_parse_simple_cases() -> None:
     a, b = cirq.LineQubit.range(2)
 
     assert quirk_url_to_circuit('http://algassert.com/quirk') == cirq.Circuit()
@@ -47,7 +47,7 @@ def test_parse_simple_cases():
         ('http://algassert.com/quirk#circuit=', json.JSONDecodeError, None),
     ],
 )
-def test_parse_url_failures(url, error_cls, msg):
+def test_parse_url_failures(url, error_cls, msg) -> None:
     with pytest.raises(error_cls, match=msg):
         _ = quirk_url_to_circuit(url)
 
@@ -70,7 +70,7 @@ def test_parse_url_failures(url, error_cls, msg):
         ),
     ],
 )
-def test_parse_failures(url, msg):
+def test_parse_failures(url, msg) -> None:
     parsed_url = urllib.parse.urlparse(url)
     data = json.loads(parsed_url.fragment[len('circuit=') :])
 
@@ -81,7 +81,7 @@ def test_parse_failures(url, msg):
         _ = quirk_json_to_circuit(data)
 
 
-def test_parse_with_qubits():
+def test_parse_with_qubits() -> None:
     a = cirq.GridQubit(0, 0)
     b = cirq.GridQubit(0, 1)
     c = cirq.GridQubit(0, 2)
@@ -103,7 +103,7 @@ def test_parse_with_qubits():
         )
 
 
-def test_extra_cell_makers():
+def test_extra_cell_makers() -> None:
     assert cirq.quirk_url_to_circuit(
         'http://algassert.com/quirk#circuit={"cols":[["iswap"]]}',
         extra_cell_makers=[
@@ -140,7 +140,7 @@ def test_extra_cell_makers():
     )
 
 
-def test_init():
+def test_init() -> None:
     b, c, d, e, f = cirq.LineQubit.range(1, 6)
     assert_url_to_circuit_returns(
         '{"cols":[],"init":[0,1,"+","-","i","-i"]}',
@@ -176,7 +176,7 @@ def test_init():
         _ = cirq.quirk_url_to_circuit('http://algassert.com/quirk#circuit={"cols":[],"init":[2]}')
 
 
-def test_custom_gate_parse_failures():
+def test_custom_gate_parse_failures() -> None:
     with pytest.raises(ValueError, match='must be a list'):
         _ = quirk_url_to_circuit('https://algassert.com/quirk#circuit={"cols":[],"gates":5}')
 
@@ -229,7 +229,7 @@ def test_custom_gate_parse_failures():
         )
 
 
-def test_custom_matrix_gate():
+def test_custom_matrix_gate() -> None:
     a, b = cirq.LineQubit.range(2)
 
     # Without name.
@@ -259,7 +259,7 @@ def test_custom_matrix_gate():
     )
 
 
-def test_survives_a_billion_laughs():
+def test_survives_a_billion_laughs() -> None:
     # If this test is timing out, it means you made a change that accidentally
     # iterated over the circuit contents before they were counted and checked
     # against the maximum. It is not possible to test for the billion laughs
@@ -301,7 +301,7 @@ def test_survives_a_billion_laughs():
         )
 
 
-def test_completes_weight_zero_billion_laughs():
+def test_completes_weight_zero_billion_laughs() -> None:
     circuit = cirq.quirk_url_to_circuit(
         'https://algassert.com/quirk#circuit={'
         '"cols":[["~z"]],'
@@ -339,7 +339,7 @@ def test_completes_weight_zero_billion_laughs():
     assert circuit == cirq.Circuit()
 
 
-def test_example_qft_circuit():
+def test_example_qft_circuit() -> None:
     qft_example_diagram = """
 0: ───×───────────────H───@───────────@────────────────────@──────────────────────────────@─────────────────────────────────────────@───────────────────────────────────────────────────@─────────────────────────────────────────────────────────────@───────────────────────────────────────────────────────────────────────
       │                   │           │                    │                              │                                         │                                                   │                                                             │

cirq/linalg/decompositions.py

@@ -518,7 +518,7 @@ def scatter_plot_normalized_kak_interaction_coefficients(
     include_frame: bool = True,
     ax: mplot3d.axes3d.Axes3D | None = None,
     **kwargs,
-):
+) -> mplot3d.axes3d.Axes3D:
     r"""Plots the interaction coefficients of many two-qubit operations.
 
     Plots: