diff --git a/mlp/data_providers.py b/mlp/data_providers.py index 91435e5..1d18794 100644 --- a/mlp/data_providers.py +++ b/mlp/data_providers.py @@ -133,11 +133,11 @@ class MNISTDataProvider(DataProvider): super(MNISTDataProvider, self).__init__( inputs, targets, batch_size, max_num_batches, shuffle_order, rng) - def next(self): - """Returns next data batch or raises `StopIteration` if at end.""" - inputs_batch, targets_batch = super(MNISTDataProvider, self).next() - return inputs_batch, self.to_one_of_k(targets_batch) - + # def next(self): + # """Returns next data batch or raises `StopIteration` if at end.""" + # inputs_batch, targets_batch = super(MNISTDataProvider, self).next() + # return inputs_batch, self.to_one_of_k(targets_batch) + # def __next__(self): return self.next()