fix bucketing bug issue for picking new bucket (#6663)

Signed-off-by: Nithin Rao Koluguri <nithinraok> Co-authored-by: Nithin Rao Koluguri <nithinraok>
NVIDIA · May 17, 2023 · 78fe893 · 78fe893
1 parent e9fcc41
commit 78fe893
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/nemo/collections/asr/data/audio_to_text.py b/nemo/collections/asr/data/audio_to_text.py
@@ -1341,8 +1341,7 @@ def __next__(self):
             try:
                 sample = next(self.wrapped_iter)
             except StopIteration:
-                self.wrapped_iter = iter(self.wrapped_ds)
-                sample = next(self.wrapped_iter)
+                break
             batches.append(sample)
         if len(batches) == 0:
             raise StopIteration