def get_test_loader()

in quant/data/data_loaders.py [0:0]

• 22 lines of code
• 1 McCabe index (conditional complexity)

    def get_test_loader(self) -> DataLoader:
        """Get a PyTorch data loader for the test set."""
        test_dir = Path(self.dataset_path) / self.val_split
        test_dataset = datasets.ImageFolder(
            test_dir,
            transforms.Compose(
                [
                    transforms.Resize(256),
                    transforms.CenterCrop(224),
                    transforms.ToTensor(),
                    self.normalize,
                ]
            ),
        )

        test_loader = torch.utils.data.DataLoader(
            test_dataset,
            batch_size=self.test_batch_size,
            shuffle=False,
            num_workers=self.workers,
            pin_memory=True,
            sampler=self.test_sampler,
        )

        return test_loader