runing linter

weinbe58 · weinbe58 · commit acfee7529506 · 2025-06-17T10:18:32.000-04:00
diff --git a/demo/log_depth_ghz.py b/demo/log_depth_ghz.py
@@ -2,12 +2,11 @@
 from typing import Any, Literal, TypeVar
 
 from bloqade.geometry.dialects import grid
-from kirin.dialects import ilist
-
 from bloqade.qourier import action, gate, init, measure, schedule
 from bloqade.qourier.dialects import spec
 from bloqade.qourier.prelude import move, tweezer
 from bloqade.qourier.visualizer import MatplotlibRenderer, PathVisualizer
+from kirin.dialects import ilist
 
 NMove = TypeVar("NMove")
 
diff --git a/src/bloqade/shuttle/analysis/zone/analysis.py b/src/bloqade/shuttle/analysis/zone/analysis.py
@@ -1,11 +1,10 @@
 from dataclasses import dataclass, field
 
 from bloqade.geometry.dialects import grid
+from bloqade.qourier.dialects.spec import Spec
 from kirin import interp, ir
 from kirin.analysis.forward import Forward, ForwardFrame
 
-from bloqade.qourier.dialects.spec import Spec
-
 from .lattice import SpecZone, UnknownZone, Zone
 
 
diff --git a/src/bloqade/shuttle/analysis/zone/impl/spec.py b/src/bloqade/shuttle/analysis/zone/impl/spec.py
@@ -1,8 +1,7 @@
+from bloqade.qourier.dialects import spec
 from kirin.analysis.forward import ForwardFrame
 from kirin.interp import MethodTable, impl
 
-from bloqade.qourier.dialects import spec
-
 from .. import analysis, lattice
 
 
diff --git a/src/bloqade/shuttle/codegen/taskgen.py b/src/bloqade/shuttle/codegen/taskgen.py
@@ -3,13 +3,12 @@
 from typing import Any, ClassVar, Dict, Optional
 
 from bloqade.geometry.dialects import grid
+from bloqade.qourier.dialects import action
 from kirin.dialects import func, ilist
 from kirin.interp import Frame, Interpreter, InterpreterError, MethodTable, impl
 from kirin.ir.method import Method
 from typing_extensions import Self
 
-from bloqade.qourier.dialects import action
-
 
 class AbstractAction(abc.ABC):
     @abc.abstractmethod
diff --git a/src/bloqade/shuttle/dialects/action/stmts.py b/src/bloqade/shuttle/dialects/action/stmts.py
@@ -1,4 +1,5 @@
 from bloqade.geometry.dialects.grid import types as zone_types
+from bloqade.qourier.dialects.action._dialect import dialect
 from kirin import ir, lowering, types
 from kirin.decl import info, statement
 from kirin.dialects import func, ilist
@@ -12,8 +13,6 @@
 )
 from kirin.print.printer import Printer
 
-from bloqade.qourier.dialects.action._dialect import dialect
-
 
 class TweezerFuncOpCallableInterface(CallableStmtInterface["TweezerFunction"]):
 
diff --git a/src/bloqade/shuttle/dialects/measure/_interface.py b/src/bloqade/shuttle/dialects/measure/_interface.py
@@ -1,9 +1,8 @@
 import typing
 
 from bloqade.geometry.dialects import grid
-from kirin.lowering import wraps
-
 from bloqade.qourier.dialects.measure.types import MeasurementArray
+from kirin.lowering import wraps
 
 from .stmts import Measure
 
diff --git a/src/bloqade/shuttle/dialects/path/concrete.py b/src/bloqade/shuttle/dialects/path/concrete.py
@@ -1,3 +1,6 @@
+from bloqade.qourier.codegen import TraceInterpreter, reverse_path
+from bloqade.qourier.dialects import schedule
+from bloqade.qourier.dialects.path import dialect, stmts, types
 from kirin.interp import (
     Frame,
     Interpreter,
@@ -6,10 +9,6 @@
     impl,
 )
 
-from bloqade.qourier.codegen import TraceInterpreter, reverse_path
-from bloqade.qourier.dialects import schedule
-from bloqade.qourier.dialects.path import dialect, stmts, types
-
 
 @dialect.register
 class PathInterpreter(MethodTable):
diff --git a/src/bloqade/shuttle/dialects/path/stmts.py b/src/bloqade/shuttle/dialects/path/stmts.py
@@ -1,9 +1,8 @@
-from kirin import decl, ir
-from kirin.decl import info
-
 from bloqade.qourier.dialects import schedule
 from bloqade.qourier.dialects.path._dialect import dialect
 from bloqade.qourier.dialects.path.types import PathType
+from kirin import decl, ir
+from kirin.decl import info
 
 
 @decl.statement(dialect=dialect)
diff --git a/src/bloqade/shuttle/dialects/path/types.py b/src/bloqade/shuttle/dialects/path/types.py
@@ -1,11 +1,10 @@
 from dataclasses import dataclass, field
 from typing import Any
 
+from bloqade.qourier.codegen.taskgen import AbstractAction
 from kirin import types
 from kirin.dialects import ilist
 
-from bloqade.qourier.codegen.taskgen import AbstractAction
-
 
 @dataclass
 class Path:
diff --git a/src/bloqade/shuttle/dialects/schedule/concrete.py b/src/bloqade/shuttle/dialects/schedule/concrete.py
@@ -1,8 +1,7 @@
-from kirin import ir
-from kirin.interp import Frame, Interpreter, InterpreterError, MethodTable, impl
-
 from bloqade.qourier.dialects.schedule import stmts, types
 from bloqade.qourier.dialects.schedule._dialect import dialect
+from kirin import ir
+from kirin.interp import Frame, Interpreter, InterpreterError, MethodTable, impl
 
 
 @dialect.register
diff --git a/src/bloqade/shuttle/dialects/schedule/stmts.py b/src/bloqade/shuttle/dialects/schedule/stmts.py
@@ -1,12 +1,11 @@
+from bloqade.qourier.dialects.schedule._dialect import dialect
+from bloqade.qourier.dialects.schedule.types import DeviceFunctionType
 from kirin import ir, lowering, types
 from kirin.decl import info, statement
 from kirin.dialects import func, ilist
 from kirin.ir.exception import ValidationError
 from kirin.lowering.python.traits import FromPythonWithSingleItem
 
-from bloqade.qourier.dialects.schedule._dialect import dialect
-from bloqade.qourier.dialects.schedule.types import DeviceFunctionType
-
 NxTones = types.TypeVar("NxTones")
 NyTones = types.TypeVar("NyTones")
 
diff --git a/src/bloqade/shuttle/dialects/spec/stmts.py b/src/bloqade/shuttle/dialects/spec/stmts.py
@@ -1,9 +1,8 @@
 from bloqade.geometry.dialects import grid
+from bloqade.qourier.dialects.spec.dialect import dialect
 from kirin import ir, lowering, types
 from kirin.decl import info, statement
 
-from bloqade.qourier.dialects.spec.dialect import dialect
-
 
 @statement(dialect=dialect)
 class GetStaticTrap(ir.Statement):
diff --git a/src/bloqade/shuttle/passes/hint_zone.py b/src/bloqade/shuttle/passes/hint_zone.py
@@ -1,13 +1,12 @@
 from dataclasses import dataclass
 
+from bloqade.qourier.analysis.zone import Zone, ZoneAnalysis
+from bloqade.qourier.dialects import spec
 from kirin import ir
 from kirin.passes import Pass
 from kirin.rewrite import Walk
 from kirin.rewrite.abc import RewriteResult, RewriteRule
 
-from bloqade.qourier.analysis.zone import Zone, ZoneAnalysis
-from bloqade.qourier.dialects import spec
-
 
 @dataclass
 class ZoneHintRewrite(RewriteRule):
diff --git a/src/bloqade/shuttle/passes/inject_spec.py b/src/bloqade/shuttle/passes/inject_spec.py
@@ -1,14 +1,13 @@
 from dataclasses import dataclass, field
 
+from bloqade.qourier.dialects import spec
 from kirin import rewrite
 from kirin.dialects.py import Constant
 from kirin.ir.method import Method
 from kirin.ir.nodes.stmt import Statement
 from kirin.passes import Fold, HintConst, Pass
 from kirin.rewrite.abc import RewriteResult, RewriteRule
 
-from bloqade.qourier.dialects import spec
-
 
 @dataclass
 class InjectStaticTrapsRule(RewriteRule):
diff --git a/src/bloqade/shuttle/passes/schedule2path.py b/src/bloqade/shuttle/passes/schedule2path.py
@@ -1,3 +1,8 @@
+from bloqade.qourier.rewrite.schedule2path import (
+    Canonicalize,
+    RewriteDeviceCall,
+    RewriteScheduleRegion,
+)
 from kirin import ir
 from kirin.passes import Pass
 from kirin.rewrite import (
@@ -8,12 +13,6 @@
     Walk,
 )
 
-from bloqade.qourier.rewrite.schedule2path import (
-    Canonicalize,
-    RewriteDeviceCall,
-    RewriteScheduleRegion,
-)
-
 
 class ScheduleToPath(Pass):
     """Pass to convert schedule dialect to path dialect."""
diff --git a/src/bloqade/shuttle/prelude.py b/src/bloqade/shuttle/prelude.py
@@ -1,12 +1,4 @@
 from bloqade.geometry.dialects import grid
-from bloqade.squin import op, qubit
-from kirin import ir
-from kirin.dialects import func, ilist
-from kirin.passes import Default, Fold, TypeInfer
-from kirin.prelude import structural
-from kirin.rewrite import Walk
-from kirin.rewrite.chain import Chain
-
 from bloqade.qourier.dialects import (
     action,
     atom,
@@ -20,6 +12,13 @@
 from bloqade.qourier.passes.inject_spec import InjectSpecsPass
 from bloqade.qourier.passes.schedule2path import ScheduleToPath
 from bloqade.qourier.rewrite.desugar import DesugarTurnOffRewrite, DesugarTurnOnRewrite
+from bloqade.squin import op, qubit
+from kirin import ir
+from kirin.dialects import func, ilist
+from kirin.passes import Default, Fold, TypeInfer
+from kirin.prelude import structural
+from kirin.rewrite import Walk
+from kirin.rewrite.chain import Chain
 
 
 @ir.dialect_group(structural.union([spec, grid, atom, gate, op, qubit]))
diff --git a/src/bloqade/shuttle/rewrite/desugar.py b/src/bloqade/shuttle/rewrite/desugar.py
@@ -1,8 +1,7 @@
+from bloqade.qourier.dialects import action
 from kirin import ir, types
 from kirin.rewrite import abc
 
-from bloqade.qourier.dialects import action
-
 
 class DesugarTurnOnRewrite(abc.RewriteRule):
 
diff --git a/src/bloqade/shuttle/rewrite/schedule2path.py b/src/bloqade/shuttle/rewrite/schedule2path.py
@@ -1,9 +1,8 @@
+from bloqade.qourier.dialects import path, schedule
 from kirin import ir
 from kirin.dialects import func
 from kirin.rewrite import abc
 
-from bloqade.qourier.dialects import path, schedule
-
 
 class RewriteDeviceCall(abc.RewriteRule):
     def rewrite_Statement(self, node: ir.Statement) -> abc.RewriteResult:
diff --git a/src/bloqade/shuttle/stdlib/moves.py b/src/bloqade/shuttle/stdlib/moves.py
@@ -1,10 +1,9 @@
 from typing import Any, TypeVar
 
 from bloqade.geometry.dialects import grid
-from kirin.dialects import ilist
-
 from bloqade.qourier import action, gate, schedule, spec
 from bloqade.qourier.prelude import move, tweezer
+from kirin.dialects import ilist
 
 NumX = TypeVar("NumX")
 NumY = TypeVar("NumY")
diff --git a/src/bloqade/shuttle/stdlib/spec.py b/src/bloqade/shuttle/stdlib/spec.py
@@ -1,7 +1,6 @@
 from itertools import repeat
 
 from bloqade.geometry.dialects import grid
-
 from bloqade.qourier.dialects import spec
 
 
diff --git a/src/bloqade/shuttle/stdlib/waypoints.py b/src/bloqade/shuttle/stdlib/waypoints.py
@@ -1,10 +1,9 @@
 from typing import TypeVar
 
 from bloqade.geometry.dialects import grid
-from kirin.dialects import ilist
-
 from bloqade.qourier import action, schedule
 from bloqade.qourier.prelude import move, tweezer
+from kirin.dialects import ilist
 
 NumX = TypeVar("NumX")
 NumY = TypeVar("NumY")
diff --git a/src/bloqade/shuttle/visualizer/impl/gate.py b/src/bloqade/shuttle/visualizer/impl/gate.py
@@ -1,7 +1,5 @@
 from typing import TypeVar
 
-from kirin import interp
-
 from bloqade.qourier.dialects.gate import (
     GlobalR,
     GlobalRz,
@@ -12,6 +10,7 @@
 )
 from bloqade.qourier.visualizer.interp import PathVisualizer
 from bloqade.qourier.visualizer.renderers import RendererInterface
+from kirin import interp
 
 
 @dialect.register(key="path.visualizer")
diff --git a/src/bloqade/shuttle/visualizer/impl/init.py b/src/bloqade/shuttle/visualizer/impl/init.py
@@ -1,10 +1,9 @@
 from typing import TypeVar
 
-from kirin import interp
-
 from bloqade.qourier.dialects.init import Fill, dialect
 from bloqade.qourier.visualizer.interp import PathVisualizer
 from bloqade.qourier.visualizer.renderers import RendererInterface
+from kirin import interp
 
 
 @dialect.register(key="path.visualizer")
diff --git a/src/bloqade/shuttle/visualizer/impl/measure.py b/src/bloqade/shuttle/visualizer/impl/measure.py
@@ -1,13 +1,12 @@
 from typing import TypeVar
 
-from kirin import interp
-
 from bloqade.qourier.dialects.measure import (
     Measure,
     dialect,
 )
 from bloqade.qourier.visualizer.interp import PathVisualizer
 from bloqade.qourier.visualizer.renderers import RendererInterface
+from kirin import interp
 
 
 class MeasureResultRuntime:
diff --git a/src/bloqade/shuttle/visualizer/impl/path.py b/src/bloqade/shuttle/visualizer/impl/path.py
@@ -1,11 +1,10 @@
 from dataclasses import dataclass
 from typing import TypeVar
 
-from kirin import interp
-
 from bloqade.qourier.dialects.path import Parallel, Path, Play, dialect
 from bloqade.qourier.visualizer.interp import PathVisualizer
 from bloqade.qourier.visualizer.renderers import RendererInterface
+from kirin import interp
 
 
 @dataclass
diff --git a/src/bloqade/shuttle/visualizer/interp.py b/src/bloqade/shuttle/visualizer/interp.py
@@ -1,11 +1,10 @@
 from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, Generic, TypeVar
 
-from kirin import interp, ir
-from typing_extensions import Self
-
 from bloqade.qourier.dialects import spec
 from bloqade.qourier.passes import inject_spec
+from kirin import interp, ir
+from typing_extensions import Self
 
 if TYPE_CHECKING:
     from bloqade.qourier.visualizer.renderers import RendererInterface
diff --git a/src/bloqade/shuttle/visualizer/renderers/interface.py b/src/bloqade/shuttle/visualizer/renderers/interface.py
@@ -1,7 +1,6 @@
 from abc import ABC, abstractmethod
 
 from bloqade.geometry.dialects import grid
-
 from bloqade.qourier.dialects import path
 
 
diff --git a/src/bloqade/shuttle/visualizer/renderers/matplotlib.py b/src/bloqade/shuttle/visualizer/renderers/matplotlib.py
@@ -3,14 +3,13 @@
 
 import numpy as np
 from bloqade.geometry.dialects.grid.types import Grid
+from bloqade.qourier.codegen import taskgen
+from bloqade.qourier.dialects import path
 from matplotlib import pyplot as plt
 from matplotlib.artist import Artist
 from matplotlib.axes import Axes
 from matplotlib.widgets import Button
 
-from bloqade.qourier.codegen import taskgen
-from bloqade.qourier.dialects import path
-
 from .interface import RendererInterface
 
 
diff --git a/test/action/test_move.py b/test/action/test_move.py
@@ -1,5 +1,4 @@
 from bloqade.geometry.dialects import grid
-
 from bloqade.qourier.dialects import action
 from bloqade.qourier.prelude import tweezer
 
diff --git a/test/analysis/test_zone.py b/test/analysis/test_zone.py
diff --git a/test/atom/test_wrapper.py b/test/atom/test_wrapper.py
diff --git a/test/codegen/test_taskgen.py b/test/codegen/test_taskgen.py
diff --git a/test/passes/test_fold.py b/test/passes/test_fold.py
diff --git a/test/passes/test_inject_spec.py b/test/passes/test_inject_spec.py
diff --git a/test/stdlib/test_spec.py b/test/stdlib/test_spec.py
diff --git a/test/unit/codegen/test_taskgen.py b/test/unit/codegen/test_taskgen.py
diff --git a/test/unit/dialects/schedule/test_concrete.py b/test/unit/dialects/schedule/test_concrete.py
diff --git a/test/unit/rewrite/test_canonicalize.py b/test/unit/rewrite/test_canonicalize.py
diff --git a/test/unit/rewrite/test_gen.py b/test/unit/rewrite/test_gen.py
diff --git a/test/unit/rewrite/test_region.py b/test/unit/rewrite/test_region.py
diff --git a/test/unit/rewrite/utils.py b/test/unit/rewrite/utils.py
