diff --git a/terratorch/models/decoders/upernet_decoder.py b/terratorch/models/decoders/upernet_decoder.py index ee4015e7..10f88def 100644 --- a/terratorch/models/decoders/upernet_decoder.py +++ b/terratorch/models/decoders/upernet_decoder.py @@ -37,7 +37,7 @@ def __init__( warnings.warn( "DeprecationWarning: scale_modules is deprecated and will be removed in future versions. " "Use LearnedInterpolateToPyramidal neck instead.", - stacklevel=2, + stacklevel=1, ) self.scale_modules = scale_modules