Fix bad tuple typing fix that was on XLA branch bust missed on master…

… merge
ares89 · Mar 22, 2022 · e1e037b · e1e037b
1 parent 341b464
commit e1e037b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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