def get_train_loader()

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

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

    def get_train_loader(self) -> DataLoader:
        """Get a PyTorch data loader for the training set."""
        train_loader = torch.utils.data.DataLoader(
            datasets.MNIST(
                self.dataset_path,
                train=True,
                download=self.download,
                transform=self.transform_fn,
            ),
            batch_size=self.train_batch_size,
            shuffle=True,
        )

        return train_loader