diff --git a/zetta_utils/layer/volumetric/constant/backend.py b/zetta_utils/layer/volumetric/constant/backend.py index 2ed0941b0..37555b682 100644 --- a/zetta_utils/layer/volumetric/constant/backend.py +++ b/zetta_utils/layer/volumetric/constant/backend.py @@ -1,6 +1,7 @@ # pylint: disable=missing-docstring from __future__ import annotations +import sys from typing import Literal, Union import attrs @@ -11,6 +12,8 @@ from .. import VolumetricBackend, VolumetricIndex +MAXSIZE_HALF = (sys.maxsize + 1) // 2 + @attrs.mutable class ConstantVolumetricBackend(VolumetricBackend): # pylint: disable=too-few-public-methods @@ -106,7 +109,11 @@ def get_dataset_size(self, resolution: Vec3D) -> Vec3D[int]: # pragma: no cover return Vec3D[int](0, 0, 0) def get_bounds(self, resolution: Vec3D) -> VolumetricIndex: # pragma: no cover - return VolumetricIndex.from_coords((0, 0, 0), (0, 0, 0), Vec3D[int](1, 1, 1)) + return VolumetricIndex.from_coords( + (-MAXSIZE_HALF, -MAXSIZE_HALF, -MAXSIZE_HALF), + (MAXSIZE_HALF, MAXSIZE_HALF, MAXSIZE_HALF), + Vec3D[int](1, 1, 1), + ) def get_chunk_aligned_index( self, idx: VolumetricIndex, mode: Literal["expand", "shrink", "round"]