cirq-core 1.6.0.dev20250507223949__py3-none-any.whl → 1.6.0.dev20250507225534__py3-none-any.whl

cirq/__init__.py

@@ -363,6 +363,7 @@ from cirq.transformers import (
     eject_z as eject_z,
     expand_composite as expand_composite,
     HardCodedInitialMapper as HardCodedInitialMapper,
+    index_tags as index_tags,
     is_negligible_turn as is_negligible_turn,
     LineInitialMapper as LineInitialMapper,
     MappingManager as MappingManager,
@@ -385,6 +386,7 @@ from cirq.transformers import (
     prepare_two_qubit_state_using_sqrt_iswap as prepare_two_qubit_state_using_sqrt_iswap,
     quantum_shannon_decomposition as quantum_shannon_decomposition,
     RouteCQC as RouteCQC,
+    remove_tags as remove_tags,
     routed_circuit_with_mapping as routed_circuit_with_mapping,
     SqrtIswapTargetGateset as SqrtIswapTargetGateset,
     single_qubit_matrix_to_gates as single_qubit_matrix_to_gates,

cirq/_version.py

@@ -28,4 +28,4 @@ if sys.version_info < (3, 10, 0):  # pragma: no cover
         'of cirq (e.g. "python -m pip install cirq==1.1.*")'
     )
 
-__version__ = "1.6.0.dev20250507223949"
+__version__ = "1.6.0.dev20250507225534"

cirq/_version_test.py

@@ -3,4 +3,4 @@ import cirq
 
 
 def test_version():
-    assert cirq.__version__ == "1.6.0.dev20250507223949"
+    assert cirq.__version__ == "1.6.0.dev20250507225534"

cirq/transformers/__init__.py

@@ -119,6 +119,8 @@ from cirq.transformers.transformer_api import (
     transformer as transformer,
 )
 
+from cirq.transformers.tag_transformers import index_tags as index_tags, remove_tags as remove_tags
+
 from cirq.transformers.transformer_primitives import (
     map_moments as map_moments,
     map_operations as map_operations,

cirq/transformers/tag_transformers.py

@@ -0,0 +1,95 @@
+# Copyright 2025 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.
+
+import itertools
+from typing import Callable, Hashable, Optional, TYPE_CHECKING
+
+from cirq.transformers import transformer_api, transformer_primitives
+
+if TYPE_CHECKING:
+    import cirq
+
+
+@transformer_api.transformer
+def index_tags(
+    circuit: 'cirq.AbstractCircuit',
+    *,
+    context: Optional['cirq.TransformerContext'] = None,
+    target_tags: Optional[set[Hashable]] = None,
+) -> 'cirq.Circuit':
+    """Indexes tags in target_tags as tag_0, tag_1, ... per tag.
+
+    Args:
+        circuit: Input circuit to apply the transformations on. The input circuit is not mutated.
+        context: `cirq.TransformerContext` storing common configurable options for transformers.
+        target_tags: Tags to be indexed.
+
+    Returns:
+        Copy of the transformed input circuit.
+    """
+    if context and context.tags_to_ignore:
+        raise ValueError("index_tags doesn't support tags_to_ignore, use function args instead.")
+    if not target_tags:
+        return circuit.unfreeze(copy=False)
+    tag_iter_by_tags = {tag: itertools.count(start=0, step=1) for tag in target_tags}
+
+    def _map_func(op: 'cirq.Operation', _) -> 'cirq.OP_TREE':
+        tag_set = set(op.tags)
+        nonlocal tag_iter_by_tags
+        for tag in target_tags.intersection(op.tags):
+            tag_set.remove(tag)
+            tag_set.add(f"{tag}_{next(tag_iter_by_tags[tag])}")
+
+        return op.untagged.with_tags(*tag_set)
+
+    return transformer_primitives.map_operations(
+        circuit, _map_func, deep=context.deep if context else False
+    ).unfreeze(copy=False)
+
+
+@transformer_api.transformer
+def remove_tags(
+    circuit: 'cirq.AbstractCircuit',
+    *,
+    context: Optional['cirq.TransformerContext'] = None,
+    target_tags: Optional[set[Hashable]] = None,
+    remove_if: Callable[[Hashable], bool] = lambda _: False,
+) -> 'cirq.Circuit':
+    """Removes tags from the operations based on the input args.
+
+    Args:
+        circuit: Input circuit to apply the transformations on. The input circuit is not mutated.
+        context: `cirq.TransformerContext` storing common configurable options for transformers.
+        target_tags: Tags to be removed.
+        remove_if: A callable(tag) that returns True if the tag should be removed.
+          Defaults to False.
+
+    Returns:
+        Copy of the transformed input circuit.
+    """
+    if context and context.tags_to_ignore:
+        raise ValueError("remove_tags doesn't support tags_to_ignore, use function args instead.")
+    target_tags = target_tags or set()
+
+    def _map_func(op: 'cirq.Operation', _) -> 'cirq.OP_TREE':
+        remaing_tags = set()
+        for tag in op.tags:
+            if not remove_if(tag) and tag not in target_tags:
+                remaing_tags.add(tag)
+
+        return op.untagged.with_tags(*remaing_tags)
+
+    return transformer_primitives.map_operations(
+        circuit, _map_func, deep=context.deep if context else False
+    ).unfreeze(copy=False)

cirq/transformers/tag_transformers_test.py

@@ -0,0 +1,101 @@
+# Copyright 2025 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.
+
+import pytest
+
+import cirq
+
+
+def check_same_circuit_with_same_tag_sets(circuit1, circuit2):
+    for op1, op2 in zip(circuit1.all_operations(), circuit2.all_operations()):
+        assert set(op1.tags) == set(op2.tags)
+        assert op1.untagged == op2.untagged
+
+
+def test_index_tags():
+    q0, q1 = cirq.LineQubit.range(2)
+    input_circuit = cirq.Circuit(
+        cirq.X(q0).with_tags("tag1", "tag2"),
+        cirq.Y(q1).with_tags("tag1"),
+        cirq.CZ(q0, q1).with_tags("tag2"),
+    )
+    expected_circuit = cirq.Circuit(
+        cirq.X(q0).with_tags("tag1_0", "tag2_0"),
+        cirq.Y(q1).with_tags("tag1_1"),
+        cirq.CZ(q0, q1).with_tags("tag2_1"),
+    )
+    check_same_circuit_with_same_tag_sets(
+        cirq.index_tags(input_circuit, target_tags={"tag1", "tag2"}), expected_circuit
+    )
+
+
+def test_index_tags_empty_target_tags():
+    q0, q1 = cirq.LineQubit.range(2)
+    input_circuit = cirq.Circuit(
+        cirq.X(q0).with_tags("tag1", "tag2"),
+        cirq.Y(q1).with_tags("tag1"),
+        cirq.CZ(q0, q1).with_tags("tag2"),
+    )
+    check_same_circuit_with_same_tag_sets(
+        cirq.index_tags(input_circuit, target_tags={}), input_circuit
+    )
+
+
+def test_remove_tags():
+    q0, q1 = cirq.LineQubit.range(2)
+    input_circuit = cirq.Circuit(
+        cirq.X(q0).with_tags("tag1", "tag2"),
+        cirq.Y(q1).with_tags("tag1"),
+        cirq.CZ(q0, q1).with_tags("tag2"),
+    )
+    expected_circuit = cirq.Circuit(
+        cirq.X(q0).with_tags("tag2"), cirq.Y(q1), cirq.CZ(q0, q1).with_tags("tag2")
+    )
+    check_same_circuit_with_same_tag_sets(
+        cirq.remove_tags(input_circuit, target_tags={"tag1"}), expected_circuit
+    )
+
+
+def test_remove_tags_via_remove_if():
+    q0, q1 = cirq.LineQubit.range(2)
+    input_circuit = cirq.Circuit(
+        cirq.X(q0).with_tags("tag1", "tag2"),
+        cirq.Y(q1).with_tags("not_tag1"),
+        cirq.CZ(q0, q1).with_tags("tag2"),
+    )
+    expected_circuit = cirq.Circuit(cirq.X(q0), cirq.Y(q1).with_tags("not_tag1"), cirq.CZ(q0, q1))
+    check_same_circuit_with_same_tag_sets(
+        cirq.remove_tags(input_circuit, remove_if=lambda tag: tag.startswith("tag")),
+        expected_circuit,
+    )
+
+
+def test_index_tags_with_tags_to_ignore():
+    with pytest.raises(
+        ValueError, match="index_tags doesn't support tags_to_ignore, use function args instead."
+    ):
+        cirq.index_tags(
+            circuit=cirq.Circuit(),
+            target_tags={"tag0"},
+            context=cirq.TransformerContext(tags_to_ignore=["tag0"]),
+        )
+
+
+def test_remove_tags_with_tags_to_ignore():
+    with pytest.raises(
+        ValueError, match="remove_tags doesn't support tags_to_ignore, use function args instead."
+    ):
+        cirq.remove_tags(
+            circuit=cirq.Circuit(), context=cirq.TransformerContext(tags_to_ignore=["tag0"])
+        )

{cirq_core-1.6.0.dev20250507223949.dist-info → cirq_core-1.6.0.dev20250507225534.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cirq-core
-Version: 1.6.0.dev20250507223949
+Version: 1.6.0.dev20250507225534
 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

{cirq_core-1.6.0.dev20250507223949.dist-info → cirq_core-1.6.0.dev20250507225534.dist-info}/RECORD

@@ -1,11 +1,11 @@
-cirq/__init__.py,sha256=rUfvQDtywCak2mJQoihOSyRjGxQahK-YOv909us0w5M,28132
+cirq/__init__.py,sha256=ecDdbOid5m5VpqO4ZPoXQRuWUanrH3Y7lGzzgNPpiyM,28194
 cirq/_compat.py,sha256=_DknO27XngcjEidNApRsCzLUWDS4QmDk9M12BaqP5Is,29531
 cirq/_compat_test.py,sha256=t51ZXkEuomg1SMI871Ws-5pk68DGBsAf2TGNjVXtZ8I,34755
 cirq/_doc.py,sha256=yDyWUD_2JDS0gShfGRb-rdqRt9-WeL7DhkqX7np0Nko,2879
 cirq/_import.py,sha256=cfocxtT1BJ4HkfZ-VO8YyIhPP-xfqHDkLrzz6eeO5U0,8421
 cirq/_import_test.py,sha256=6K_v0riZJXOXUphHNkGA8MY-JcmGlezFaGmvrNhm3OQ,1015
-cirq/_version.py,sha256=ej5Fg_x84YN77p1cdtLyrTatULWldkQMY3wr9uVW6Ao,1206
-cirq/_version_test.py,sha256=SLvxISQ8V1GpwYAJkTGScRjoLHudE0Z08eAlyTEUacU,147
+cirq/_version.py,sha256=hTB6q5KLlQenUB_P9SCUo9H4op-PatyQxgikmCkf8U0,1206
+cirq/_version_test.py,sha256=Yt9Dr26wo6Uhn7hGIoGN0kOoBnN7KUfd5ymMAIBGxCE,147
 cirq/conftest.py,sha256=X7yLFL8GLhg2CjPw0hp5e_dGASfvHx1-QT03aUbhKJw,1168
 cirq/json_resolver_cache.py,sha256=-4KqEEYb6aps-seafnFTHTp3SZc0D8mr4O-pCKIajn8,13653
 cirq/py.typed,sha256=VFSlmh_lNwnaXzwY-ZuW-C2Ws5PkuDoVgBdNCs0jXJE,63
@@ -1046,7 +1046,7 @@ 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=QjYovkmQHRi7D6nPHkWrszVBM_1__thvOXG4wcYvX_E,6905
+cirq/transformers/__init__.py,sha256=8k80MycPGQzZdJztV4eDpDAo6qpV7SA7GbG2vFiBAcg,7006
 cirq/transformers/align.py,sha256=XmARu30-wwsWMB-3z0K8WJ7zyErrdsTtgXgMDm4co6A,3355
 cirq/transformers/align_test.py,sha256=AI26PxERbtJDfjKtoz17xVAKn6TSr-9pyImqfr5LJWc,7686
 cirq/transformers/drop_empty_moments.py,sha256=j7zAoEsPs9JQ_zYpQX2yEvfiViDiYejnaOJ9GsXNbx4,1548
@@ -1081,6 +1081,8 @@ cirq/transformers/stratify.py,sha256=jfZEQuKv1YT8RdtcbGNsUNp4cs0WzZoiet-e5zwfLTc
 cirq/transformers/stratify_test.py,sha256=17ic2VAUPEGuPG2o5j98yDxQ2j2J_PN3EsPsfh5xwUk,15220
 cirq/transformers/synchronize_terminal_measurements.py,sha256=uh3u53xLjQLyZvh6KY-oOk_i6j8VveMeOi_zGdi748I,3856
 cirq/transformers/synchronize_terminal_measurements_test.py,sha256=VTiw5S3s_Y31qR7ME8Mzv50LdJ_6M3DOtgwvtziQzPI,7742
+cirq/transformers/tag_transformers.py,sha256=7Y5w6sKjT5Ccevwu3EysSMLZ4mZvW9jM2wcAFDqBOnU,3510
+cirq/transformers/tag_transformers_test.py,sha256=qyA9Z6lVk5lRnml6x63EREMUeyCAnxVfRQeA9KN4a0o,3403
 cirq/transformers/transformer_api.py,sha256=zH4suvb0iLPIJ_znCIpJGag4GSiycdPOk2nbbS14C1w,16961
 cirq/transformers/transformer_api_test.py,sha256=YBkIX-R6vYeQz1Y_sqpzDlvNYszEtfvkegoA8dAVVVc,13286
 cirq/transformers/transformer_primitives.py,sha256=q88fl6KGdJvx5mZqnorZv4oR92JK1k6Jm2JZBbjx4Ms,36642
@@ -1212,8 +1214,8 @@ cirq/work/sampler.py,sha256=b7O3B8bc77KQb8ReLx7qeF8owP1Qwb5_I-RwC6-M_C8,19118
 cirq/work/sampler_test.py,sha256=TBJm3gepuOURwklJTXNdqj0thvdqKUvrZwZqdytJxNY,13313
 cirq/work/zeros_sampler.py,sha256=vHCfqkXmUcPkaDuKHlY-UQ71dUHVroEtm_XW51mZpHs,2390
 cirq/work/zeros_sampler_test.py,sha256=TR3AXYSfg3ETpeaEtrmE-GgZsPtfZkUZ36kyH9JquJk,3313
-cirq_core-1.6.0.dev20250507223949.dist-info/licenses/LICENSE,sha256=tAkwu8-AdEyGxGoSvJ2gVmQdcicWw3j1ZZueVV74M-E,11357
-cirq_core-1.6.0.dev20250507223949.dist-info/METADATA,sha256=N8AvAV22NN2JNnFf298hHfQGPHnVd4I2-Yz1_DamQYk,4908
-cirq_core-1.6.0.dev20250507223949.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-cirq_core-1.6.0.dev20250507223949.dist-info/top_level.txt,sha256=Sz9iOxHU0IEMLSFGwiwOCaN2e9K-jFbBbtpPN1hB73g,5
-cirq_core-1.6.0.dev20250507223949.dist-info/RECORD,,
+cirq_core-1.6.0.dev20250507225534.dist-info/licenses/LICENSE,sha256=tAkwu8-AdEyGxGoSvJ2gVmQdcicWw3j1ZZueVV74M-E,11357
+cirq_core-1.6.0.dev20250507225534.dist-info/METADATA,sha256=t80B3z06_iMsNyOWwLvSh4xNe_TlVBNQ6tOCihY7rkA,4908
+cirq_core-1.6.0.dev20250507225534.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+cirq_core-1.6.0.dev20250507225534.dist-info/top_level.txt,sha256=Sz9iOxHU0IEMLSFGwiwOCaN2e9K-jFbBbtpPN1hB73g,5
+cirq_core-1.6.0.dev20250507225534.dist-info/RECORD,,