diff --git a/src/anomalib/data/transforms/center_crop.py b/src/anomalib/data/transforms/center_crop.py index 88b8655aae..3747c4f6c6 100644 --- a/src/anomalib/data/transforms/center_crop.py +++ b/src/anomalib/data/transforms/center_crop.py @@ -9,6 +9,7 @@ # Copyright (C) 2024 Intel Corporation # SPDX-License-Identifier: Apache-2.0 +from collections.abc import Sequence from typing import Any import torch @@ -77,9 +78,9 @@ class ExportableCenterCrop(Transform): size (int | tuple[int, int]): Desired output size of the crop. """ - def __init__(self, size: int | tuple[int, int]) -> None: + def __init__(self, size: int | Sequence[int]) -> None: super().__init__() - self.size = list(size) if isinstance(size, tuple) else [size, size] + self.size = list(size) if isinstance(size, Sequence) else [size, size] def _transform(self, inpt: torch.Tensor, params: dict[str, Any]) -> torch.Tensor: """Apply the transform."""