diff --git a/pesto/model.py b/pesto/model.py index 27241271f15211799e8573bac077f56fa95f400d..a728e3522000b07e07e305d615b6c7f2f66785fd 100644 --- a/pesto/model.py +++ b/pesto/model.py @@ -161,7 +161,7 @@ class PESTO(nn.Module): def __init__(self, encoder: nn.Module, preprocessor: nn.Module, - crop_kwargs: Mapping[str, Any] | None = None, + crop_kwargs: Optional[Mapping[str, Any]] = None, reduction: str = "alwa"): super(PESTO, self).__init__() self.encoder = encoder