return []
seen[x] = (color_id[x], k // width, k % width)
- square_infos = zip(*seen.values())
+ square_infos = tuple(zip(*seen.values()))
square_c = torch.tensor(square_infos[0])
square_i = torch.tensor(square_infos[1])
square_j = torch.tensor(square_infos[2])