From be6e8261aa56b6a6040ed0dd853f0f43adae8592 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 28 Nov 2023 15:57:43 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- src/torchmetrics/functional/segmentation/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/torchmetrics/functional/segmentation/utils.py b/src/torchmetrics/functional/segmentation/utils.py index 78dfcf524e3..dfd219ceb47 100644 --- a/src/torchmetrics/functional/segmentation/utils.py +++ b/src/torchmetrics/functional/segmentation/utils.py @@ -384,7 +384,7 @@ def surface_distance( return dis[preds] -@functools.lru_cache() +@functools.lru_cache def get_neighbour_tables( spacing: Union[Tuple[int, int], Tuple[int, int, int]], device: Optional[torch.device] = None ) -> Tuple[Tensor, Tensor]: @@ -449,7 +449,7 @@ def table_contour_length(spacing: Tuple[int, int], device: Optional[torch.device return table, kernel -@functools.lru_cache() +@functools.lru_cache def table_surface_area(spacing: Tuple[int, int, int], device: Optional[torch.device] = None) -> Tuple[Tensor, Tensor]: """Create a table that maps neighbour codes to the surface area of the corresponding surface.