- square_c = torch.tensor(square_infos[0])
- square_i = torch.tensor(square_infos[1])
- square_j = torch.tensor(square_infos[2])
+ if square_infos:
+ square_c = torch.tensor(square_infos[0])
+ square_i = torch.tensor(square_infos[1])
+ square_j = torch.tensor(square_infos[2])
+ else:
+ square_c = torch.tensor([])
+ square_i = torch.tensor([])
+ square_j = torch.tensor([])