bugfix in BatchGenerator, iter should return self not next(self)

This commit is contained in:
Joerg Franke 2018-12-28 20:45:31 +01:00
parent b2ae8de21e
commit 390fb1f62e

View File

@ -62,7 +62,7 @@ class BatchGenerator():
self.order = self.data_set.rng.permutation(self.order) self.order = self.data_set.rng.permutation(self.order)
def __iter__(self): def __iter__(self):
return next(self) return self
def __next__(self): def __next__(self):
""" """