Merge branch 'main' into fix/v2-NaT-fill_value

Author: moradology

.pre-commit-config.yaml

@@ -37,6 +37,8 @@ repos:
           - obstore>=0.5.1
           # Tests
           - pytest
+          - hypothesis
+          - s3fs
   - repo: https://github.com/scientific-python/cookie
     rev: 2025.01.22
     hooks:

changes/2862.bugfix.rst

@@ -0,0 +1 @@
+Fix a bug that prevented the number of initialized chunks being counted properly.

changes/2962.fix.rst

@@ -0,0 +1 @@
+Internally use `typesize` constructor parameter for :class:`numcodecs.blosc.Blosc` to improve compression ratios back to the v2-package levels.

changes/2978.bugfix.rst

@@ -0,0 +1 @@
+Fixed sharding with GPU buffers.

changes/3027.misc.rst

@@ -0,0 +1 @@
+Simplified scalar indexing of size-1 arrays.

changes/3045.bugfix.rst

@@ -0,0 +1 @@
+Fixed the typing of ``dimension_names`` arguments throughout so that it now accepts iterables that contain `None` alongside `str`.

docs/user-guide/arrays.rst

@@ -209,8 +209,8 @@ prints additional diagnostics, e.g.::
    Serializer         : BytesCodec(endian=<Endian.little: 'little'>)
    Compressors        : (BloscCodec(typesize=4, cname=<BloscCname.zstd: 'zstd'>, clevel=3, shuffle=<BloscShuffle.bitshuffle: 'bitshuffle'>, blocksize=0),)
    No. bytes          : 400000000 (381.5M)
-   No. bytes stored   : 9696520
-   Storage ratio      : 41.3
+   No. bytes stored   : 3558573
+   Storage ratio      : 112.4
    Chunks Initialized : 100
 
 .. note::

docs/user-guide/groups.rst

@@ -140,7 +140,7 @@ property. E.g.::
    No. bytes          : 8000000 (7.6M)
    No. bytes stored   : 1614
    Storage ratio      : 4956.6
-   Chunks Initialized : 0
+   Chunks Initialized : 10
    >>> baz.info
    Type               : Array
    Zarr format        : 3

pyproject.toml

@@ -83,7 +83,7 @@ test = [
 remote_tests = [
     'zarr[remote]',
     "botocore",
-    "s3fs",
+    "s3fs>=2023.10.0",
     "moto[s3,server]",
     "requests",
 ]
@@ -104,7 +104,7 @@ docs = [
     # Optional dependencies to run examples
     'numcodecs[msgpack]',
     'rich',
-    's3fs',
+    's3fs>=2023.10.0',
     'astroid<4'
 ]
 
@@ -348,27 +348,35 @@ python_version = "3.11"
 ignore_missing_imports = true
 namespace_packages = false
 
-
 strict = true
 warn_unreachable = true
-
 enable_error_code = ["ignore-without-code", "redundant-expr", "truthy-bool"]
 
+
 [[tool.mypy.overrides]]
 module = [
-    "zarr.v2.*",
+    "tests.package_with_entrypoint.*",
+    "zarr.testing.stateful",
+    "tests.test_codecs.test_transpose",
+    "tests.test_config",
+    "tests.test_store.test_zip",
+    "tests.test_store.test_local",
+    "tests.test_store.test_fsspec",
+    "tests.test_store.test_memory",
 ]
-ignore_errors = true
+strict = false
 
+# TODO: Move the next modules up to the strict = false section
+# and fix the errors
 [[tool.mypy.overrides]]
 module = [
-    "zarr.testing.stateful",  # lots of hypothesis decorator errors
-    "tests.package_with_entrypoint.*",
     "tests.test_codecs.test_codecs",
-    "tests.test_codecs.test_transpose",
     "tests.test_metadata.*",
-    "tests.test_store.*",
-    "tests.test_config",
+    "tests.test_store.test_core",
+    "tests.test_store.test_logging",
+    "tests.test_store.test_object",
+    "tests.test_store.test_stateful",
+    "tests.test_store.test_wrapper",
     "tests.test_group",
     "tests.test_indexing",
     "tests.test_properties",

src/zarr/api/asynchronous.py

@@ -16,6 +16,7 @@
     JSON,
     AccessModeLiteral,
     ChunkCoords,
+    DimensionNames,
     MemoryOrder,
     ZarrFormat,
     _default_zarr_format,
@@ -865,7 +866,7 @@ async def create(
         | None
     ) = None,
     codecs: Iterable[Codec | dict[str, JSON]] | None = None,
-    dimension_names: Iterable[str] | None = None,
+    dimension_names: DimensionNames = None,
     storage_options: dict[str, Any] | None = None,
     config: ArrayConfigLike | None = None,
     **kwargs: Any,