cubed-dev · lgray · Jul 16, 2025 · Jul 16, 2025
diff --git a/.github/workflows/mypy.yml b/.github/workflows/mypy.yml
@@ -46,4 +46,4 @@ jobs:
 
       - name: Run mypy
         run: |
-          python -m mypy cubed
+          python -m mypy src/cubed
diff --git a/.gitignore b/.gitignore
@@ -34,6 +34,9 @@ MANIFEST
 *.manifest
 *.spec
 
+# scikit-build-core dynamic version
+src/cubed/version.py
+
 # Installer logs
 pip-log.txt
 pip-delete-this-directory.txt

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,10 +1,10 @@
 [build-system]
-requires = ["setuptools"]
-build-backend = "setuptools.build_meta"
+requires = ["hatchling>=1.17.1", "hatch-vcs"]
+build-backend = "hatchling.build"
 
 [project]
 name = "cubed"
-version = "0.21.0"
+dynamic = ["version"]
 authors = [
     {name = "Tom White", email = "[email protected]"},
 ]
@@ -94,11 +94,15 @@ homepage = "https://github.com/cubed-dev/cubed"
 documentation = "https://tomwhite.github.io/cubed"
 repository = "https://github.com/cubed-dev/cubed"
 
+[tool.hatch]
+version.source = "vcs"
+build.hooks.vcs.version-file = "src/cubed/version.py"
+
 [tool.pytest.ini_options]
 filterwarnings = [
     "error::UserWarning",
 ]
-testpaths = "cubed/tests"
+testpaths = "src/cubed/tests"
 
 [tool.ruff]
 target-version = "py311"
@@ -145,3 +149,54 @@ known-third-party = [
     "toolz",
     "zarr",
 ]
+
+[tool.mypy]
+exclude = "tests|doc"
+files = "src/"
+show_error_codes = true
+allow_redefinition = true
+
+[[tool.mypy.overrides]]
+# Ignore vendored libraries
+modules = "mypy-cubed.vendor.*"
+ignore_errors = true
+
+[[tool.mypy.overrides]]
+# Most of the numerical computing stack doesn't have type annotations yet.
+module = [
+    "mypy-apache_beam.*",
+    "mypy-aiostream.*",
+    "mypy-array_api_compat.*",
+    "mypy-cloudpickle.*",
+    "mypy-coiled.*",
+    "mypy-dask.*",
+    "mypy-donfig.*",
+    "mypy-distributed.*",
+    "mypy-fsspec.*",
+    "mypy-icechunk.*",
+    "mypy-lithops.*",
+    "mypy-IPython.*",
+    "mypy-memray.*",
+    "mypy-modal.*",
+    "mypy-matplotlib.*",
+    "mypy-ndindex.*",
+    "mypy-networkx.*",
+    "mypy-numcodecs.*",
+    "mypy-numpy.*",
+    "mypy-obstore.*",
+    "mypy-pandas.*",
+    "mypy-psutil.*",
+    "mypy-pytest.*",
+    "mypy-ray.*",
+    "mypy-rechunker.*",
+    "mypy-seaborn.*",
+    "mypy-six.*",
+    "mypy-tenacity.*",
+    "mypy-tensorstore.*",
+    "mypy-tlz.*",
+    "mypy-toolz.*",
+    "mypy-tqdm.*",
+    "mypy-zarr.*",
+    "mypy-pyspark.*",
+]
+ignore_missing_imports = true
diff --git a/setup.cfg b/setup.cfg
diff --git a/setup.py b/setup.py
diff --git a/cubed/__init__.py → src/cubed/__init__.py b/cubed/__init__.py → src/cubed/__init__.py
diff --git a/cubed/array/__init__.py → src/cubed/array/__init__.py b/cubed/array/__init__.py → src/cubed/array/__init__.py
diff --git a/cubed/array/nan_functions.py → src/cubed/array/nan_functions.py b/cubed/array/nan_functions.py → src/cubed/array/nan_functions.py
diff --git a/cubed/array/overlap.py → src/cubed/array/overlap.py b/cubed/array/overlap.py → src/cubed/array/overlap.py
diff --git a/cubed/array/pad.py → src/cubed/array/pad.py b/cubed/array/pad.py → src/cubed/array/pad.py
diff --git a/cubed/array_api/__init__.py → src/cubed/array_api/__init__.py b/cubed/array_api/__init__.py → src/cubed/array_api/__init__.py
diff --git a/cubed/array_api/array_object.py → src/cubed/array_api/array_object.py b/cubed/array_api/array_object.py → src/cubed/array_api/array_object.py
diff --git a/cubed/array_api/constants.py → src/cubed/array_api/constants.py b/cubed/array_api/constants.py → src/cubed/array_api/constants.py
diff --git a/cubed/array_api/creation_functions.py → src/cubed/array_api/creation_functions.py b/cubed/array_api/creation_functions.py → src/cubed/array_api/creation_functions.py
diff --git a/cubed/array_api/data_type_functions.py → src/cubed/array_api/data_type_functions.py b/cubed/array_api/data_type_functions.py → src/cubed/array_api/data_type_functions.py
diff --git a/cubed/array_api/dtypes.py → src/cubed/array_api/dtypes.py b/cubed/array_api/dtypes.py → src/cubed/array_api/dtypes.py
diff --git a/cubed/array_api/elementwise_functions.py → src/cubed/array_api/elementwise_functions.py b/cubed/array_api/elementwise_functions.py → src/cubed/array_api/elementwise_functions.py
diff --git a/cubed/array_api/indexing_functions.py → src/cubed/array_api/indexing_functions.py b/cubed/array_api/indexing_functions.py → src/cubed/array_api/indexing_functions.py
diff --git a/cubed/array_api/inspection.py → src/cubed/array_api/inspection.py b/cubed/array_api/inspection.py → src/cubed/array_api/inspection.py
diff --git a/cubed/array_api/linalg.py → src/cubed/array_api/linalg.py b/cubed/array_api/linalg.py → src/cubed/array_api/linalg.py
diff --git a/cubed/array_api/linear_algebra_functions.py → ...bed/array_api/linear_algebra_functions.py b/cubed/array_api/linear_algebra_functions.py → ...bed/array_api/linear_algebra_functions.py
diff --git a/cubed/array_api/manipulation_functions.py → ...cubed/array_api/manipulation_functions.py b/cubed/array_api/manipulation_functions.py → ...cubed/array_api/manipulation_functions.py
diff --git a/cubed/array_api/searching_functions.py → src/cubed/array_api/searching_functions.py b/cubed/array_api/searching_functions.py → src/cubed/array_api/searching_functions.py
diff --git a/cubed/array_api/statistical_functions.py → src/cubed/array_api/statistical_functions.py b/cubed/array_api/statistical_functions.py → src/cubed/array_api/statistical_functions.py
diff --git a/cubed/array_api/utility_functions.py → src/cubed/array_api/utility_functions.py b/cubed/array_api/utility_functions.py → src/cubed/array_api/utility_functions.py
diff --git a/cubed/backend_array_api.py → src/cubed/backend_array_api.py b/cubed/backend_array_api.py → src/cubed/backend_array_api.py
diff --git a/cubed/core/__init__.py → src/cubed/core/__init__.py b/cubed/core/__init__.py → src/cubed/core/__init__.py
diff --git a/cubed/core/array.py → src/cubed/core/array.py b/cubed/core/array.py → src/cubed/core/array.py
diff --git a/cubed/core/groupby.py → src/cubed/core/groupby.py b/cubed/core/groupby.py → src/cubed/core/groupby.py
diff --git a/cubed/core/gufunc.py → src/cubed/core/gufunc.py b/cubed/core/gufunc.py → src/cubed/core/gufunc.py
diff --git a/cubed/core/indexing.py → src/cubed/core/indexing.py b/cubed/core/indexing.py → src/cubed/core/indexing.py
diff --git a/cubed/core/ops.py → src/cubed/core/ops.py b/cubed/core/ops.py → src/cubed/core/ops.py
diff --git a/cubed/core/optimization.py → src/cubed/core/optimization.py b/cubed/core/optimization.py → src/cubed/core/optimization.py
diff --git a/cubed/core/plan.py → src/cubed/core/plan.py b/cubed/core/plan.py → src/cubed/core/plan.py
diff --git a/cubed/diagnostics/__init__.py → src/cubed/diagnostics/__init__.py b/cubed/diagnostics/__init__.py → src/cubed/diagnostics/__init__.py
diff --git a/cubed/diagnostics/history.py → src/cubed/diagnostics/history.py b/cubed/diagnostics/history.py → src/cubed/diagnostics/history.py
diff --git a/cubed/diagnostics/mem_usage.py → src/cubed/diagnostics/mem_usage.py b/cubed/diagnostics/mem_usage.py → src/cubed/diagnostics/mem_usage.py
diff --git a/cubed/diagnostics/mem_warn.py → src/cubed/diagnostics/mem_warn.py b/cubed/diagnostics/mem_warn.py → src/cubed/diagnostics/mem_warn.py
diff --git a/cubed/diagnostics/memray.py → src/cubed/diagnostics/memray.py b/cubed/diagnostics/memray.py → src/cubed/diagnostics/memray.py
diff --git a/cubed/diagnostics/rich.py → src/cubed/diagnostics/rich.py b/cubed/diagnostics/rich.py → src/cubed/diagnostics/rich.py
diff --git a/cubed/diagnostics/timeline.py → src/cubed/diagnostics/timeline.py b/cubed/diagnostics/timeline.py → src/cubed/diagnostics/timeline.py
diff --git a/cubed/diagnostics/tqdm.py → src/cubed/diagnostics/tqdm.py b/cubed/diagnostics/tqdm.py → src/cubed/diagnostics/tqdm.py
diff --git a/cubed/icechunk.py → src/cubed/icechunk.py b/cubed/icechunk.py → src/cubed/icechunk.py
diff --git a/cubed/primitive/__init__.py → src/cubed/primitive/__init__.py b/cubed/primitive/__init__.py → src/cubed/primitive/__init__.py
diff --git a/cubed/primitive/blockwise.py → src/cubed/primitive/blockwise.py b/cubed/primitive/blockwise.py → src/cubed/primitive/blockwise.py
diff --git a/cubed/primitive/memory.py → src/cubed/primitive/memory.py b/cubed/primitive/memory.py → src/cubed/primitive/memory.py
diff --git a/cubed/primitive/rechunk.py → src/cubed/primitive/rechunk.py b/cubed/primitive/rechunk.py → src/cubed/primitive/rechunk.py
diff --git a/cubed/primitive/types.py → src/cubed/primitive/types.py b/cubed/primitive/types.py → src/cubed/primitive/types.py
diff --git a/cubed/random.py → src/cubed/random.py b/cubed/random.py → src/cubed/random.py
diff --git a/cubed/runtime/__init__.py → src/cubed/runtime/__init__.py b/cubed/runtime/__init__.py → src/cubed/runtime/__init__.py
diff --git a/cubed/runtime/asyncio.py → src/cubed/runtime/asyncio.py b/cubed/runtime/asyncio.py → src/cubed/runtime/asyncio.py
diff --git a/cubed/runtime/backup.py → src/cubed/runtime/backup.py b/cubed/runtime/backup.py → src/cubed/runtime/backup.py
diff --git a/cubed/runtime/create.py → src/cubed/runtime/create.py b/cubed/runtime/create.py → src/cubed/runtime/create.py
diff --git a/cubed/runtime/executors/__init__.py → src/cubed/runtime/executors/__init__.py b/cubed/runtime/executors/__init__.py → src/cubed/runtime/executors/__init__.py
diff --git a/cubed/runtime/executors/beam.py → src/cubed/runtime/executors/beam.py b/cubed/runtime/executors/beam.py → src/cubed/runtime/executors/beam.py
diff --git a/cubed/runtime/executors/coiled.py → src/cubed/runtime/executors/coiled.py b/cubed/runtime/executors/coiled.py → src/cubed/runtime/executors/coiled.py
diff --git a/cubed/runtime/executors/dask.py → src/cubed/runtime/executors/dask.py b/cubed/runtime/executors/dask.py → src/cubed/runtime/executors/dask.py
diff --git a/cubed/runtime/executors/lithops.py → src/cubed/runtime/executors/lithops.py b/cubed/runtime/executors/lithops.py → src/cubed/runtime/executors/lithops.py
diff --git a/cubed/runtime/executors/local.py → src/cubed/runtime/executors/local.py b/cubed/runtime/executors/local.py → src/cubed/runtime/executors/local.py
diff --git a/cubed/runtime/executors/modal.py → src/cubed/runtime/executors/modal.py b/cubed/runtime/executors/modal.py → src/cubed/runtime/executors/modal.py
diff --git a/cubed/runtime/executors/ray.py → src/cubed/runtime/executors/ray.py b/cubed/runtime/executors/ray.py → src/cubed/runtime/executors/ray.py
diff --git a/cubed/runtime/executors/spark.py → src/cubed/runtime/executors/spark.py b/cubed/runtime/executors/spark.py → src/cubed/runtime/executors/spark.py
diff --git a/cubed/runtime/pipeline.py → src/cubed/runtime/pipeline.py b/cubed/runtime/pipeline.py → src/cubed/runtime/pipeline.py
diff --git a/cubed/runtime/types.py → src/cubed/runtime/types.py b/cubed/runtime/types.py → src/cubed/runtime/types.py
diff --git a/cubed/runtime/utils.py → src/cubed/runtime/utils.py b/cubed/runtime/utils.py → src/cubed/runtime/utils.py
diff --git a/cubed/spec.py → src/cubed/spec.py b/cubed/spec.py → src/cubed/spec.py
diff --git a/cubed/storage/__init__.py → src/cubed/storage/__init__.py b/cubed/storage/__init__.py → src/cubed/storage/__init__.py
diff --git a/cubed/storage/backend.py → src/cubed/storage/backend.py b/cubed/storage/backend.py → src/cubed/storage/backend.py
diff --git a/cubed/storage/backends/__init__.py → src/cubed/storage/backends/__init__.py b/cubed/storage/backends/__init__.py → src/cubed/storage/backends/__init__.py
diff --git a/cubed/storage/backends/tensorstore.py → src/cubed/storage/backends/tensorstore.py b/cubed/storage/backends/tensorstore.py → src/cubed/storage/backends/tensorstore.py
diff --git a/cubed/storage/backends/zarr_python.py → src/cubed/storage/backends/zarr_python.py b/cubed/storage/backends/zarr_python.py → src/cubed/storage/backends/zarr_python.py
diff --git a/cubed/storage/backends/zarr_python_v3.py → src/cubed/storage/backends/zarr_python_v3.py b/cubed/storage/backends/zarr_python_v3.py → src/cubed/storage/backends/zarr_python_v3.py
diff --git a/cubed/storage/types.py → src/cubed/storage/types.py b/cubed/storage/types.py → src/cubed/storage/types.py
diff --git a/cubed/storage/virtual.py → src/cubed/storage/virtual.py b/cubed/storage/virtual.py → src/cubed/storage/virtual.py
diff --git a/cubed/storage/zarr.py → src/cubed/storage/zarr.py b/cubed/storage/zarr.py → src/cubed/storage/zarr.py
diff --git a/cubed/tests/__init__.py → src/cubed/tests/__init__.py b/cubed/tests/__init__.py → src/cubed/tests/__init__.py
diff --git a/cubed/tests/array/test_nan_functions.py → src/cubed/tests/array/test_nan_functions.py b/cubed/tests/array/test_nan_functions.py → src/cubed/tests/array/test_nan_functions.py
diff --git a/cubed/tests/array/test_overlap.py → src/cubed/tests/array/test_overlap.py b/cubed/tests/array/test_overlap.py → src/cubed/tests/array/test_overlap.py
diff --git a/cubed/tests/array/test_pad.py → src/cubed/tests/array/test_pad.py b/cubed/tests/array/test_pad.py → src/cubed/tests/array/test_pad.py
diff --git a/cubed/tests/primitive/__init__.py → src/cubed/tests/primitive/__init__.py b/cubed/tests/primitive/__init__.py → src/cubed/tests/primitive/__init__.py
diff --git a/cubed/tests/primitive/test_blockwise.py → src/cubed/tests/primitive/test_blockwise.py b/cubed/tests/primitive/test_blockwise.py → src/cubed/tests/primitive/test_blockwise.py
diff --git a/.../tests/primitive/test_blockwise_fusion.py → .../tests/primitive/test_blockwise_fusion.py b/.../tests/primitive/test_blockwise_fusion.py → .../tests/primitive/test_blockwise_fusion.py
diff --git a/cubed/tests/primitive/test_memory.py → src/cubed/tests/primitive/test_memory.py b/cubed/tests/primitive/test_memory.py → src/cubed/tests/primitive/test_memory.py
diff --git a/cubed/tests/primitive/test_rechunk.py → src/cubed/tests/primitive/test_rechunk.py b/cubed/tests/primitive/test_rechunk.py → src/cubed/tests/primitive/test_rechunk.py
diff --git a/cubed/tests/runtime/__init__.py → src/cubed/tests/runtime/__init__.py b/cubed/tests/runtime/__init__.py → src/cubed/tests/runtime/__init__.py
diff --git a/cubed/tests/runtime/test_backup.py → src/cubed/tests/runtime/test_backup.py b/cubed/tests/runtime/test_backup.py → src/cubed/tests/runtime/test_backup.py
diff --git a/cubed/tests/runtime/test_dask.py → src/cubed/tests/runtime/test_dask.py b/cubed/tests/runtime/test_dask.py → src/cubed/tests/runtime/test_dask.py
diff --git a/cubed/tests/runtime/test_lithops.py → src/cubed/tests/runtime/test_lithops.py b/cubed/tests/runtime/test_lithops.py → src/cubed/tests/runtime/test_lithops.py
diff --git a/cubed/tests/runtime/test_local.py → src/cubed/tests/runtime/test_local.py b/cubed/tests/runtime/test_local.py → src/cubed/tests/runtime/test_local.py
diff --git a/cubed/tests/runtime/test_modal.py → src/cubed/tests/runtime/test_modal.py b/cubed/tests/runtime/test_modal.py → src/cubed/tests/runtime/test_modal.py
diff --git a/cubed/tests/runtime/utils.py → src/cubed/tests/runtime/utils.py b/cubed/tests/runtime/utils.py → src/cubed/tests/runtime/utils.py
diff --git a/cubed/tests/storage/__init__.py → src/cubed/tests/storage/__init__.py b/cubed/tests/storage/__init__.py → src/cubed/tests/storage/__init__.py
diff --git a/cubed/tests/storage/test_virtual.py → src/cubed/tests/storage/test_virtual.py b/cubed/tests/storage/test_virtual.py → src/cubed/tests/storage/test_virtual.py
diff --git a/cubed/tests/storage/test_zarr.py → src/cubed/tests/storage/test_zarr.py b/cubed/tests/storage/test_zarr.py → src/cubed/tests/storage/test_zarr.py
diff --git a/cubed/tests/test_array_api.py → src/cubed/tests/test_array_api.py b/cubed/tests/test_array_api.py → src/cubed/tests/test_array_api.py
diff --git a/cubed/tests/test_core.py → src/cubed/tests/test_core.py b/cubed/tests/test_core.py → src/cubed/tests/test_core.py
diff --git a/cubed/tests/test_executor_features.py → src/cubed/tests/test_executor_features.py b/cubed/tests/test_executor_features.py → src/cubed/tests/test_executor_features.py
diff --git a/cubed/tests/test_groupby.py → src/cubed/tests/test_groupby.py b/cubed/tests/test_groupby.py → src/cubed/tests/test_groupby.py
diff --git a/cubed/tests/test_gufunc.py → src/cubed/tests/test_gufunc.py b/cubed/tests/test_gufunc.py → src/cubed/tests/test_gufunc.py
diff --git a/cubed/tests/test_html.py → src/cubed/tests/test_html.py b/cubed/tests/test_html.py → src/cubed/tests/test_html.py
diff --git a/cubed/tests/test_icechunk.py → src/cubed/tests/test_icechunk.py b/cubed/tests/test_icechunk.py → src/cubed/tests/test_icechunk.py
diff --git a/cubed/tests/test_indexing.py → src/cubed/tests/test_indexing.py b/cubed/tests/test_indexing.py → src/cubed/tests/test_indexing.py
diff --git a/cubed/tests/test_inspection.py → src/cubed/tests/test_inspection.py b/cubed/tests/test_inspection.py → src/cubed/tests/test_inspection.py
diff --git a/cubed/tests/test_linalg.py → src/cubed/tests/test_linalg.py b/cubed/tests/test_linalg.py → src/cubed/tests/test_linalg.py
diff --git a/cubed/tests/test_mem_utilization.py → src/cubed/tests/test_mem_utilization.py b/cubed/tests/test_mem_utilization.py → src/cubed/tests/test_mem_utilization.py
diff --git a/cubed/tests/test_mem_utilization_rechunk.py → ...bed/tests/test_mem_utilization_rechunk.py b/cubed/tests/test_mem_utilization_rechunk.py → ...bed/tests/test_mem_utilization_rechunk.py
diff --git a/cubed/tests/test_ops.py → src/cubed/tests/test_ops.py b/cubed/tests/test_ops.py → src/cubed/tests/test_ops.py
diff --git a/cubed/tests/test_optimization.py → src/cubed/tests/test_optimization.py b/cubed/tests/test_optimization.py → src/cubed/tests/test_optimization.py
diff --git a/cubed/tests/test_random.py → src/cubed/tests/test_random.py b/cubed/tests/test_random.py → src/cubed/tests/test_random.py
diff --git a/cubed/tests/test_rechunk.py → src/cubed/tests/test_rechunk.py b/cubed/tests/test_rechunk.py → src/cubed/tests/test_rechunk.py
diff --git a/cubed/tests/test_types.py → src/cubed/tests/test_types.py b/cubed/tests/test_types.py → src/cubed/tests/test_types.py
diff --git a/cubed/tests/test_utils.py → src/cubed/tests/test_utils.py b/cubed/tests/test_utils.py → src/cubed/tests/test_utils.py
diff --git a/cubed/tests/utils.py → src/cubed/tests/utils.py b/cubed/tests/utils.py → src/cubed/tests/utils.py
diff --git a/cubed/types.py → src/cubed/types.py b/cubed/types.py → src/cubed/types.py
diff --git a/cubed/utils.py → src/cubed/utils.py b/cubed/utils.py → src/cubed/utils.py
diff --git a/cubed/vendor/__init__.py → src/cubed/vendor/__init__.py b/cubed/vendor/__init__.py → src/cubed/vendor/__init__.py
diff --git a/cubed/vendor/dask/LICENSE.txt → src/cubed/vendor/dask/LICENSE.txt b/cubed/vendor/dask/LICENSE.txt → src/cubed/vendor/dask/LICENSE.txt
diff --git a/cubed/vendor/dask/__init__.py → src/cubed/vendor/dask/__init__.py b/cubed/vendor/dask/__init__.py → src/cubed/vendor/dask/__init__.py
diff --git a/cubed/vendor/dask/array/__init__.py → src/cubed/vendor/dask/array/__init__.py b/cubed/vendor/dask/array/__init__.py → src/cubed/vendor/dask/array/__init__.py
diff --git a/cubed/vendor/dask/array/core.py → src/cubed/vendor/dask/array/core.py b/cubed/vendor/dask/array/core.py → src/cubed/vendor/dask/array/core.py
@@ -4,7 +4,7 @@
 from numbers import Number
 
 import numpy as np
-from tlz import first, frequencies
+from tlz import frequencies
 
 from ..utils import is_integer, parse_bytes
 
@@ -312,7 +312,7 @@ def auto_chunks(chunks, shape, limit, dtype, previous_chunks=None):
         return tuple(chunks)
 
     if limit is None:
-        limit = "128MiB" # config.get("array.chunk-size")
+        limit = "128MiB"  # config.get("array.chunk-size")
     if isinstance(limit, str):
         limit = parse_bytes(limit)
 

diff --git a/cubed/vendor/dask/array/core.pyi → src/cubed/vendor/dask/array/core.pyi b/cubed/vendor/dask/array/core.pyi → src/cubed/vendor/dask/array/core.pyi
@@ -3,18 +3,12 @@ from typing import List, Optional, Tuple
 from cubed.types import T_Chunks, T_DType, T_RectangularChunks, T_Shape
 
 def _check_regular_chunks(chunkset: T_RectangularChunks) -> bool: ...
-
-def broadcast_chunks(
-    *chunkss
-) -> Tuple[Tuple[int, ...]]: ...
-
+def broadcast_chunks(*chunkss) -> Tuple[Tuple[int, ...]]: ...
 def common_blockdim(blockdims: List[Tuple[int, ...]]) -> Tuple[int, ...]: ...
-
 def normalize_chunks(
     chunks: T_Chunks,
     shape: Optional[T_Shape] = None,
     limit: Optional[int] = None,
     dtype: Optional[T_DType] = None,
     previous_chunks: Optional[T_RectangularChunks] = None,
-) -> T_RectangularChunks:
-    ...
+) -> T_RectangularChunks: ...
diff --git a/cubed/vendor/dask/array/gufunc.py → src/cubed/vendor/dask/array/gufunc.py b/cubed/vendor/dask/array/gufunc.py → src/cubed/vendor/dask/array/gufunc.py
diff --git a/cubed/vendor/dask/array/overlap.py → src/cubed/vendor/dask/array/overlap.py b/cubed/vendor/dask/array/overlap.py → src/cubed/vendor/dask/array/overlap.py
diff --git a/cubed/vendor/dask/array/reshape.py → src/cubed/vendor/dask/array/reshape.py b/cubed/vendor/dask/array/reshape.py → src/cubed/vendor/dask/array/reshape.py
diff --git a/cubed/vendor/dask/array/svg.py → src/cubed/vendor/dask/array/svg.py b/cubed/vendor/dask/array/svg.py → src/cubed/vendor/dask/array/svg.py
diff --git a/cubed/vendor/dask/array/utils.py → src/cubed/vendor/dask/array/utils.py b/cubed/vendor/dask/array/utils.py → src/cubed/vendor/dask/array/utils.py
diff --git a/cubed/vendor/dask/blockwise.py → src/cubed/vendor/dask/blockwise.py b/cubed/vendor/dask/blockwise.py → src/cubed/vendor/dask/blockwise.py
@@ -358,11 +358,6 @@ def lol_product(head, values):
         return lol_product(head + (values[0],), values[1:])
 
 
-
-
-
-
-
 def broadcast_dimensions(argpairs, numblocks, sentinels=(1, (1,)), consolidate=None):
     """Find block dimensions from arguments
 

diff --git a/cubed/vendor/dask/core.py → src/cubed/vendor/dask/core.py b/cubed/vendor/dask/core.py → src/cubed/vendor/dask/core.py
diff --git a/cubed/vendor/dask/utils.py → src/cubed/vendor/dask/utils.py b/cubed/vendor/dask/utils.py → src/cubed/vendor/dask/utils.py
diff --git a/cubed/vendor/dask/widgets/__init__.py → src/cubed/vendor/dask/widgets/__init__.py b/cubed/vendor/dask/widgets/__init__.py → src/cubed/vendor/dask/widgets/__init__.py
diff --git a/...ndor/dask/widgets/templates/array.html.j2 → ...ndor/dask/widgets/templates/array.html.j2 b/...ndor/dask/widgets/templates/array.html.j2 → ...ndor/dask/widgets/templates/array.html.j2
diff --git a/cubed/vendor/dask/widgets/widgets.py → src/cubed/vendor/dask/widgets/widgets.py b/cubed/vendor/dask/widgets/widgets.py → src/cubed/vendor/dask/widgets/widgets.py
diff --git a/cubed/vendor/rechunker/LICENSE → src/cubed/vendor/rechunker/LICENSE b/cubed/vendor/rechunker/LICENSE → src/cubed/vendor/rechunker/LICENSE
diff --git a/cubed/vendor/rechunker/__init__.py → src/cubed/vendor/rechunker/__init__.py b/cubed/vendor/rechunker/__init__.py → src/cubed/vendor/rechunker/__init__.py
diff --git a/cubed/vendor/rechunker/algorithm.py → src/cubed/vendor/rechunker/algorithm.py b/cubed/vendor/rechunker/algorithm.py → src/cubed/vendor/rechunker/algorithm.py
@@ -1,4 +1,5 @@
 """Core rechunking algorithm stuff."""
+
 import logging
 import warnings
 from math import ceil, floor, prod
@@ -270,7 +271,6 @@ def multistage_rechunking_plan(
 
     # increase the number of stages until min_mem is exceeded
     for stage_count in range(1, MAX_STAGES):
-
         stage_chunks = calculate_stage_chunks(read_chunks, write_chunks, stage_count)
         pre_chunks = [read_chunks] + stage_chunks
         post_chunks = stage_chunks + [write_chunks]
@@ -297,6 +297,7 @@ def multistage_rechunking_plan(
                 f"Consider decreasing min_mem ({min_mem}) or increasing "
                 f"({max_mem}) to find a more efficient plan.",
                 category=ExcessiveIOWarning,
+                stacklevel=2,
             )
             assert prev_plan is not None
             return prev_plan

diff --git a/cubed/vendor/rechunker/compat.py → src/cubed/vendor/rechunker/compat.py b/cubed/vendor/rechunker/compat.py → src/cubed/vendor/rechunker/compat.py
diff --git a/cubed/vendor/rechunker/types.py → src/cubed/vendor/rechunker/types.py b/cubed/vendor/rechunker/types.py → src/cubed/vendor/rechunker/types.py
@@ -1,4 +1,5 @@
 """Types definitions used by executors."""
+
 from dataclasses import dataclass
 from typing import (
     Any,
@@ -61,12 +62,8 @@ class CopySpec(NamedTuple):
 
 
 Config = Any  # TODO: better typing for config
-SingleArgumentStageFunction = Callable[
-    [Any, NamedArg(type=Any, name="config")], None
-]  # noqa: F821
-NoArgumentStageFunction = Callable[
-    [NamedArg(type=Any, name="config")], None
-]  # noqa: F821
+SingleArgumentStageFunction = Callable[[Any, NamedArg(type=Any, name="config")], None]  # noqa: F821
+NoArgumentStageFunction = Callable[[NamedArg(type=Any, name="config")], None]  # noqa: F821
 StageFunction = Union[NoArgumentStageFunction, SingleArgumentStageFunction]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -358,11 +358,6 @@ def lol_product(head, values):
		return lol_product(head + (values[0],), values[1:])







		def broadcast_dimensions(argpairs, numblocks, sentinels=(1, (1,)), consolidate=None):
		"""Find block dimensions from arguments

Expand Down