From 2c30e1e506535fb653c8ae8715883fdbc941092f Mon Sep 17 00:00:00 2001 From: Sergiy Date: Tue, 28 Jan 2025 21:32:40 +0000 Subject: [PATCH] chore: black reformatting --- tests/unit/layer/volumetric/cloudvol/test_backend.py | 8 ++++---- tests/unit/tensor_ops/test_generators.py | 2 -- zetta_utils/geometry/bbox_strider.py | 6 +++--- zetta_utils/layer/volumetric/cloudvol/backend.py | 2 +- 4 files changed, 8 insertions(+), 10 deletions(-) diff --git a/tests/unit/layer/volumetric/cloudvol/test_backend.py b/tests/unit/layer/volumetric/cloudvol/test_backend.py index d14cb4082..6ae51a9ba 100644 --- a/tests/unit/layer/volumetric/cloudvol/test_backend.py +++ b/tests/unit/layer/volumetric/cloudvol/test_backend.py @@ -195,8 +195,8 @@ def test_cv_backend_write_scalar(clear_caches_reset_mocks, mocker): def test_cv_backend_read_uint63(clear_caches_reset_mocks, mocker): - data_read = np.array([[[[2**63 - 1]]]], dtype=np.uint64) - expected = np.array([[[[2**63 - 1]]]], dtype=np.int64) + data_read = np.array([[[[2 ** 63 - 1]]]], dtype=np.uint64) + expected = np.array([[[[2 ** 63 - 1]]]], dtype=np.int64) cv_m = mocker.MagicMock() cv_m.__getitem__ = mocker.MagicMock(return_value=data_read) mocker.patch("cloudvolume.CloudVolume.__new__", return_value=cv_m) @@ -222,8 +222,8 @@ def test_cv_backend_write_scalar_uint63(clear_caches_reset_mocks, mocker): cv_m.dtype = "uint64" mocker.patch("cloudvolume.CloudVolume.__new__", return_value=cv_m) cvb = CVBackend(path=LAYER_SCRATCH0_PATH, info_spec=info_spec, on_info_exists="overwrite") - value = np.array([2**63 - 1], dtype=np.int64) - expected_written = np.uint64(2**63 - 1) + value = np.array([2 ** 63 - 1], dtype=np.int64) + expected_written = np.uint64(2 ** 63 - 1) index = VolumetricIndex( bbox=BBox3D.from_slices((slice(0, 1), slice(0, 1), slice(0, 1))), diff --git a/tests/unit/tensor_ops/test_generators.py b/tests/unit/tensor_ops/test_generators.py index 2a9cc58f6..72028f717 100644 --- a/tests/unit/tensor_ops/test_generators.py +++ b/tests/unit/tensor_ops/test_generators.py @@ -251,7 +251,6 @@ def test_get_field_from_matrix_exceptions(mat, size): [[[[nan, nan], [0.0, nan]], [[nan, nan], [-1.0, nan]]]], dtype=torch.float32 ), ], - # Two vectors, same target pixel -- should average [ [ @@ -267,7 +266,6 @@ def test_get_field_from_matrix_exceptions(mat, size): [[[[0.0, nan], [nan, nan]], [[0.0, nan], [nan, nan]]]], dtype=torch.float32 ), ], - # Annotation outside ROI [ [ diff --git a/zetta_utils/geometry/bbox_strider.py b/zetta_utils/geometry/bbox_strider.py index 61e3eb3d2..994e30973 100644 --- a/zetta_utils/geometry/bbox_strider.py +++ b/zetta_utils/geometry/bbox_strider.py @@ -83,9 +83,9 @@ def __attrs_post_init__(self) -> None: object.__setattr__(self, "stride", superchunk_size) object.__setattr__(self, "mode", "exact") object.__setattr__(self, "max_superchunk_size", None) - object.__setattr__(self, - "stride_start_offset", - self.bbox_snapped.start / self.resolution) + object.__setattr__( + self, "stride_start_offset", self.bbox_snapped.start / self.resolution + ) object.__setattr__(self, "stride_start_offset_in_unit", self.bbox_snapped.start) self.__attrs_post_init__() diff --git a/zetta_utils/layer/volumetric/cloudvol/backend.py b/zetta_utils/layer/volumetric/cloudvol/backend.py index bdffea6de..89cddb7d8 100644 --- a/zetta_utils/layer/volumetric/cloudvol/backend.py +++ b/zetta_utils/layer/volumetric/cloudvol/backend.py @@ -21,7 +21,7 @@ _cv_cache: cachetools.LRUCache = cachetools.LRUCache(maxsize=16) _cv_cached: Dict[str, set] = {} -IN_MEM_CACHE_NUM_BYTES_PER_CV = 128 * 1024**2 +IN_MEM_CACHE_NUM_BYTES_PER_CV = 128 * 1024 ** 2 # To avoid reloading info file - note that an empty provenance is passed # since otherwise the CloudVolume's __new__ will download the provenance