fixing tests and package

Author: weinbe58

README.md

@@ -1 +1,3 @@
-# kirin-qourier
+# Bloqade-Shuttle
+
+SDK for simulation and running QuEra's neutral atom quantum computers with explicit shuttling.

demo/log_depth_ghz.py

@@ -2,12 +2,13 @@
 from typing import Any, Literal, TypeVar
 
 from bloqade.geometry.dialects import grid
-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
 
+from bloqade.shuttle import action, gate, init, measure, schedule
+from bloqade.shuttle.dialects import spec
+from bloqade.shuttle.prelude import move, tweezer
+from bloqade.shuttle.visualizer import MatplotlibRenderer, PathVisualizer
+
 NMove = TypeVar("NMove")
 
 

pyproject.toml

@@ -1,5 +1,5 @@
 [project]
-name = "bloqade-shuttle-internal" # This is the name of your package
+name = "bloqade-qourier" # This is the name of your package
 version = "0.6.0-dev"
 description = "Add your description here"
 authors = [

src/bloqade/shuttle/analysis/zone/analysis.py

@@ -1,10 +1,11 @@
 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.shuttle.dialects.spec import Spec
+
 from .lattice import SpecZone, UnknownZone, Zone
 
 

src/bloqade/shuttle/analysis/zone/impl/spec.py

@@ -1,7 +1,8 @@
-from bloqade.qourier.dialects import spec
 from kirin.analysis.forward import ForwardFrame
 from kirin.interp import MethodTable, impl
 
+from bloqade.shuttle.dialects import spec
+
 from .. import analysis, lattice
 
 

src/bloqade/shuttle/codegen/taskgen.py

@@ -3,12 +3,13 @@
 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.shuttle.dialects import action
+
 
 class AbstractAction(abc.ABC):
     @abc.abstractmethod

src/bloqade/shuttle/dialects/action/stmts.py

@@ -1,5 +1,4 @@
 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
@@ -13,6 +12,8 @@
 )
 from kirin.print.printer import Printer
 
+from bloqade.shuttle.dialects.action._dialect import dialect
+
 
 class TweezerFuncOpCallableInterface(CallableStmtInterface["TweezerFunction"]):
 

src/bloqade/shuttle/dialects/measure/_dialect.py

@@ -1,3 +1,3 @@
 from kirin import ir
 
-dialect = ir.Dialect("bloqade.qourier.measure")
+dialect = ir.Dialect("bloqade.shuttle.measure")

src/bloqade/shuttle/dialects/measure/_interface.py

@@ -1,9 +1,10 @@
 import typing
 
 from bloqade.geometry.dialects import grid
-from bloqade.qourier.dialects.measure.types import MeasurementArray
 from kirin.lowering import wraps
 
+from bloqade.shuttle.dialects.measure.types import MeasurementArray
+
 from .stmts import Measure
 
 NumX = typing.TypeVar("NumX", bound=int)

src/bloqade/shuttle/dialects/path/concrete.py

@@ -1,6 +1,3 @@
-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,
@@ -9,13 +6,17 @@
     impl,
 )
 
+from bloqade.shuttle.codegen import TraceInterpreter, reverse_path
+from bloqade.shuttle.dialects import schedule
+from bloqade.shuttle.dialects.path import dialect, stmts, types
+
 
 @dialect.register
 class PathInterpreter(MethodTable):
 
     @impl(stmts.Gen)
     def gen(self, interp: Interpreter, frame: Frame, stmt: stmts.Gen):
-        from bloqade.qourier.prelude import tweezer
+        from bloqade.shuttle.prelude import tweezer
 
         device_task = frame.get(stmt.device_task)
         if isinstance(device_task, schedule.DeviceFunction):