Adding and updating __init__ files for modules

examples/mpi/global_timings.py

@@ -3,7 +3,7 @@
 import numpy as np
 from mpi4py import MPI
 
-from ndsl.performance.timer import Timer
+from ndsl import Timer
 
 
 @contextlib.contextmanager

examples/mpi/zarr_monitor.py

@@ -5,11 +5,14 @@
 import zarr
 from mpi4py import MPI
 
-from ndsl.comm.partitioner import CubedSpherePartitioner, TilePartitioner
+from ndsl import (
+    CubedSpherePartitioner,
+    QuantityFactory,
+    SubtileGridSizer,
+    TilePartitioner,
+    ZarrMonitor,
+)
 from ndsl.constants import X_DIM, Y_DIM, Z_DIM
-from ndsl.initialization.allocator import QuantityFactory
-from ndsl.initialization.sizer import SubtileGridSizer
-from ndsl.monitor import ZarrMonitor
 
 
 OUTPUT_PATH = "output/zarr_monitor.zarr"

ndsl/__init__.py

@@ -1 +1,62 @@
+from .buffer import Buffer
+from .comm.boundary import Boundary, SimpleBoundary
+from .comm.communicator import CubedSphereCommunicator, TileCommunicator
+from .comm.local_comm import AsyncResult, ConcurrencyError, LocalComm
+from .comm.mpi import MPIComm
+from .comm.null_comm import NullAsyncResult, NullComm
+from .comm.partitioner import CubedSpherePartitioner, TilePartitioner
+from .constants import ConstantVersions
+from .dsl.caches.codepath import FV3CodePath
+from .dsl.dace.dace_config import DaceConfig, DaCeOrchestration, FrozenCompiledSDFG
+from .dsl.dace.orchestration import orchestrate, orchestrate_function
+from .dsl.dace.utils import (
+    ArrayReport,
+    DaCeProgress,
+    MaxBandwithBenchmarkProgram,
+    StorageReport,
+)
+from .dsl.dace.wrapped_halo_exchange import WrappedHaloUpdater
+from .dsl.stencil import (
+    CompareToNumpyStencil,
+    FrozenStencil,
+    GridIndexing,
+    StencilFactory,
+    TimingCollector,
+)
+from .dsl.stencil_config import CompilationConfig, RunMode, StencilConfig
+from .exceptions import OutOfBoundsError
+from .halo.data_transformer import (
+    HaloDataTransformer,
+    HaloDataTransformerCPU,
+    HaloDataTransformerGPU,
+    HaloExchangeSpec,
+)
+from .halo.updater import HaloUpdater, HaloUpdateRequest, VectorInterfaceHaloUpdater
+from .initialization.allocator import QuantityFactory, StorageNumpy
+from .initialization.sizer import GridSizer, SubtileGridSizer
 from .logging import ndsl_log
+from .monitor.netcdf_monitor import NetCDFMonitor
+from .monitor.protocol import Protocol
+from .monitor.zarr_monitor import ZarrMonitor
+from .namelist import Namelist
+from .optional_imports import RaiseWhenAccessed
+from .performance.collector import (
+    AbstractPerformanceCollector,
+    NullPerformanceCollector,
+    PerformanceCollector,
+)
+from .performance.config import PerformanceConfig
+from .performance.profiler import NullProfiler, Profiler
+from .performance.report import Experiment, Report, TimeReport
+from .performance.timer import NullTimer, Timer
+from .quantity import (
+    BoundaryArrayView,
+    BoundedArrayView,
+    Quantity,
+    QuantityHaloSpec,
+    QuantityMetadata,
+)
+from .testing.dummy_comm import DummyComm
+from .types import Allocator, AsyncRequest, NumpyModule
+from .units import UnitsError
+from .utils import MetaEnumStr

ndsl/checkpointer/__init__.py

@@ -1,6 +1,5 @@
-from .base import Checkpointer
 from .null import NullCheckpointer
-from .snapshots import SnapshotCheckpointer
+from .snapshots import SnapshotCheckpointer, _Snapshots
 from .thresholds import (
     InsufficientTrialsError,
     SavepointThresholds,

ndsl/checkpointer/null.py

@@ -1,4 +1,4 @@
-from .base import Checkpointer
+from ndsl.typing import Checkpointer
 
 
 class NullCheckpointer(Checkpointer):

ndsl/checkpointer/snapshots.py

@@ -4,8 +4,7 @@
 
 from ndsl.optional_imports import cupy as cp
 from ndsl.optional_imports import xarray as xr
-
-from .base import Checkpointer
+from ndsl.typing import Checkpointer
 
 
 def make_dims(savepoint_dim, label, data_list):

ndsl/checkpointer/thresholds.py

@@ -5,8 +5,8 @@
 
 import numpy as np
 
-from ..quantity import Quantity
-from .base import Checkpointer
+from ndsl.quantity import Quantity
+from ndsl.typing import Checkpointer
 
 
 try:

ndsl/checkpointer/validation.py

@@ -5,10 +5,14 @@
 
 import numpy as np
 
+from ndsl.checkpointer.thresholds import (
+    ArrayLike,
+    SavepointName,
+    SavepointThresholds,
+    cast_to_ndarray,
+)
 from ndsl.optional_imports import xarray as xr
-
-from .base import Checkpointer
-from .thresholds import ArrayLike, SavepointName, SavepointThresholds, cast_to_ndarray
+from ndsl.typing import Checkpointer
 
 
 def _clip_pace_array_to_target(

ndsl/comm/__init__.py

@@ -0,0 +1,9 @@
+from .caching_comm import (
+    CachingCommData,
+    CachingCommReader,
+    CachingCommWriter,
+    CachingRequestReader,
+    CachingRequestWriter,
+    NullRequest,
+)
+from .comm_abc import Comm, Request

ndsl/comm/boundary.py

@@ -1,8 +1,8 @@
 import dataclasses
 from typing import Tuple
 
-from ..quantity import Quantity, QuantityHaloSpec
-from ._boundary_utils import get_boundary_slice
+from ndsl.comm._boundary_utils import get_boundary_slice
+from ndsl.quantity import Quantity, QuantityHaloSpec
 
 
 @dataclasses.dataclass