From e1e037ba52a5c18068fb9f0392a8895f760c2676 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 21 Mar 2022 22:00:33 -0700 Subject: [PATCH] Fix bad tuple typing fix that was on XLA branch bust missed on master merge --- timm/models/swin_transformer_v2_cr.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/swin_transformer_v2_cr.py b/timm/models/swin_transformer_v2_cr.py index 85cbddba..de0c4f14 100644 --- a/timm/models/swin_transformer_v2_cr.py +++ b/timm/models/swin_transformer_v2_cr.py @@ -120,8 +120,8 @@ 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) window_size (Tuple[int, int]): Window size