Move qubitization_walk_operator.py to a new module qubitization/ (#1001)

* Move qubitization_walk_operator.py to a new module qubitization/

* Update references in notebook

* Revert qrom.ipynb

* Fix merge conflict in resolve_dict

Author: tanujkhattar

dev_tools/autogenerate-bloqs-notebooks-v2.py

@@ -92,7 +92,7 @@
 import qualtran.bloqs.qft.approximate_qft
 import qualtran.bloqs.qft.two_bit_ffft
 import qualtran.bloqs.qsp.generalized_qsp
-import qualtran.bloqs.qubitization_walk_operator
+import qualtran.bloqs.qubitization.qubitization_walk_operator
 import qualtran.bloqs.reflections
 import qualtran.bloqs.rotations.phasing_via_cost_function
 import qualtran.bloqs.rotations.quantum_variable_rotation
@@ -459,9 +459,9 @@
     ),
     NotebookSpecV2(
         title='Qubitization Walk Operator',
-        module=qualtran.bloqs.qubitization_walk_operator,
+        module=qualtran.bloqs.qubitization.qubitization_walk_operator,
         bloq_specs=[
-            qualtran.bloqs.qubitization_walk_operator._QUBITIZATION_WALK_DOC,
+            qualtran.bloqs.qubitization.qubitization_walk_operator._QUBITIZATION_WALK_DOC,
             qualtran.bloqs.block_encoding.lcu_select_and_prepare._SELECT_ORACLE_DOC,
             qualtran.bloqs.block_encoding.lcu_select_and_prepare._PREPARE_ORACLE_DOC,
         ],

docs/bloqs/index.rst

@@ -84,7 +84,7 @@ Bloqs Library
     qft/approximate_qft.ipynb
     phase_estimation/lp_resource_state.ipynb
     phase_estimation/text_book_qpe.ipynb
-    qubitization_walk_operator.ipynb
+    qubitization/qubitization_walk_operator.ipynb
     phase_estimation/qubitization_qpe.ipynb
 
 .. toctree::

qualtran/bloqs/chemistry/sparse/walk_operator.py

@@ -14,7 +14,7 @@
 from numpy.typing import NDArray
 
 from qualtran.bloqs.chemistry.sparse import PrepareSparse, SelectSparse
-from qualtran.bloqs.qubitization_walk_operator import QubitizationWalkOperator
+from qualtran.bloqs.qubitization.qubitization_walk_operator import QubitizationWalkOperator
 
 
 def get_walk_operator_for_sparse_chem_ham(

qualtran/bloqs/chemistry/thc/walk_operator.py

@@ -16,7 +16,7 @@
 from numpy.typing import NDArray
 
 from qualtran.bloqs.chemistry.thc import PrepareTHC, SelectTHC
-from qualtran.bloqs.qubitization_walk_operator import QubitizationWalkOperator
+from qualtran.bloqs.qubitization.qubitization_walk_operator import QubitizationWalkOperator
 
 
 def get_walk_operator_for_thc_ham(

qualtran/bloqs/for_testing/qubitization_walk_test.py

@@ -22,7 +22,7 @@
 from qualtran import Signature
 from qualtran.bloqs.block_encoding.lcu_select_and_prepare import PrepareOracle
 from qualtran.bloqs.multiplexers.select_pauli_lcu import SelectPauliLCU
-from qualtran.bloqs.qubitization_walk_operator import QubitizationWalkOperator
+from qualtran.bloqs.qubitization.qubitization_walk_operator import QubitizationWalkOperator
 from qualtran.bloqs.state_preparation import PrepareUniformSuperposition
 from qualtran.symbolics import SymbolicFloat
 

qualtran/bloqs/for_testing/random_select_and_prepare.py

@@ -23,7 +23,7 @@
 from qualtran._infra.gate_with_registers import SpecializedSingleQubitControlledGate
 from qualtran.bloqs.block_encoding.lcu_select_and_prepare import PrepareOracle, SelectOracle
 from qualtran.bloqs.for_testing.matrix_gate import MatrixGate
-from qualtran.bloqs.qubitization_walk_operator import QubitizationWalkOperator
+from qualtran.bloqs.qubitization.qubitization_walk_operator import QubitizationWalkOperator
 
 
 @frozen

qualtran/bloqs/hamiltonian_simulation/hamiltonian_simulation_by_gqsp.py

@@ -20,7 +20,7 @@
 
 from qualtran import bloq_example, BloqDocSpec, GateWithRegisters, Signature, Soquet
 from qualtran.bloqs.qsp.generalized_qsp import GeneralizedQSP, scale_down_to_qsp_polynomial
-from qualtran.bloqs.qubitization_walk_operator import QubitizationWalkOperator
+from qualtran.bloqs.qubitization.qubitization_walk_operator import QubitizationWalkOperator
 from qualtran.linalg.jacobi_anger_approximations import (
     approx_exp_cos_by_jacobi_anger,
     degree_jacobi_anger_approximation,

qualtran/bloqs/hamiltonian_simulation/hamiltonian_simulation_by_gqsp_test.py

@@ -26,7 +26,7 @@
     check_polynomial_pair_on_random_points_on_unit_circle,
     verify_generalized_qsp,
 )
-from qualtran.bloqs.qubitization_walk_operator import QubitizationWalkOperator
+from qualtran.bloqs.qubitization.qubitization_walk_operator import QubitizationWalkOperator
 from qualtran.cirq_interop.t_complexity_protocol import TComplexity
 from qualtran.resource_counting import big_O
 from qualtran.symbolics import Shaped

qualtran/bloqs/hubbard_model.py

@@ -66,7 +66,7 @@
 from qualtran.bloqs.mod_arithmetic import ModAddK
 from qualtran.bloqs.multiplexers.apply_gate_to_lth_target import ApplyGateToLthQubit
 from qualtran.bloqs.multiplexers.selected_majorana_fermion import SelectedMajoranaFermion
-from qualtran.bloqs.qubitization_walk_operator import QubitizationWalkOperator
+from qualtran.bloqs.qubitization.qubitization_walk_operator import QubitizationWalkOperator
 from qualtran.bloqs.state_preparation.prepare_uniform_superposition import (
     PrepareUniformSuperposition,
 )

qualtran/bloqs/phase_estimation/qubitization_qpe.py

@@ -21,7 +21,7 @@
 from qualtran import Bloq, bloq_example, BloqDocSpec, GateWithRegisters, QFxp, Register, Signature
 from qualtran.bloqs.phase_estimation.lp_resource_state import LPResourceState
 from qualtran.bloqs.qft.qft_text_book import QFTTextBook
-from qualtran.bloqs.qubitization_walk_operator import QubitizationWalkOperator
+from qualtran.bloqs.qubitization.qubitization_walk_operator import QubitizationWalkOperator
 from qualtran.symbolics import ceil, is_symbolic, log2, pi, SymbolicFloat, SymbolicInt
 
 if TYPE_CHECKING: