def collate()

in data_loaders.py [0:0]

• 10 lines of code
• 8 McCabe index (conditional complexity)

	def collate(self, batch):
		data = {}
		data['names'] = [item['names'] for item in batch]
		data['class'] = [item['class'] for item in batch]
		data['sheets'] = torch.cat([item['sheets'].unsqueeze(0) for item in batch])
		data['gt_points'] = torch.cat([item['gt_points'].unsqueeze(0) for item in batch])
		data['img_occ'] = torch.cat([item['img_occ'].unsqueeze(0) for item in batch])
		data['img_unocc'] = torch.cat([item['img_unocc'].unsqueeze(0) for item in batch])
		data['successful'] = [item['successful'] for item in batch]

		return data