fix(mypy): address tensorstore issues

Author: nkemnitz

tests/unit/layer/volumetric/tensorstore/test_backend.py

@@ -1,4 +1,5 @@
 # pylint: disable=missing-docstring,redefined-outer-name,unused-argument,pointless-statement,line-too-long,protected-access,too-few-public-methods
+# mypy: disable-error-code="misc,method-assign,attr-defined"
 import os
 import pathlib
 from copy import deepcopy

zetta_utils/layer/volumetric/tensorstore/backend.py

@@ -309,11 +309,15 @@ def with_changes(self, **kwargs) -> TSBackend:
 
     def get_voxel_offset(self, resolution: Vec3D) -> Vec3D[int]:
         ts = _get_ts_at_resolution(self.path, self.cache_bytes_limit, str(list(resolution)))
-        return Vec3D[int](*ts.chunk_layout.grid_origin[0:3])
+        grid_origin = ts.chunk_layout.grid_origin
+        assert grid_origin is not None
+        return Vec3D[int](*grid_origin[0:3])
 
     def get_chunk_size(self, resolution: Vec3D) -> Vec3D[int]:
         ts = _get_ts_at_resolution(self.path, self.cache_bytes_limit, str(list(resolution)))
-        return Vec3D[int](*ts.chunk_layout.read_chunk.shape[0:3])
+        chunk_shape = ts.chunk_layout.read_chunk.shape
+        assert chunk_shape is not None
+        return Vec3D[int](*chunk_shape[0:3])
 
     def get_dataset_size(self, resolution: Vec3D) -> Vec3D[int]:
         ts = _get_ts_at_resolution(self.path, self.cache_bytes_limit, str(list(resolution)))

zetta_utils/layer/volumetric/tensorstore/deprecated/backend.py

@@ -294,11 +294,15 @@ def with_changes(self, **kwargs) -> TSBackend:
 
     def get_voxel_offset(self, resolution: Vec3D) -> Vec3D[int]:
         ts = _get_ts_at_resolution(self.path, self.cache_bytes_limit, str(list(resolution)))
-        return Vec3D[int](*ts.chunk_layout.grid_origin[0:3])
+        grid_origin = ts.chunk_layout.grid_origin
+        assert grid_origin is not None
+        return Vec3D[int](*grid_origin[0:3])
 
     def get_chunk_size(self, resolution: Vec3D) -> Vec3D[int]:
         ts = _get_ts_at_resolution(self.path, self.cache_bytes_limit, str(list(resolution)))
-        return Vec3D[int](*ts.chunk_layout.read_chunk.shape[0:3])
+        chunk_shape = ts.chunk_layout.read_chunk.shape
+        assert chunk_shape is not None
+        return Vec3D[int](*chunk_shape[0:3])
 
     def get_dataset_size(self, resolution: Vec3D) -> Vec3D[int]:
         ts = _get_ts_at_resolution(self.path, self.cache_bytes_limit, str(list(resolution)))