ngio 0.4.0a3__py3-none-any.whl → 0.4.0a4__py3-none-any.whl

ngio/images/_label.py

@@ -3,6 +3,8 @@
 from collections.abc import Sequence
 from typing import Literal
 
+from zarr.types import DIMENSION_SEPARATOR
+
 from ngio.common import compute_masking_roi
 from ngio.images._abstract_image import AbstractImage
 from ngio.images._create import create_empty_label_container
@@ -94,7 +96,7 @@ class Label(AbstractImage[LabelMetaHandler]):
     ) -> None:
         """Consolidate the label on disk."""
         self._consolidate(
-            order=0,
+            order="nearest",
             mode=mode,
         )
 
@@ -167,6 +169,8 @@ class LabelsContainer:
         axes_names: Sequence[str] | None = None,
         chunks: Sequence[int] | None = None,
         dtype: str = "uint32",
+        dimension_separator: DIMENSION_SEPARATOR | None = None,
+        compressor=None,
         overwrite: bool = False,
     ) -> "Label":
         """Create an empty OME-Zarr label from a reference image.
@@ -184,6 +188,10 @@ class LabelsContainer:
                 For labels, the channel axis is not allowed.
             chunks (Sequence[int] | None): The chunk shape of the new image.
             dtype (str): The data type of the new label.
+            dimension_separator (DIMENSION_SEPARATOR | None): The separator to use for
+                dimensions. If None it will use the same as the reference image.
+            compressor: The compressor to use. If None it will use
+                the same as the reference image.
             overwrite (bool): Whether to overwrite an existing image.
 
         Returns:
@@ -226,6 +234,8 @@ def derive_label(
     pixel_size: PixelSize | None = None,
     axes_names: Sequence[str] | None = None,
     chunks: Sequence[int] | None = None,
+    dimension_separator: DIMENSION_SEPARATOR | None = None,
+    compressor=None,
     dtype: str = "uint32",
     overwrite: bool = False,
 ) -> None:
@@ -242,6 +252,10 @@ def derive_label(
             For labels, the channel axis is not allowed.
         chunks (Sequence[int] | None): The chunk shape of the new image.
         dtype (str): The data type of the new label.
+        dimension_separator (DIMENSION_SEPARATOR | None): The separator to use for
+            dimensions. If None it will use the same as the reference image.
+        compressor: The compressor to use. If None it will use
+            the same as the reference image.
         overwrite (bool): Whether to overwrite an existing image.
 
     Returns:
@@ -257,8 +271,8 @@ def derive_label(
         pixel_size = ref_image.pixel_size
 
     if axes_names is None:
-        axes_names = ref_meta.axes_mapper.axes_names
-        c_axis = ref_meta.axes_mapper.get_index("c")
+        axes_names = ref_meta.axes_handler.axes_names
+        c_axis = ref_meta.axes_handler.get_index("c")
     else:
         if "c" in axes_names:
             raise NgioValidationError(
@@ -291,6 +305,11 @@ def derive_label(
         axes_names = list(axes_names)
         axes_names = axes_names[:c_axis] + axes_names[c_axis + 1 :]
 
+    if dimension_separator is None:
+        dimension_separator = ref_image.zarr_array._dimension_separator  # type: ignore
+    if compressor is None:
+        compressor = ref_image.zarr_array.compressor  # type: ignore
+
     _ = create_empty_label_container(
         store=store,
         shape=shape,
@@ -305,6 +324,8 @@ def derive_label(
         axes_names=axes_names,
         chunks=chunks,
         dtype=dtype,
+        dimension_separator=dimension_separator,  # type: ignore
+        compressor=compressor,  # type: ignore
         overwrite=overwrite,
         version=ref_meta.version,
         name=name,

ngio/images/_masked_image.py

@@ -6,14 +6,6 @@ from typing import Literal
 import dask.array as da
 import numpy as np
 
-from ngio.common import (
-    ArrayLike,
-    TransformProtocol,
-    build_roi_masked_dask_getter,
-    build_roi_masked_dask_setter,
-    build_roi_masked_numpy_getter,
-    build_roi_masked_numpy_setter,
-)
 from ngio.images._image import (
     ChannelSlicingInputType,
     Image,
@@ -21,6 +13,13 @@ from ngio.images._image import (
     add_channel_selection_to_slicing_dict,
 )
 from ngio.images._label import Label
+from ngio.io_pipes import (
+    DaskGetterMasked,
+    DaskSetterMasked,
+    NumpyGetterMasked,
+    NumpySetterMasked,
+    TransformProtocol,
+)
 from ngio.ome_zarr_meta import ImageMetaHandler, LabelMetaHandler
 from ngio.tables import MaskingRoiTable
 from ngio.utils import (
@@ -70,7 +69,7 @@ class MaskedImage(Image):
         axes_order: Sequence[str] | None = None,
         transforms: Sequence[TransformProtocol] | None = None,
         **slicing_kwargs: slice | int | Sequence[int],
-    ) -> ArrayLike:
+    ) -> np.ndarray:
         """Return the array for a given ROI."""
         roi = self._masking_roi_table.get_label(label)
         roi = roi.zoom(zoom_factor)
@@ -111,7 +110,7 @@ class MaskedImage(Image):
         transforms: Sequence[TransformProtocol] | None = None,
         mode: Literal["numpy", "dask"] = "numpy",
         **slicing_kwargs: slice | int | Sequence[int],
-    ) -> ArrayLike:
+    ) -> np.ndarray | da.Array:
         """Return the array for a given ROI."""
         roi = self._masking_roi_table.get_label(label)
         roi = roi.zoom(zoom_factor)
@@ -127,7 +126,7 @@ class MaskedImage(Image):
     def set_roi(  # type: ignore (this ignore the method override issue)
         self,
         label: int,
-        patch: ArrayLike,
+        patch: np.ndarray | da.Array,
         zoom_factor: float = 1.0,
         channel_selection: ChannelSlicingInputType | None = None,
         axes_order: Sequence[str] | None = None,
@@ -153,7 +152,7 @@ class MaskedImage(Image):
         zoom_factor: float = 1.0,
         axes_order: Sequence[str] | None = None,
         transforms: Sequence[TransformProtocol] | None = None,
-        allow_scaling: bool = True,
+        allow_rescaling: bool = True,
         **slicing_kwargs: SlicingInputType,
     ) -> np.ndarray:
         """Return the masked array for a given label as a NumPy array."""
@@ -163,18 +162,16 @@ class MaskedImage(Image):
 
         roi = self._masking_roi_table.get_label(label)
         roi = roi.zoom(zoom_factor)
-        masked_getter = build_roi_masked_numpy_getter(
+        masked_getter = NumpyGetterMasked(
             roi=roi,
             zarr_array=self.zarr_array,
             label_zarr_array=self._label.zarr_array,
             dimensions=self.dimensions,
-            pixel_size=self.pixel_size,
             label_dimensions=self._label.dimensions,
-            label_pixel_size=self._label.pixel_size,
             axes_order=axes_order,
             transforms=transforms,
             slicing_dict=slicing_kwargs,
-            allow_scaling=allow_scaling,
+            allow_rescaling=allow_rescaling,
         )
         return masked_getter()
 
@@ -185,7 +182,7 @@ class MaskedImage(Image):
         zoom_factor: float = 1.0,
         axes_order: Sequence[str] | None = None,
         transforms: Sequence[TransformProtocol] | None = None,
-        allow_scaling: bool = True,
+        allow_rescaling: bool = True,
         **slicing_kwargs: SlicingInputType,
     ) -> da.Array:
         """Return the masked array for a given label as a Dask array."""
@@ -195,18 +192,16 @@ class MaskedImage(Image):
 
         roi = self._masking_roi_table.get_label(label)
         roi = roi.zoom(zoom_factor)
-        masked_getter = build_roi_masked_dask_getter(
+        masked_getter = DaskGetterMasked(
             roi=roi,
             zarr_array=self.zarr_array,
             label_zarr_array=self._label.zarr_array,
             dimensions=self.dimensions,
-            pixel_size=self.pixel_size,
             label_dimensions=self._label.dimensions,
-            label_pixel_size=self._label.pixel_size,
             axes_order=axes_order,
             transforms=transforms,
             slicing_dict=slicing_kwargs,
-            allow_scaling=allow_scaling,
+            allow_rescaling=allow_rescaling,
         )
         return masked_getter()
 
@@ -218,42 +213,38 @@ class MaskedImage(Image):
         axes_order: Sequence[str] | None = None,
         transforms: Sequence[TransformProtocol] | None = None,
         mode: Literal["numpy", "dask"] = "numpy",
-        allow_scaling: bool = True,
+        allow_rescaling: bool = True,
         **slicing_kwargs: SlicingInputType,
-    ) -> ArrayLike:
+    ) -> np.ndarray | da.Array:
         """Return the masked array for a given label."""
-        if mode == "numpy":
-            return self.get_roi_masked_as_numpy(
-                label=label,
-                channel_selection=channel_selection,
-                zoom_factor=zoom_factor,
-                axes_order=axes_order,
-                transforms=transforms,
-                allow_scaling=allow_scaling,
-                **slicing_kwargs,
-            )
-        elif mode == "dask":
-            return self.get_roi_masked_as_dask(
-                label=label,
-                channel_selection=channel_selection,
-                zoom_factor=zoom_factor,
-                axes_order=axes_order,
-                transforms=transforms,
-                allow_scaling=allow_scaling,
-                **slicing_kwargs,
-            )
-        else:
-            raise ValueError(f"Unknown mode: {mode}")
+        slicing_kwargs = add_channel_selection_to_slicing_dict(
+            image=self, channel_selection=channel_selection, slicing_dict=slicing_kwargs
+        )
+
+        roi = self._masking_roi_table.get_label(label)
+        roi = roi.zoom(zoom_factor)
+        masked_getter = NumpyGetterMasked(
+            roi=roi,
+            zarr_array=self.zarr_array,
+            label_zarr_array=self._label.zarr_array,
+            dimensions=self.dimensions,
+            label_dimensions=self._label.dimensions,
+            axes_order=axes_order,
+            transforms=transforms,
+            slicing_dict=slicing_kwargs,
+            allow_rescaling=allow_rescaling,
+        )
+        return masked_getter()
 
     def set_roi_masked(
         self,
         label: int,
-        patch: ArrayLike,
+        patch: np.ndarray | da.Array,
         channel_selection: ChannelSlicingInputType | None = None,
         axes_order: Sequence[str] | None = None,
         zoom_factor: float = 1.0,
         transforms: Sequence[TransformProtocol] | None = None,
-        allow_scaling: bool = True,
+        allow_rescaling: bool = True,
         **slicing_kwargs: SlicingInputType,
     ) -> None:
         """Set the masked array for a given label."""
@@ -264,33 +255,29 @@ class MaskedImage(Image):
         roi = self._masking_roi_table.get_label(label)
         roi = roi.zoom(zoom_factor)
         if isinstance(patch, da.Array):
-            path_setter = build_roi_masked_dask_setter(
+            path_setter = DaskSetterMasked(
                 roi=roi,
                 zarr_array=self.zarr_array,
                 label_zarr_array=self._label.zarr_array,
                 dimensions=self.dimensions,
-                pixel_size=self.pixel_size,
                 label_dimensions=self._label.dimensions,
-                label_pixel_size=self._label.pixel_size,
                 axes_order=axes_order,
                 transforms=transforms,
                 slicing_dict=slicing_kwargs,
-                allow_scaling=allow_scaling,
+                allow_rescaling=allow_rescaling,
             )
             path_setter(patch)
         elif isinstance(patch, np.ndarray):
-            path_setter = build_roi_masked_numpy_setter(
+            path_setter = NumpySetterMasked(
                 roi=roi,
                 zarr_array=self.zarr_array,
                 label_zarr_array=self._label.zarr_array,
                 dimensions=self.dimensions,
-                pixel_size=self.pixel_size,
                 label_dimensions=self._label.dimensions,
-                label_pixel_size=self._label.pixel_size,
                 axes_order=axes_order,
                 transforms=transforms,
                 slicing_dict=slicing_kwargs,
-                allow_scaling=allow_scaling,
+                allow_rescaling=allow_rescaling,
             )
             path_setter(patch)
         else:
@@ -378,7 +365,7 @@ class MaskedLabel(Label):
         mode: Literal["numpy", "dask"] = "numpy",
         transforms: Sequence[TransformProtocol] | None = None,
         **slicing_kwargs: slice | int | Sequence[int],
-    ) -> ArrayLike:
+    ) -> np.ndarray | da.Array:
         """Return the array for a given ROI."""
         roi = self._masking_roi_table.get_label(label)
         roi = roi.zoom(zoom_factor)
@@ -393,7 +380,7 @@ class MaskedLabel(Label):
     def set_roi(
         self,
         label: int,
-        patch: ArrayLike,
+        patch: np.ndarray | da.Array,
         zoom_factor: float = 1.0,
         axes_order: Sequence[str] | None = None,
         transforms: Sequence[TransformProtocol] | None = None,
@@ -416,24 +403,22 @@ class MaskedLabel(Label):
         zoom_factor: float = 1.0,
         axes_order: Sequence[str] | None = None,
         transforms: Sequence[TransformProtocol] | None = None,
-        allow_scaling: bool = True,
+        allow_rescaling: bool = True,
         **slicing_kwargs: SlicingInputType,
     ) -> np.ndarray:
         """Return the masked array for a given label as a NumPy array."""
         roi = self._masking_roi_table.get_label(label)
         roi = roi.zoom(zoom_factor)
-        masked_getter = build_roi_masked_numpy_getter(
+        masked_getter = NumpyGetterMasked(
             roi=roi,
             zarr_array=self.zarr_array,
             label_zarr_array=self._label.zarr_array,
             dimensions=self.dimensions,
-            pixel_size=self.pixel_size,
             label_dimensions=self._label.dimensions,
-            label_pixel_size=self._label.pixel_size,
             axes_order=axes_order,
             transforms=transforms,
             slicing_dict=slicing_kwargs,
-            allow_scaling=allow_scaling,
+            allow_rescaling=allow_rescaling,
         )
         return masked_getter()
 
@@ -443,24 +428,22 @@ class MaskedLabel(Label):
         zoom_factor: float = 1.0,
         axes_order: Sequence[str] | None = None,
         transforms: Sequence[TransformProtocol] | None = None,
-        allow_scaling: bool = True,
+        allow_rescaling: bool = True,
         **slicing_kwargs: SlicingInputType,
     ) -> da.Array:
         """Return the masked array for a given label as a Dask array."""
         roi = self._masking_roi_table.get_label(label)
         roi = roi.zoom(zoom_factor)
-        masked_getter = build_roi_masked_dask_getter(
+        masked_getter = DaskGetterMasked(
             roi=roi,
             zarr_array=self.zarr_array,
             label_zarr_array=self._label.zarr_array,
             dimensions=self.dimensions,
-            pixel_size=self.pixel_size,
             label_dimensions=self._label.dimensions,
-            label_pixel_size=self._label.pixel_size,
             axes_order=axes_order,
             transforms=transforms,
             slicing_dict=slicing_kwargs,
-            allow_scaling=allow_scaling,
+            allow_rescaling=allow_rescaling,
         )
         return masked_getter()
 
@@ -471,9 +454,9 @@ class MaskedLabel(Label):
         axes_order: Sequence[str] | None = None,
         mode: Literal["numpy", "dask"] = "numpy",
         transforms: Sequence[TransformProtocol] | None = None,
-        allow_scaling: bool = True,
+        allow_rescaling: bool = True,
         **slicing_kwargs: SlicingInputType,
-    ) -> ArrayLike:
+    ) -> np.ndarray | da.Array:
         """Return the masked array for a given label."""
         if mode == "numpy":
             return self.get_roi_masked_as_numpy(
@@ -481,7 +464,7 @@ class MaskedLabel(Label):
                 zoom_factor=zoom_factor,
                 axes_order=axes_order,
                 transforms=transforms,
-                allow_scaling=allow_scaling,
+                allow_rescaling=allow_rescaling,
                 **slicing_kwargs,
             )
 
@@ -491,7 +474,7 @@ class MaskedLabel(Label):
                 zoom_factor=zoom_factor,
                 axes_order=axes_order,
                 transforms=transforms,
-                allow_scaling=allow_scaling,
+                allow_rescaling=allow_rescaling,
                 **slicing_kwargs,
             )
         else:
@@ -500,44 +483,40 @@ class MaskedLabel(Label):
     def set_roi_masked(
         self,
         label: int,
-        patch: ArrayLike,
+        patch: np.ndarray | da.Array,
         axes_order: Sequence[str] | None = None,
         zoom_factor: float = 1.0,
         transforms: Sequence[TransformProtocol] | None = None,
-        allow_scaling: bool = True,
+        allow_rescaling: bool = True,
         **slicing_kwargs: SlicingInputType,
     ) -> None:
         """Set the masked array for a given label."""
         roi = self._masking_roi_table.get_label(label)
         roi = roi.zoom(zoom_factor)
         if isinstance(patch, da.Array):
-            path_setter = build_roi_masked_dask_setter(
+            path_setter = DaskSetterMasked(
                 roi=roi,
                 zarr_array=self.zarr_array,
                 label_zarr_array=self._label.zarr_array,
                 dimensions=self.dimensions,
-                pixel_size=self.pixel_size,
                 label_dimensions=self._label.dimensions,
-                label_pixel_size=self._label.pixel_size,
                 axes_order=axes_order,
                 transforms=transforms,
                 slicing_dict=slicing_kwargs,
-                allow_scaling=allow_scaling,
+                allow_rescaling=allow_rescaling,
             )
             path_setter(patch)
         elif isinstance(patch, np.ndarray):
-            path_setter = build_roi_masked_numpy_setter(
+            path_setter = NumpySetterMasked(
                 roi=roi,
                 zarr_array=self.zarr_array,
                 label_zarr_array=self._label.zarr_array,
                 dimensions=self.dimensions,
-                pixel_size=self.pixel_size,
                 label_dimensions=self._label.dimensions,
-                label_pixel_size=self._label.pixel_size,
                 axes_order=axes_order,
                 transforms=transforms,
                 slicing_dict=slicing_kwargs,
-                allow_scaling=allow_scaling,
+                allow_rescaling=allow_rescaling,
             )
             path_setter(patch)
         else:

ngio/images/_ome_zarr_container.py

@@ -4,6 +4,7 @@ import warnings
 from collections.abc import Sequence
 
 import numpy as np
+from zarr.types import DIMENSION_SEPARATOR
 
 from ngio.images._create import create_empty_image_container
 from ngio.images._image import Image, ImagesContainer
@@ -408,6 +409,8 @@ class OmeZarrContainer:
         name: str | None = None,
         chunks: Sequence[int] | None = None,
         dtype: str | None = None,
+        dimension_separator: DIMENSION_SEPARATOR | None = None,
+        compressor=None,
         copy_labels: bool = False,
         copy_tables: bool = False,
         overwrite: bool = False,
@@ -425,6 +428,11 @@ class OmeZarrContainer:
             chunks (Sequence[int] | None): The chunk shape of the new image.
             dtype (str | None): The data type of the new image.
             name (str | None): The name of the new image.
+            dimension_separator (DIMENSION_SEPARATOR | None): The dimension
+                separator to use. If None, the dimension separator of the
+                reference image will be used.
+            compressor: The compressor to use. If None, the compressor of the
+                reference image will be used.
             copy_labels (bool): Whether to copy the labels from the reference image.
             copy_tables (bool): Whether to copy the tables from the reference image.
             overwrite (bool): Whether to overwrite an existing image.
@@ -443,6 +451,8 @@ class OmeZarrContainer:
             name=name,
             chunks=chunks,
             dtype=dtype,
+            dimension_separator=dimension_separator,
+            compressor=compressor,
             overwrite=overwrite,
         )
 
@@ -683,6 +693,8 @@ class OmeZarrContainer:
         axes_names: Sequence[str] | None = None,
         chunks: Sequence[int] | None = None,
         dtype: str = "uint32",
+        dimension_separator: DIMENSION_SEPARATOR | None = None,
+        compressor=None,
         overwrite: bool = False,
     ) -> "Label":
         """Create an empty OME-Zarr label from a reference image.
@@ -699,6 +711,11 @@ class OmeZarrContainer:
                 For labels, the channel axis is not allowed.
             chunks (Sequence[int] | None): The chunk shape of the new image.
             dtype (str): The data type of the new label.
+            dimension_separator (DIMENSION_SEPARATOR | None): The dimension
+                separator to use. If None, the dimension separator of the
+                reference image will be used.
+            compressor: The compressor to use. If None, the compressor of the
+                reference image will be used.
             overwrite (bool): Whether to overwrite an existing image.
 
         Returns:
@@ -715,6 +732,8 @@ class OmeZarrContainer:
             axes_names=axes_names,
             chunks=chunks,
             dtype=dtype,
+            dimension_separator=dimension_separator,
+            compressor=compressor,
             overwrite=overwrite,
         )
 
@@ -819,6 +838,8 @@ def create_empty_ome_zarr(
     name: str | None = None,
     chunks: Sequence[int] | None = None,
     dtype: str = "uint16",
+    dimension_separator: DIMENSION_SEPARATOR = "/",
+    compressor="default",
     channel_labels: list[str] | None = None,
     channel_wavelengths: list[str] | None = None,
     channel_colors: Sequence[str] | None = None,
@@ -851,6 +872,9 @@ def create_empty_ome_zarr(
         chunks (Sequence[int] | None, optional): The chunk shape. If None the shape
             is used. Defaults to None.
         dtype (str, optional): The data type of the image. Defaults to "uint16".
+        dimension_separator (DIMENSION_SEPARATOR): The dimension
+            separator to use. Defaults to "/".
+        compressor: The compressor to use. Defaults to "default".
         channel_labels (list[str] | None, optional): The labels of the channels.
             Defaults to None.
         channel_wavelengths (list[str] | None, optional): The wavelengths of the
@@ -879,6 +903,8 @@ def create_empty_ome_zarr(
         name=name,
         chunks=chunks,
         dtype=dtype,
+        dimension_separator=dimension_separator,
+        compressor=compressor,
         overwrite=overwrite,
         version=version,
     )
@@ -913,6 +939,8 @@ def create_ome_zarr_from_array(
     channel_active: Sequence[bool] | None = None,
     name: str | None = None,
     chunks: Sequence[int] | None = None,
+    dimension_separator: DIMENSION_SEPARATOR = "/",
+    compressor: str = "default",
     overwrite: bool = False,
     version: NgffVersions = DefaultNgffVersion,
 ) -> OmeZarrContainer:
@@ -950,6 +978,9 @@ def create_ome_zarr_from_array(
             Defaults to None.
         channel_active (Sequence[bool] | None, optional): Whether the channels are
             active. Defaults to None.
+        dimension_separator (DIMENSION_SEPARATOR): The separator to use for
+            dimensions. Defaults to "/".
+        compressor: The compressor to use. Defaults to "default".
         overwrite (bool, optional): Whether to overwrite an existing image.
             Defaults to True.
         version (str, optional): The version of the OME-Zarr specification.
@@ -971,6 +1002,8 @@ def create_ome_zarr_from_array(
         chunks=chunks,
         dtype=str(array.dtype),
         overwrite=overwrite,
+        dimension_separator=dimension_separator,
+        compressor=compressor,
         version=version,
     )
 

ngio/io_pipes/__init__.py

@@ -0,0 +1,49 @@
+"""I/O pipes for reading and writing data from zarr to numpy and dask arrays."""
+
+from ngio.io_pipes._io_pipes import (
+    DaskGetter,
+    DaskSetter,
+    DataGetter,
+    DataSetter,
+    NumpyGetter,
+    NumpySetter,
+)
+from ngio.io_pipes._io_pipes_masked import (
+    DaskGetterMasked,
+    DaskSetterMasked,
+    NumpyGetterMasked,
+    NumpySetterMasked,
+)
+from ngio.io_pipes._io_pipes_roi import (
+    DaskRoiGetter,
+    DaskRoiSetter,
+    NumpyRoiGetter,
+    NumpyRoiSetter,
+)
+from ngio.io_pipes._io_pipes_utils import SlicingInputType
+from ngio.io_pipes._match_shape import dask_match_shape, numpy_match_shape
+from ngio.io_pipes._ops_slices import SlicingOps, SlicingType
+from ngio.io_pipes._ops_transforms import TransformProtocol
+
+__all__ = [
+    "DaskGetter",
+    "DaskGetterMasked",
+    "DaskRoiGetter",
+    "DaskRoiSetter",
+    "DaskSetter",
+    "DaskSetterMasked",
+    "DataGetter",
+    "DataSetter",
+    "NumpyGetter",
+    "NumpyGetterMasked",
+    "NumpyRoiGetter",
+    "NumpyRoiSetter",
+    "NumpySetter",
+    "NumpySetterMasked",
+    "SlicingInputType",
+    "SlicingOps",
+    "SlicingType",
+    "TransformProtocol",
+    "dask_match_shape",
+    "numpy_match_shape",
+]