def reset_iteration()

in grok/data.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

    def reset_iteration(self, shuffle=True):
        self.index = 0
        if shuffle and self.dataset.train:
            self.permutation = torch.randperm(len(self.dataset))
        else:
            self.permutation = torch.arange(len(self.dataset))