def sample_partial_test()

in part_selector.py [0:0]

• 16 lines of code
• 5 McCabe index (conditional complexity)

    def sample_partial_test(self, n):
        sample_ids = [np.random.randint(self.__len__()) for _ in range(n)]
        samples_partid = [torch.LongTensor([self.parts_id[sample_id]]) for sample_id in sample_ids]
        sample_jsons = [json.load(open(self.paths[sample_id]))for sample_id in sample_ids]
        samples_partial = []
        for sample_json in sample_jsons:
            input_parts_json = sample_json['input_parts']
            img_partial_test = []
            vector_input_part = []
            for i in range(self.n_part):
                key = self.id_to_part[i]
                vector_input_part += input_parts_json[key]
                img_partial_test.append(self.processed_part_to_raster(input_parts_json[key], side=self.image_size))
            img_partial_test.append(self.processed_part_to_raster(vector_input_part, side=self.image_size))
            samples_partial.append(torch.cat(img_partial_test, 0))
        return torch.stack(samples_partid), torch.stack(samples_partial)