Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Nov 28, 2023
1 parent ec8a63e commit be6e826
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/torchmetrics/functional/segmentation/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]:
Expand Down Expand Up @@ -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.
Expand Down

0 comments on commit be6e826

Please sign in to comment.