clean up import

Author: kaihsin

src/bloqade/qasm2/dialects/core/__init__.py

@@ -1,11 +1,11 @@
 from . import _emit as _emit, address as address, _typeinfer as _typeinfer
 from .stmts import (
-    QRegNew as QRegNew,
-    QRegGet as QRegGet,
-    CRegNew as CRegNew,
-    CRegGet as CRegGet,
     Reset as Reset,
-    Measure as Measure,
     CRegEq as CRegEq,
+    CRegGet as CRegGet,
+    CRegNew as CRegNew,
+    Measure as Measure,
+    QRegGet as QRegGet,
+    QRegNew as QRegNew,
 )
 from ._dialect import dialect as dialect

src/bloqade/qasm2/dialects/expr/__init__.py

@@ -1,20 +1,20 @@
 from . import _emit as _emit, _interp as _interp, _from_python as _from_python
 from .stmts import (
-    GateFunction as GateFunction,
-    ConstFloat as ConstFloat,
-    ConstInt as ConstInt,
-    ConstPI as ConstPI,
-    Neg as Neg,
-    Sin as Sin,
+    Add as Add,
     Cos as Cos,
-    Tan as Tan,
+    Div as Div,
     Exp as Exp,
     Log as Log,
-    Sqrt as Sqrt,
-    Add as Add,
-    Sub as Sub,
     Mul as Mul,
-    Div as Div,
+    Neg as Neg,
     Pow as Pow,
+    Sin as Sin,
+    Sub as Sub,
+    Tan as Tan,
+    Sqrt as Sqrt,
+    ConstPI as ConstPI,
+    ConstInt as ConstInt,
+    ConstFloat as ConstFloat,
+    GateFunction as GateFunction,
 )
 from ._dialect import dialect as dialect

src/bloqade/qasm2/dialects/uop/__init__.py

@@ -1,41 +1,40 @@
-from . import _emit as _emit, stmts as stmts
+from . import _emit as _emit, stmts as stmts, schedule as schedule
 from .stmts import (
-    UGate as UGate,
+    CH as CH,
+    CU as CU,
     CX as CX,
-    Barrier as Barrier,
-    Id as Id,
-    H as H,
-    X as X,
-    Y as Y,
-    Z as Z,
-    S as S,
-    Sdag as Sdag,
-    SX as SX,
-    SXdag as SXdag,
-    T as T,
-    Tdag as Tdag,
+    CY as CY,
+    CZ as CZ,
     RX as RX,
     RY as RY,
     RZ as RZ,
+    SX as SX,
     U1 as U1,
     U2 as U2,
-    CZ as CZ,
-    CY as CY,
-    CSX as CSX,
-    Swap as Swap,
-    CH as CH,
     CCX as CCX,
-    CSwap as CSwap,
     CRX as CRX,
     CRY as CRY,
     CRZ as CRZ,
+    CSX as CSX,
     CU1 as CU1,
     CU3 as CU3,
-    CU as CU,
     RXX as RXX,
     RZZ as RZZ,
+    H as H,
+    S as S,
+    T as T,
+    X as X,
+    Y as Y,
+    Z as Z,
+    Id as Id,
+    Sdag as Sdag,
+    Swap as Swap,
+    Tdag as Tdag,
+    CSwap as CSwap,
+    SXdag as SXdag,
+    UGate as UGate,
+    Barrier as Barrier,
     SingleQubitGate as SingleQubitGate,
     TwoQubitCtrlGate as TwoQubitCtrlGate,
 )
 from ._dialect import dialect as dialect
-from . import schedule as schedule

src/bloqade/qasm2/rewrite/desugar.py

@@ -5,8 +5,8 @@
 from kirin.rewrite import abc, walk
 from kirin.dialects import py
 
-from bloqade.qasm2.dialects import core
 from bloqade.qasm2 import types
+from bloqade.qasm2.dialects import core
 
 
 class IndexingDesugarRule(abc.RewriteRule):

src/bloqade/squin/groups.py

@@ -44,4 +44,3 @@ def run_pass(method):
         py_mult_to_mult_pass(method)
 
     return run_pass
-

src/bloqade/squin/noise/__init__.py

@@ -1,10 +1,8 @@
 from . import stmts as stmts
-from ._dialect import dialect as dialect    
+from ._dialect import dialect as dialect
 from ._wrapper import (
-    pauli_channel as pauli_channel,
     pp_error as pp_error,
     depolarize as depolarize,
-    pauli_channel as pauli_channel,     
     qubit_loss as qubit_loss,
+    pauli_channel as pauli_channel,
 )
-

src/bloqade/squin/noise/_wrapper.py

@@ -1,8 +1,10 @@
-from kirin.lowering import wraps 
+from kirin.lowering import wraps
 
 from bloqade.squin.op.types import Op
+
 from . import stmts
 
+
 @wraps(stmts.PauliError)
 def pauli_error(basis: Op, p: float) -> Op: ...
 

src/bloqade/squin/op/__init__.py

@@ -1,36 +1,36 @@
 from . import stmts as stmts, types as types, rewrite as rewrite
+from .stdlib import (
+    ch as ch,
+    cx as cx,
+    cy as cy,
+    cz as cz,
+    rx as rx,
+    ry as ry,
+    rz as rz,
+    cphase as cphase,
+)
 from .traits import Unitary as Unitary, MaybeUnitary as MaybeUnitary
 from ._dialect import dialect as dialect
 from ._wrapper import (
-    kron as kron,
-    mult as mult,
-    scale as scale,
-    adjoint as adjoint,
-    control as control,
-    identity as identity,
-    rot as rot,
-    shift as shift,
-    phase as phase,
-    x as x,
-    y as y,
-    z as z, 
     h as h,
     s as s,
     t as t,
+    u as u,
+    x as x,
+    y as y,
+    z as z,
     p0 as p0,
     p1 as p1,
+    rot as rot,
+    kron as kron,
+    mult as mult,
+    phase as phase,
+    scale as scale,
+    shift as shift,
     spin_n as spin_n,
     spin_p as spin_p,
-    u as u,
+    adjoint as adjoint,
+    control as control,
+    identity as identity,
     pauli_string as pauli_string,
 )
-from .stdlib import (
-    rx as rx,
-    ry as ry,
-    rz as rz,
-    cx as cx,
-    cz as cz,
-    cy as cy,
-    ch as ch,
-    cphase as cphase,
-)

src/bloqade/squin/op/_wrapper.py

@@ -1,6 +1,8 @@
 from kirin.lowering import wraps
+
 from . import stmts, types
 
+
 @wraps(stmts.Kron)
 def kron(lhs: types.Op, rhs: types.Op) -> types.Op: ...
 
@@ -97,4 +99,3 @@ def u(theta: float, phi: float, lam: float) -> types.Op: ...
 
 @wraps(stmts.PauliString)
 def pauli_string(*, string: str) -> types.Op: ...
-

src/bloqade/squin/op/stdlib.py

@@ -1,17 +1,10 @@
-
 from kirin import ir
-from kirin.prelude import structural_no_opt    
-from ._dialect import dialect
+from kirin.prelude import structural_no_opt
+
 from . import types
-from ._wrapper import (
-    rot, 
-    x,
-    y,
-    z,
-    h,
-    phase,
-    control
-)
+from ._dialect import dialect
+from ._wrapper import h, x, y, z, rot, phase, control
+
 
 @ir.dialect_group(structural_no_opt.add(dialect))
 def op(self):