setup isort

Author: Roger-luo

.pre-commit-config.yaml

@@ -14,7 +14,7 @@ repos:
     - id: isort
       name: isort (python)
 - repo: https://github.com/psf/black
-  rev: 24.10.0
+  rev: 25.1.0
   hooks:
     - id: black
 - repo: https://github.com/charliermarsh/ruff-pre-commit

pyproject.toml

@@ -51,6 +51,13 @@ packages = ["src/bloqade"]
 [tool.black]
 line-length = 88
 
+[tool.isort]
+profile = "black"
+combine_as_imports = true
+multi_line_output = 3
+length_sort = true
+src_paths = ["src/bloqade"]
+
 [tool.ruff]
 exclude = [
     ".bzr",

src/bloqade-analog/__init__.py

@@ -0,0 +1,67 @@
+try:
+    __import__("pkg_resources").declare_namespace(__name__)
+except ImportError:
+    __path__ = __import__("pkgutil").extend_path(__path__, __name__)
+
+import importlib.metadata
+
+import bloqade.analog.ir as _ir
+from bloqade.analog.ir import (
+    Literal,
+    Variable,
+    var,
+    cast,
+    start,
+    to_waveform as waveform,
+)
+from bloqade.analog.factory import (
+    linear,
+    constant,
+    rydberg_h,
+    get_capabilities,
+    piecewise_linear,
+    piecewise_constant,
+)
+from bloqade.analog.constants import RB_C6
+from bloqade.analog.serialize import load, save, dumps, loads
+
+__version__ = importlib.metadata.version("bloqade")
+
+
+def tree_depth(depth: int | None = None):
+    """Setting globally maximum depth for tree printing
+
+    If `depth=None`, return current depth.
+    If `depth` is provided, setting current depth to `depth`
+
+    Args:
+        depth (int, optional): the user specified depth. Defaults to None.
+
+    Returns:
+        int: current updated depth
+    """
+    if depth is not None:
+        _ir.tree_print.MAX_TREE_DEPTH = depth
+    return _ir.tree_print.MAX_TREE_DEPTH
+
+
+__all__ = [
+    "RB_C6",
+    "start",
+    "var",
+    "cast",
+    "Variable",
+    "Literal",
+    "piecewise_linear",
+    "piecewise_constant",
+    "linear",
+    "constant",
+    "set_print_depth",
+    "load",
+    "save",
+    "loads",
+    "dumps",
+    "rydberg_h",
+    "waveform",
+    "get_capabilities",
+]

src/bloqade/__init__.py renamed to src/bloqade/analog/__init__.py

@@ -3,22 +3,27 @@
 except ImportError:
     __path__ = __import__("pkgutil").extend_path(__path__, __name__)
 
-from bloqade.ir import var, cast, Variable, Literal, start
-from bloqade.ir import to_waveform as waveform
-from bloqade.serialize import load, save, loads, dumps
+import importlib.metadata
 
-from bloqade.factory import (
-    get_capabilities,
-    piecewise_linear,
-    piecewise_constant,
+import bloqade.analog.ir as _ir
+from bloqade.analog.ir import (
+    Literal,
+    Variable,
+    var,
+    cast,
+    start,
+    to_waveform as waveform,
+)
+from bloqade.analog.factory import (
     linear,
     constant,
     rydberg_h,
+    get_capabilities,
+    piecewise_linear,
+    piecewise_constant,
 )
-import bloqade.ir as _ir
-from bloqade.constants import RB_C6
-
-import importlib.metadata
+from bloqade.analog.constants import RB_C6
+from bloqade.analog.serialize import load, save, dumps, loads
 
 __version__ = importlib.metadata.version("bloqade")
 

src/bloqade/atom_arrangement.py renamed to src/bloqade/analog/atom_arrangement.py

@@ -1,16 +1,15 @@
-from bloqade.ir.location import (
-    AtomArrangement,
+from bloqade.analog.ir.location import (
+    Lieb,
     Chain,
+    Kagome,
     Square,
-    Rectangular,
     Honeycomb,
     Triangular,
-    Lieb,
-    Kagome,
+    Rectangular,
+    AtomArrangement,
     ListOfLocations,
 )
 
-
 __all__ = [
     "AtomArrangement",
     "Chain",

src/bloqade/builder/__init__.py renamed to src/bloqade/analog/builder/__init__.py

@@ -1,9 +1,10 @@
 from beartype import beartype
-from beartype.typing import List, Optional, Union
-from bloqade.ir.scalar import Variable
-from bloqade.builder.base import Builder
-from bloqade.builder.pragmas import Parallelizable
-from bloqade.builder.backend import BackendRoute
+from beartype.typing import List, Union, Optional
+
+from bloqade.analog.ir.scalar import Variable
+from bloqade.analog.builder.base import Builder
+from bloqade.analog.builder.backend import BackendRoute
+from bloqade.analog.builder.pragmas import Parallelizable
 
 
 class Args(Parallelizable, BackendRoute, Builder):

src/bloqade/builder/args.py renamed to src/bloqade/analog/builder/args.py

@@ -1,12 +1,14 @@
-from itertools import repeat
-from beartype.typing import Optional, List, Dict, Set, Sequence, Union
-from bloqade.builder.typing import ParamType
-from bloqade.builder.base import Builder
-from bloqade.builder.pragmas import Parallelizable, AddArgs, BatchAssignable
-from bloqade.builder.backend import BackendRoute
-from numbers import Real
 from decimal import Decimal
+from numbers import Real
+from itertools import repeat
+
 import numpy as np
+from beartype.typing import Set, Dict, List, Union, Optional, Sequence
+
+from bloqade.analog.builder.base import Builder
+from bloqade.analog.builder.typing import ParamType
+from bloqade.analog.builder.backend import BackendRoute
+from bloqade.analog.builder.pragmas import AddArgs, Parallelizable, BatchAssignable
 
 
 class CastParams:
@@ -69,7 +71,7 @@ class Assign(BatchAssignable, AddArgs, Parallelizable, BackendRoute, AssignBase)
     def __init__(
         self, assignments: Dict[str, ParamType], parent: Optional[Builder] = None
     ) -> None:
-        from bloqade.compiler.analysis.common.scan_variables import ScanVariables
+        from bloqade.analog.compiler.analysis.common.scan_variables import ScanVariables
 
         super().__init__(parent)
 
@@ -85,7 +87,7 @@ class BatchAssign(AddArgs, Parallelizable, BackendRoute, AssignBase):
     def __init__(
         self, assignments: Dict[str, List[ParamType]], parent: Optional[Builder] = None
     ) -> None:
-        from bloqade.compiler.analysis.common.scan_variables import ScanVariables
+        from bloqade.analog.compiler.analysis.common.scan_variables import ScanVariables
 
         super().__init__(parent)
 
@@ -120,7 +122,7 @@ def __init__(
         batch_params: Sequence[Dict[str, ParamType]],
         parent: Optional[Builder] = None,
     ) -> None:
-        from bloqade.compiler.analysis.common.scan_variables import ScanVariables
+        from bloqade.analog.compiler.analysis.common.scan_variables import ScanVariables
 
         super().__init__(parent)
 

src/bloqade/builder/assign.py renamed to src/bloqade/analog/builder/assign.py

@@ -1,6 +1,6 @@
-from bloqade.builder.backend.quera import QuEraService
-from bloqade.builder.backend.braket import BraketService
-from bloqade.builder.backend.bloqade import BloqadeService
+from bloqade.analog.builder.backend.quera import QuEraService
+from bloqade.analog.builder.backend.braket import BraketService
+from bloqade.analog.builder.backend.bloqade import BloqadeService
 
 
 class BackendRoute(QuEraService, BraketService, BloqadeService):

src/bloqade/builder/backend/__init__.py renamed to src/bloqade/analog/builder/backend/__init__.py

@@ -1,4 +1,4 @@
-from bloqade.builder.base import Builder
+from bloqade.analog.builder.base import Builder
 
 
 class BloqadeService(Builder):