From 15cc9eae3efb54019668b1761312c273cd709764 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 28 Feb 2022 16:44:24 -0800 Subject: [PATCH] Fix Swin v2 tuple type hint --- timm/models/swin_transformer_v2_cr.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/swin_transformer_v2_cr.py b/timm/models/swin_transformer_v2_cr.py index bb77466f..39ea993e 100644 --- a/timm/models/swin_transformer_v2_cr.py +++ b/timm/models/swin_transformer_v2_cr.py @@ -117,7 +117,7 @@ def window_partition(x, window_size: Tuple[int, int]): @register_notrace_function # reason: int argument is a Proxy -def window_reverse(windows, window_size: tuple[int, int], img_size: tuple[int, int]): +def window_reverse(windows, window_size: Tuple[int, int], img_size: Tuple[int, int]): """ Args: windows: (num_windows * B, window_size[0], window_size[1], C)