From 43c6de2722ed5b93142c57db8016a94f035b9e5d Mon Sep 17 00:00:00 2001
From: Michal Januszewski <mjanusz@google.com>
Date: Thu, 5 Sep 2024 09:18:45 -0700
Subject: [PATCH] to_slice -> to_slice4d
PiperOrigin-RevId: 671397639
---
connectomics/volume/mask.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/connectomics/volume/mask.py b/connectomics/volume/mask.py
index b7ce35b..9804309 100644
--- a/connectomics/volume/mask.py
+++ b/connectomics/volume/mask.py
@@ -122,7 +122,7 @@ class MaskConfigs(dataclasses_json.DataClassJsonMixin):
# TODO(timblakely): Return a Subvolume.
def build_mask(
masks: Sequence[MaskConfig] | MaskConfigs,
- box: bounding_box.BoundingBoxBase,
+ box: bounding_box.BoundingBox,
decorated_volume_loader: Callable[[metadata.DecoratedVolume], np.ndarray],
mask_volume_map: dict[str, Any] | None = None,
image: np.ndarray | None = None,
@@ -187,7 +187,7 @@ def build_mask(
decorated_volume_loader(config.volume.mask)
)
volume = mask_volume_map[volume_key]
- mask = volume[box.to_slice()]
+ mask = volume[box.to_slice4d()]
else:
logging.fatal('Unsupported mask source: %s', config.to_json())