code: def reset(self):
          
          self.curr_idx = 0
          random.shuffle(self.idx)
          for buck in self.data:
              np.random.shuffle(buck)
docs: Resets the iterator to the beginning of the data.