cirq-core 1.4.0.dev20240509162830__py3-none-any.whl → 1.4.0.dev20240509224851__py3-none-any.whl

cirq/__init__.py

@@ -336,6 +336,7 @@ from cirq.ops import (
 
 from cirq.transformers import (
     AbstractInitialMapper,
+    add_dynamical_decoupling,
     align_left,
     align_right,
     CompilationTargetGateset,

cirq/_version.py

@@ -1 +1 @@
-__version__ = "1.4.0.dev20240509162830"
+__version__ = "1.4.0.dev20240509224851"

cirq/ops/qid_util.py

@@ -41,8 +41,7 @@ def q(*args: Union[int, str]) -> Union['cirq.LineQubit', 'cirq.GridQubit', 'cirq
     >>> cirq.q("foo") == cirq.NamedQubit("foo")
     True
 
-    Note that arguments should be treated as positional only, even
-    though this is only enforceable in python 3.8 or later.
+    Note that arguments should be treated as positional only.
 
     Args:
         *args: One or two ints, or a single str, as described above.

cirq/protocols/json_serialization_test.py

@@ -11,6 +11,7 @@
 # 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.
+
 import contextlib
 import dataclasses
 import datetime
@@ -19,7 +20,6 @@ import io
 import json
 import os
 import pathlib
-import sys
 import warnings
 from typing import Dict, List, Optional, Tuple, Type
 from unittest import mock
@@ -56,12 +56,6 @@ TESTED_MODULES: Dict[str, Optional[_ModuleDeprecation]] = {
 }
 
 
-# pyQuil 3.0, necessary for cirq_rigetti module requires
-# python >= 3.9
-if sys.version_info < (3, 9):  # pragma: no cover
-    del TESTED_MODULES['cirq_rigetti']
-
-
 def _get_testspecs_for_modules() -> List[ModuleJsonTestSpec]:
     modules = []
     for m in TESTED_MODULES.keys():

cirq/transformers/__init__.py

@@ -78,6 +78,8 @@ from cirq.transformers.drop_empty_moments import drop_empty_moments
 
 from cirq.transformers.drop_negligible_operations import drop_negligible_operations
 
+from cirq.transformers.dynamical_decoupling import add_dynamical_decoupling
+
 from cirq.transformers.eject_z import eject_z
 
 from cirq.transformers.measurement_transformers import (

cirq/transformers/dynamical_decoupling.py

@@ -0,0 +1,122 @@
+# Copyright 2024 The Cirq Developers
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+
+"""Transformer pass that adds dynamical decoupling operations to a circuit."""
+
+from functools import reduce
+from typing import Dict, Optional, Sequence, Tuple, Union
+
+from cirq.transformers import transformer_api
+import cirq
+import numpy as np
+
+
+def _repeat_sequence(
+    base_sequence: Sequence['cirq.Gate'], num_idle_moments: int
+) -> Sequence['cirq.Gate']:
+    """Returns the longest possible dynamical decoupling sequence."""
+    repeat_times = num_idle_moments // len(base_sequence)
+    return list(base_sequence) * repeat_times
+
+
+def _get_dd_sequence_from_schema_name(schema: str) -> Sequence['cirq.Gate']:
+    """Gets dynamical decoupling sequence from a schema name."""
+    dd_sequence: Sequence['cirq.Gate']
+    match schema:
+        case 'XX_PAIR':
+            dd_sequence = (cirq.X, cirq.X)
+        case 'X_XINV':
+            dd_sequence = (cirq.X, cirq.X**-1)
+        case 'YY_PAIR':
+            dd_sequence = (cirq.Y, cirq.Y)
+        case 'Y_YINV':
+            dd_sequence = (cirq.Y, cirq.Y**-1)
+        case _:
+            raise ValueError('Invalid schema name.')
+    return dd_sequence
+
+
+def _validate_dd_sequence(dd_sequence: Sequence['cirq.Gate']) -> None:
+    """Validates a given dynamical decoupling sequence.
+
+    Args:
+        dd_sequence: Input dynamical sequence to be validated.
+
+    Returns:
+        A tuple containing:
+            - is_valid (bool): True if the dd sequence is valid, False otherwise.
+            - error_message (str): An error message if the dd sequence is invalid, else None.
+
+    Raises:
+        ValueError: If dd_sequence is not valid.
+    """
+    if len(dd_sequence) < 2:
+        raise ValueError('Invalid dynamical decoupling sequence. Expect more than one gates.')
+    matrices = [cirq.unitary(gate) for gate in dd_sequence]
+    product = reduce(np.matmul, matrices)
+
+    if not cirq.equal_up_to_global_phase(product, np.eye(2)):
+        raise ValueError(
+            'Invalid dynamical decoupling sequence. Expect sequence production equals'
+            f' identity up to a global phase, got {product}.'.replace('\n', ' ')
+        )
+
+
+def _parse_dd_sequence(schema: Union[str, Sequence['cirq.Gate']]) -> Sequence['cirq.Gate']:
+    """Parses and returns dynamical decoupling sequence from schema."""
+    if isinstance(schema, str):
+        dd_sequence = _get_dd_sequence_from_schema_name(schema)
+    else:
+        _validate_dd_sequence(schema)
+        dd_sequence = schema
+    return dd_sequence
+
+
+@transformer_api.transformer
+def add_dynamical_decoupling(
+    circuit: 'cirq.AbstractCircuit',
+    *,
+    context: Optional['cirq.TransformerContext'] = None,
+    schema: Union[str, Sequence['cirq.Gate']] = 'X_XINV',
+) -> 'cirq.Circuit':
+    """Adds dynamical decoupling gate operations to idle moments of a given circuit.
+    This transformer preserves the moment structure of the circuit.
+
+    Args:
+          circuit: Input circuit to transform.
+          context: `cirq.TransformerContext` storing common configurable options for transformers.
+          schema: Dynamical decoupling schema name or a dynamical decoupling sequence.
+            If a schema is specified, provided dynamical decouping sequence will be used.
+            Otherwise, customized dynamical decoupling sequence will be applied.
+
+    Returns:
+          A copy of the input circuit with dynamical decoupling operations.
+    """
+    last_busy_moment_by_qubits: Dict['cirq.Qid', int] = {q: 0 for q in circuit.all_qubits()}
+    insert_into: list[Tuple[int, 'cirq.OP_TREE']] = []
+
+    base_dd_sequence = _parse_dd_sequence(schema)
+
+    for moment_id, moment in enumerate(circuit):
+        for q in moment.qubits:
+            insert_gates = _repeat_sequence(
+                base_dd_sequence, num_idle_moments=moment_id - last_busy_moment_by_qubits[q] - 1
+            )
+            for idx, gate in enumerate(insert_gates):
+                insert_into.append((last_busy_moment_by_qubits[q] + idx + 1, gate.on(q)))
+            last_busy_moment_by_qubits[q] = moment_id
+
+    updated_circuit = circuit.unfreeze(copy=True)
+    updated_circuit.batch_insert_into(insert_into)
+    return updated_circuit

cirq/transformers/dynamical_decoupling_test.py

@@ -0,0 +1,123 @@
+# Copyright 2024 The Cirq Developers
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+
+from typing import Sequence, Union
+import cirq
+from cirq import add_dynamical_decoupling
+import pytest
+
+
+def assert_dd(
+    input_circuit: cirq.Circuit,
+    expected_circuit: cirq.Circuit,
+    schema: Union[str, Sequence['cirq.Gate']],
+):
+    updated_circuit = add_dynamical_decoupling(input_circuit, schema=schema)
+    cirq.testing.assert_same_circuits(updated_circuit, expected_circuit)
+
+
+def test_no_insert_due_to_no_consecutive_moments():
+    a = cirq.NamedQubit('a')
+    b = cirq.NamedQubit('b')
+
+    # No insertion as there is no room for a dd sequence.
+    assert_dd(
+        input_circuit=cirq.Circuit(
+            cirq.Moment(cirq.H(a)), cirq.Moment(cirq.CNOT(a, b)), cirq.Moment(cirq.H(b))
+        ),
+        expected_circuit=cirq.Circuit(
+            cirq.Moment(cirq.H(a)), cirq.Moment(cirq.CNOT(a, b)), cirq.Moment(cirq.H(b))
+        ),
+        schema='XX_PAIR',
+    )
+
+
+@pytest.mark.parametrize(
+    'schema,inserted_gates',
+    [
+        ('XX_PAIR', (cirq.X, cirq.X)),
+        ('X_XINV', (cirq.X, cirq.X**-1)),
+        ('YY_PAIR', (cirq.Y, cirq.Y)),
+        ('Y_YINV', (cirq.Y, cirq.Y**-1)),
+    ],
+)
+def test_insert_provided_schema(schema: str, inserted_gates: Sequence['cirq.Gate']):
+    a = cirq.NamedQubit('a')
+    b = cirq.NamedQubit('b')
+    c = cirq.NamedQubit('c')
+
+    input_circuit = cirq.Circuit(
+        cirq.Moment(cirq.H(a)),
+        cirq.Moment(cirq.CNOT(a, b)),
+        cirq.Moment(cirq.CNOT(b, c)),
+        cirq.Moment(cirq.CNOT(b, c)),
+        cirq.Moment(cirq.measure_each(a, b, c)),
+    )
+    expected_circuit = cirq.Circuit(
+        cirq.Moment(cirq.H(a)),
+        cirq.Moment(cirq.CNOT(a, b)),
+        cirq.Moment(cirq.CNOT(b, c), inserted_gates[0](a)),
+        cirq.Moment(cirq.CNOT(b, c), inserted_gates[1](a)),
+        cirq.Moment(cirq.measure_each(a, b, c)),
+    )
+
+    # Insert one dynamical decoupling sequence in idle moments.
+    assert_dd(input_circuit, expected_circuit, schema=schema)
+
+
+def test_insert_by_customized_dd_sequence():
+    a = cirq.NamedQubit('a')
+    b = cirq.NamedQubit('b')
+    c = cirq.NamedQubit('c')
+
+    assert_dd(
+        input_circuit=cirq.Circuit(
+            cirq.Moment(cirq.H(a)),
+            cirq.Moment(cirq.CNOT(a, b)),
+            cirq.Moment(cirq.CNOT(b, c)),
+            cirq.Moment(cirq.CNOT(b, c)),
+            cirq.Moment(cirq.CNOT(b, c)),
+            cirq.Moment(cirq.CNOT(b, c)),
+            cirq.Moment(cirq.measure_each(a, b, c)),
+        ),
+        expected_circuit=cirq.Circuit(
+            cirq.Moment(cirq.H(a)),
+            cirq.Moment(cirq.CNOT(a, b)),
+            cirq.Moment(cirq.CNOT(b, c), cirq.X(a)),
+            cirq.Moment(cirq.CNOT(b, c), cirq.X(a)),
+            cirq.Moment(cirq.CNOT(b, c), cirq.Y(a)),
+            cirq.Moment(cirq.CNOT(b, c), cirq.Y(a)),
+            cirq.Moment(cirq.measure_each(a, b, c)),
+        ),
+        schema=[cirq.X, cirq.X, cirq.Y, cirq.Y],
+    )
+
+
+@pytest.mark.parametrize(
+    'schema,error_msg_regex',
+    [
+        ('INVALID_SCHEMA', 'Invalid schema name.'),
+        ([cirq.X], 'Invalid dynamical decoupling sequence. Expect more than one gates.'),
+        (
+            [cirq.X, cirq.H],
+            'Invalid dynamical decoupling sequence. Expect sequence production equals identity'
+            ' up to a global phase, got',
+        ),
+    ],
+)
+def test_invalid_dd_schema(schema: Union[str, Sequence['cirq.Gate']], error_msg_regex):
+    a = cirq.NamedQubit('a')
+    input_circuit = cirq.Circuit(cirq.H(a))
+    with pytest.raises(ValueError, match=error_msg_regex):
+        add_dynamical_decoupling(input_circuit, schema=schema)

{cirq_core-1.4.0.dev20240509162830.dist-info → cirq_core-1.4.0.dev20240509224851.dist-info}/METADATA

@@ -1,18 +1,18 @@
 Metadata-Version: 2.1
 Name: cirq-core
-Version: 1.4.0.dev20240509162830
+Version: 1.4.0.dev20240509224851
 Summary: A framework for creating, editing, and invoking Noisy Intermediate Scale Quantum (NISQ) circuits.
 Home-page: http://github.com/quantumlib/cirq
 Author: The Cirq Developers
 Author-email: cirq-dev@googlegroups.com
 License: Apache 2
-Requires-Python: >=3.9.0
+Requires-Python: >=3.10.0
 License-File: LICENSE
 Requires-Dist: attrs
 Requires-Dist: duet >=0.2.8
 Requires-Dist: matplotlib ~=3.0
 Requires-Dist: networkx >=2.4
-Requires-Dist: numpy ~=1.16
+Requires-Dist: numpy ~=1.22
 Requires-Dist: pandas
 Requires-Dist: sortedcontainers ~=2.0
 Requires-Dist: scipy <1.13.0

{cirq_core-1.4.0.dev20240509162830.dist-info → cirq_core-1.4.0.dev20240509224851.dist-info}/RECORD

@@ -1,10 +1,10 @@
-cirq/__init__.py,sha256=Pisx61VoUhqDdggQ31ch0Mx3uMwVtZ0q7Z69Ib4C92s,15736
+cirq/__init__.py,sha256=gQkxjgGyat-egE2jZlrLCVMofvpG_VA1021yh8Tv9j8,15766
 cirq/_compat.py,sha256=2tkJ50ID2PXzmMNuTfQrsZqqRSblTmfu3Y7g742gojs,29342
 cirq/_compat_test.py,sha256=Qq3ZcfgD-Nb81cEppQdJqhAyrVqXKtfXZYGXT0p-Wh0,34718
 cirq/_doc.py,sha256=yDyWUD_2JDS0gShfGRb-rdqRt9-WeL7DhkqX7np0Nko,2879
 cirq/_import.py,sha256=p9gMHJscbtDDkfHOaulvd3Aer0pwUF5AXpL89XR8dNw,8402
 cirq/_import_test.py,sha256=6K_v0riZJXOXUphHNkGA8MY-JcmGlezFaGmvrNhm3OQ,1015
-cirq/_version.py,sha256=HvH8RsJtt9j4_dutS6yEhtOpobrHfzOwzWux-vblBDw,40
+cirq/_version.py,sha256=Oe2oO7fLniWMge5QU8jj7I7z689VwMkUcRfBX4DFRrU,40
 cirq/_version_test.py,sha256=yYS6xm5-nuBRQJa9R3n41WdvFtVyY7Lb5Q8bea3VgBI,133
 cirq/conftest.py,sha256=X7yLFL8GLhg2CjPw0hp5e_dGASfvHx1-QT03aUbhKJw,1168
 cirq/json_resolver_cache.py,sha256=9g_JQMmfBzTuV-3s2flUbXIgcLjs4K7LjAFFgngdG1U,13204
@@ -349,7 +349,7 @@ cirq/ops/phased_x_z_gate.py,sha256=eHYMyPiDhlEyzphJDnaTeLdRn96Tns1VfiYv6FtPShU,1
 cirq/ops/phased_x_z_gate_test.py,sha256=8PCwOgGGWL32QIR8m1kBd5JG3Mq9BMDeeVD0aPk1E90,9541
 cirq/ops/projector.py,sha256=isDlNLR8YS5CxsVCihtRAgfKZXFFQxk7kuFXY14t8ys,5690
 cirq/ops/projector_test.py,sha256=Wq7ddj-PV30yUXJxJoT3XIw2sIUC1AilnZ9m9N5Ejr8,9063
-cirq/ops/qid_util.py,sha256=IO7vhq5RgeIujO2EhG_FCIvkIMMr2jTyz8gG5M0mrIU,2136
+cirq/ops/qid_util.py,sha256=SxyoMy_s070lcqXV7ny6vE8pp082OrTjBtawZtHOhXs,2071
 cirq/ops/qid_util_test.py,sha256=JdViBgFfH4bZJyPKTjUf9MuPxMQe08JV_Tl6tusekfk,1061
 cirq/ops/qubit_manager.py,sha256=U8dML-iosNfQn16Ug62z2-b2PFf-iDWc1ZCLuUsIc-g,3539
 cirq/ops/qubit_manager_test.py,sha256=SQ_6kOehIaMgRvU43nTOEy6UoDxZSeVzSYQ2Mfouexg,3371
@@ -400,7 +400,7 @@ cirq/protocols/has_unitary_protocol_test.py,sha256=IjmJ3dqvteFUUO4FLwCTokgUvYy8H
 cirq/protocols/inverse_protocol.py,sha256=CEqtGRRj86WQyyALonRXxQrNq-fENOs_Zqrlr_BVau8,4115
 cirq/protocols/inverse_protocol_test.py,sha256=pqqIU4_G4Npc9Z-SeoM9eCB2T5JRTeI02NCXhP0UtaI,2017
 cirq/protocols/json_serialization.py,sha256=Otrrzsi74rfE_2wbK-xBUI3QzQaRnEXqsY1D_ktW4jk,24360
-cirq/protocols/json_serialization_test.py,sha256=qPL80GpJBEN3pWe_vhNFCvUFQuuta3DKhCPwnTRnzdI,27635
+cirq/protocols/json_serialization_test.py,sha256=Cn116fKZ-k7KNxwJXrCQSct7ioeP1hKIcZZ4HE5q2gk,27461
 cirq/protocols/kraus_protocol.py,sha256=GKQ748yxo6B6i0jUzq8OyR6lcEFYbh4FV2UryjUgSuk,9158
 cirq/protocols/kraus_protocol_test.py,sha256=QpPwkk7XBiGKNdrOcjXqtvhESVbUzujnUPz47uUNyqc,5401
 cirq/protocols/measurement_key_protocol.py,sha256=hl7xFi1Evebh8MQNLuUffhsUkggr4SdWUm4kIr9s2iI,13425
@@ -1013,13 +1013,15 @@ cirq/testing/test_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 cirq/testing/test_data/test_module_missing_json_test_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cirq/testing/test_data/test_module_missing_testspec/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cirq/testing/test_data/test_module_missing_testspec/json_test_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cirq/transformers/__init__.py,sha256=Omi-8NLq9lMbMHV-eXxwrpX5_zw-ZyRJKPmIL_mIzMg,4145
+cirq/transformers/__init__.py,sha256=-buJqTF2c6rUM18JDGSb5aMKWtyT24jE_mMrHo6DsdM,4222
 cirq/transformers/align.py,sha256=B4DuX84DWd4tLfMH21JviMZSOFYu7KCOMSEMALYsQpw,2727
 cirq/transformers/align_test.py,sha256=M4etT2cgESv1RdkKASguiGxEuqY7kmI1IswjSi-1jjY,7174
 cirq/transformers/drop_empty_moments.py,sha256=Rtn_BrpwkLXyZBdLzwdnsnEGWTdYuf1xOPakzbpv7-w,1517
 cirq/transformers/drop_empty_moments_test.py,sha256=G8pZmTfi8NG2NpGz_K3LZu5NQoqa-xPMCuZjwEu07xk,1907
 cirq/transformers/drop_negligible_operations.py,sha256=8eyOMy7bra2wJAjORbk6QjwHiLdL5SfwRaz8D2Dazbw,2083
 cirq/transformers/drop_negligible_operations_test.py,sha256=gqL6RoDPm6Zf4RxtprBenFyIsZQPUxmPur9oRl0Yr3U,3823
+cirq/transformers/dynamical_decoupling.py,sha256=iALBcjaO3uwcAla-smjapp2z9YezGecGFo_OGgJkD_8,4704
+cirq/transformers/dynamical_decoupling_test.py,sha256=hGUzTDV1UL3VBIueri91j3_ycxGmr20mGfQSnRAjegM,4149
 cirq/transformers/eject_phased_paulis.py,sha256=mTgRT5aw5_c9ccTkP4Np_4YTWnLzxsMKRO8pOQ7CtYM,13955
 cirq/transformers/eject_phased_paulis_test.py,sha256=-mXsfbi3V0ojC_YqoQM5otzdW4kjGusCx6F-kCv8M98,15834
 cirq/transformers/eject_z.py,sha256=0kOyvh6FDfrCrrTCVfpHKNc_kNC_pBdEKoXv11kuqGA,5803
@@ -1169,8 +1171,8 @@ cirq/work/sampler.py,sha256=JEAeQQRF3bqlO9AkOf4XbrTATDI5f5JgyM_FAUCNxao,19751
 cirq/work/sampler_test.py,sha256=B2ZsuqGT854gQtBIAh8k0LiG9Vj5wSzcGvkxOUoTcW4,13217
 cirq/work/zeros_sampler.py,sha256=x1C7cup66a43n-3tm8QjhiqJa07qcJW10FxNp9jJ59Q,2356
 cirq/work/zeros_sampler_test.py,sha256=JIkpBBFPJe5Ba4142vzogyWyboG1Q1ZAm0UVGgOoZn8,3279
-cirq_core-1.4.0.dev20240509162830.dist-info/LICENSE,sha256=tAkwu8-AdEyGxGoSvJ2gVmQdcicWw3j1ZZueVV74M-E,11357
-cirq_core-1.4.0.dev20240509162830.dist-info/METADATA,sha256=U2n22LwiwY8DdkbHvLf9F9IpC_6OAx-5-uV9SWHwizk,1999
-cirq_core-1.4.0.dev20240509162830.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-cirq_core-1.4.0.dev20240509162830.dist-info/top_level.txt,sha256=Sz9iOxHU0IEMLSFGwiwOCaN2e9K-jFbBbtpPN1hB73g,5
-cirq_core-1.4.0.dev20240509162830.dist-info/RECORD,,
+cirq_core-1.4.0.dev20240509224851.dist-info/LICENSE,sha256=tAkwu8-AdEyGxGoSvJ2gVmQdcicWw3j1ZZueVV74M-E,11357
+cirq_core-1.4.0.dev20240509224851.dist-info/METADATA,sha256=VWh4JaX499yyPSZoAlVl2F0SUMTk6zuqukkKI7O7D3g,2000
+cirq_core-1.4.0.dev20240509224851.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+cirq_core-1.4.0.dev20240509224851.dist-info/top_level.txt,sha256=Sz9iOxHU0IEMLSFGwiwOCaN2e9K-jFbBbtpPN1hB73g,5
+cirq_core-1.4.0.dev20240509224851.dist-info/RECORD,,