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

pull/1187/head
Ross Wightman 3 years ago
parent 341b464a5a
commit e1e037ba52

@ -120,8 +120,8 @@ def window_partition(x, window_size: Tuple[int, int]):
@register_notrace_function # reason: int argument is a Proxy @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: Args:
windows: (num_windows * B, window_size[0], window_size[1], C) windows: (num_windows * B, window_size[0], window_size[1], C)
window_size (Tuple[int, int]): Window size window_size (Tuple[int, int]): Window size

Loading…
Cancel
Save