cirq.transform_op_tree

Maps transformation functions onto the nodes of an OP_TREE.

cirq.transform_op_tree(
    root: cirq.OP_TREE,
    op_transformation: Callable[[cirq.Operation], cirq.OP_TREE] = (lambda e: e),
    iter_transformation: Callable[[Iterable[OP_TREE]], OP_TREE] = (lambda e: e),
    preserve_moments: bool = False
) -> cirq.OP_TREE

root The operation or tree of operations to transform.
op_transformation How to transform the operations (i.e. leaves).
iter_transformation How to transform the iterables (i.e. internal nodes).
preserve_moments Whether to leave Moments alone. If True, the transformation functions will not be applied to Moments or the operations within them.

A transformed operation tree.

TypeError root isn't a valid OP_TREE.