From e1dabca3e35e5205a3a9128fb682cd654a356a9d Mon Sep 17 00:00:00 2001 From: Anurudh Peduri Date: Wed, 30 Oct 2024 20:29:54 +0100 Subject: [PATCH] imports and mypy --- qualtran/bloqs/mcmt/specialized_ctrl.py | 15 ++------------- qualtran/bloqs/mcmt/specialized_ctrl_test.py | 9 +++++++-- .../qubitization/qubitization_walk_operator.py | 15 +++++++++++---- 3 files changed, 20 insertions(+), 19 deletions(-) diff --git a/qualtran/bloqs/mcmt/specialized_ctrl.py b/qualtran/bloqs/mcmt/specialized_ctrl.py index da228cc41..1e649665d 100644 --- a/qualtran/bloqs/mcmt/specialized_ctrl.py +++ b/qualtran/bloqs/mcmt/specialized_ctrl.py @@ -18,22 +18,11 @@ import attrs import numpy as np -from qualtran import ( - AddControlledT, - Adjoint, - Bloq, - BloqBuilder, - CompositeBloq, - CtrlSpec, - QBit, - Register, - Signature, - SoquetT, -) +from qualtran import Adjoint, Bloq, BloqBuilder, CompositeBloq, QBit, Register, Signature from qualtran.bloqs.bookkeeping import AutoPartition if TYPE_CHECKING: - from qualtran import AddControlledT, BloqBuilder, CtrlSpec, SoquetT + from qualtran import AddControlledT, CtrlSpec, SoquetT from qualtran._infra.controlled import ControlBit from qualtran.resource_counting import BloqCountDictT, SympySymbolAllocator diff --git a/qualtran/bloqs/mcmt/specialized_ctrl_test.py b/qualtran/bloqs/mcmt/specialized_ctrl_test.py index bf95a8bb6..fa385c177 100644 --- a/qualtran/bloqs/mcmt/specialized_ctrl_test.py +++ b/qualtran/bloqs/mcmt/specialized_ctrl_test.py @@ -183,7 +183,12 @@ def test_bloq_with_controlled_bloq(): def test_ctrl_adjoint(): assert TestAtom('a').adjoint().controlled() == CTestAtom('a').adjoint() - _, sigma = TestAtom('g').adjoint().controlled(CtrlSpec(cvs=[1, 1])).call_graph(keep=_keep_and) + _, sigma = ( + TestAtom('g') + .adjoint() + .controlled(ctrl_spec=CtrlSpec(cvs=[1, 1])) + .call_graph(keep=_keep_and) + ) assert sigma == {And(): 1, And().adjoint(): 1, CTestAtom('g').adjoint(): 1} _, sigma = CTestAtom('c').adjoint().controlled().call_graph(keep=_keep_and) @@ -191,7 +196,7 @@ def test_ctrl_adjoint(): for cv in [0, 1]: assert ( - AtomWithSpecializedControl().adjoint().controlled(CtrlSpec(cvs=(cv,))) + AtomWithSpecializedControl().adjoint().controlled(ctrl_spec=CtrlSpec(cvs=cv)) == AtomWithSpecializedControl(cv=cv).adjoint() ) diff --git a/qualtran/bloqs/qubitization/qubitization_walk_operator.py b/qualtran/bloqs/qubitization/qubitization_walk_operator.py index a752fdaba..9e270d8ec 100644 --- a/qualtran/bloqs/qubitization/qubitization_walk_operator.py +++ b/qualtran/bloqs/qubitization/qubitization_walk_operator.py @@ -33,8 +33,15 @@ import cirq import numpy as np -from qualtran import bloq_example, BloqBuilder, BloqDocSpec, Register, Signature, SoquetT -from qualtran._infra.gate_with_registers import GateWithRegisters, total_bits +from qualtran import ( + bloq_example, + BloqBuilder, + BloqDocSpec, + GateWithRegisters, + Register, + Signature, + SoquetT, +) from qualtran.bloqs.block_encoding.lcu_block_encoding import ( BlackBoxPrepare, LCUBlockEncoding, @@ -51,7 +58,7 @@ @attrs.frozen(cache_hash=True) -class QubitizationWalkOperator(GateWithRegisters): # type: ignore[misc] +class QubitizationWalkOperator(GateWithRegisters): r"""Construct a Szegedy Quantum Walk operator using LCU oracles SELECT and PREPARE. For a Hamiltonian $H = \sum_l w_l H_l$ (where coefficients $w_l > 0$ and $H_l$ are unitaries), @@ -130,7 +137,7 @@ def build_composite_bloq(self, bb: 'BloqBuilder', **soqs: 'SoquetT') -> dict[str return soqs def _circuit_diagram_info_(self, args: cirq.CircuitDiagramInfoArgs) -> cirq.CircuitDiagramInfo: - wire_symbols = ['W'] * total_bits(self.signature) + wire_symbols = ['W'] * self.signature.n_qubits() return cirq.CircuitDiagramInfo(wire_symbols=wire_symbols) @cached_property