def __getitem__()

in src/data_manager.py [0:0]

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

    def __getitem__(self, index):
        target = self.targets[index]
        path = self.samples[index][0]
        img = self.dataset.loader(path)

        if self.dataset.target_transform is not None:
            target = self.dataset.target_transform(target)

        if self.dataset.transform is not None:
            if self.supervised:
                return *[self.dataset.transform(img) for _ in range(self.supervised_views)], target
            else:
                img_1 = self.dataset.transform(img)
                img_2 = self.dataset.transform(img)

                multicrop, mc_transform = self.multicrop_transform
                if multicrop > 0 and mc_transform is not None:
                    mc_imgs = [mc_transform(img) for _ in range(int(multicrop))]
                    return img_1, img_2, *mc_imgs, target

                return img_1, img_2, target

        return img, target