diff --git a/mlp/data_providers.py b/mlp/data_providers.py index 3d26d5e..c20063b 100644 --- a/mlp/data_providers.py +++ b/mlp/data_providers.py @@ -101,7 +101,7 @@ class DataProvider(object): self.shuffle() def __next__(self): - self.next() + return self.next() def reset(self): """Resets the provider to the initial state."""