fix: float resolutions for

ZettaAI · Apr 29, 2024 · 8469c1a · 8469c1a
1 parent 88f57ce
commit 8469c1a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/zetta_utils/layer/volumetric/cloudvol/build.py b/zetta_utils/layer/volumetric/cloudvol/build.py
@@ -34,7 +34,7 @@ def build_cv_layer(  # pylint: disable=too-many-locals
     info_dataset_size_map: dict[str, Sequence[int]] | None = None,
     info_voxel_offset: Sequence[int] | None = None,
     info_voxel_offset_map: dict[str, Sequence[int]] | None = None,
-    info_add_scales: Sequence[Sequence[int] | dict[str, Any]] | None = None,
+    info_add_scales: Sequence[Sequence[float] | dict[str, Any]] | None = None,
     info_add_scales_ref: str | dict[str, Any] | None = None,
     info_add_scales_mode: Literal["merge", "replace"] = "merge",
     on_info_exists: InfoExistsModes = "expect_same",

diff --git a/zetta_utils/layer/volumetric/precomputed/precomputed.py b/zetta_utils/layer/volumetric/precomputed/precomputed.py
@@ -86,7 +86,7 @@ def _get_ref_scale(
     return matched[0]
 
 
-def _make_scale(ref: dict[str, Any], target: Sequence[int] | dict[str, Any]) -> dict[str, Any]:
+def _make_scale(ref: dict[str, Any], target: Sequence[float] | dict[str, Any]) -> dict[str, Any]:
     """Make a single scale based on the reference scale"""
     ret = {}
     if isinstance(target, dict):
@@ -148,7 +148,7 @@ class PrecomputedInfoSpec:
     voxel_offset_map: dict[str, Sequence[int]] | None = None
     dataset_size_map: dict[str, Sequence[int]] | None = None
     data_type: str | None = None
-    add_scales: Sequence[Sequence[int] | dict[str, Any]] | None = None
+    add_scales: Sequence[Sequence[float] | dict[str, Any]] | None = None
     add_scales_ref: str | dict[str, Any] | None = None
     add_scales_mode: str = "merge"
     # ensure_scales: Optional[Iterable[int]] = None