From 390fb1f62e6fa3dd4b4a899403dbb62dffe81dc1 Mon Sep 17 00:00:00 2001
From: Joerg Franke <joerg-franke@outlook.com>
Date: Fri, 28 Dec 2018 20:45:31 +0100
Subject: [PATCH] bugfix in BatchGenerator, iter should return self not
 next(self)

---
 adnc/data/utils/batch_generator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/adnc/data/utils/batch_generator.py b/adnc/data/utils/batch_generator.py
index 6f052b4..0b34763 100755
--- a/adnc/data/utils/batch_generator.py
+++ b/adnc/data/utils/batch_generator.py
@@ -62,7 +62,7 @@ class BatchGenerator():
                     self.order = self.data_set.rng.permutation(self.order)
 
     def __iter__(self):
-        return next(self)
+        return self
 
     def __next__(self):
         """