Clean up zarr.codecs namespace

Author: dstansby

src/zarr/buffer.py

@@ -19,7 +19,7 @@
 if TYPE_CHECKING:
     from typing_extensions import Self
 
-    from zarr.codecs.bytes import Endian
+    from zarr.codecs._bytes import Endian
     from zarr.common import BytesLike
 
 
@@ -409,7 +409,7 @@ def shape(self) -> tuple[int, ...]:
 
     @property
     def byteorder(self) -> Endian:
-        from zarr.codecs.bytes import Endian
+        from zarr.codecs._bytes import Endian
 
         if self.dtype.byteorder == "<":
             return Endian.little

src/zarr/codecs/__init__.py

@@ -1,10 +1,10 @@
 from __future__ import annotations
 
-from zarr.codecs.blosc import BloscCname, BloscCodec, BloscShuffle  # noqa: F401
-from zarr.codecs.bytes import BytesCodec, Endian  # noqa: F401
-from zarr.codecs.crc32c_ import Crc32cCodec  # noqa: F401
-from zarr.codecs.gzip import GzipCodec  # noqa: F401
-from zarr.codecs.pipeline import BatchedCodecPipeline  # noqa: F401
-from zarr.codecs.sharding import ShardingCodec, ShardingCodecIndexLocation  # noqa: F401
-from zarr.codecs.transpose import TransposeCodec  # noqa: F401
-from zarr.codecs.zstd import ZstdCodec  # noqa: F401
+from zarr.codecs._blosc import BloscCname, BloscCodec, BloscShuffle  # noqa: F401
+from zarr.codecs._bytes import BytesCodec, Endian  # noqa: F401
+from zarr.codecs._crc32c_ import Crc32cCodec  # noqa: F401
+from zarr.codecs._gzip import GzipCodec  # noqa: F401
+from zarr.codecs._pipeline import BatchedCodecPipeline  # noqa: F401
+from zarr.codecs._sharding import ShardingCodec, ShardingCodecIndexLocation  # noqa: F401
+from zarr.codecs._transpose import TransposeCodec  # noqa: F401
+from zarr.codecs._zstd import ZstdCodec  # noqa: F401

src/zarr/codecs/blosc.py renamed to src/zarr/codecs/_blosc.py

@@ -94,7 +94,7 @@ def evolve_from_array_spec(self, array_spec: ArraySpec) -> Self:
     def codecs_from_list(
         codecs: list[Codec],
     ) -> tuple[tuple[ArrayArrayCodec, ...], ArrayBytesCodec, tuple[BytesBytesCodec, ...]]:
-        from zarr.codecs.sharding import ShardingCodec
+        from zarr.codecs._sharding import ShardingCodec
 
         if not any(isinstance(codec, ArrayBytesCodec) for codec in codecs):
             raise ValueError("Exactly one array-to-bytes codec is required.")

src/zarr/codecs/bytes.py renamed to src/zarr/codecs/_bytes.py

@@ -21,9 +21,9 @@
 )
 from zarr.buffer import Buffer, NDBuffer
 from zarr.chunk_grids import RegularChunkGrid
-from zarr.codecs.bytes import BytesCodec
-from zarr.codecs.crc32c_ import Crc32cCodec
-from zarr.codecs.pipeline import BatchedCodecPipeline
+from zarr.codecs._bytes import BytesCodec
+from zarr.codecs._crc32c_ import Crc32cCodec
+from zarr.codecs._pipeline import BatchedCodecPipeline
 from zarr.codecs.registry import register_codec
 from zarr.common import (
     ArraySpec,