bugfix (NVIDIA#6122)

Signed-off-by: Dima Rekesh <[email protected]> Co-authored-by: Dima Rekesh <[email protected]> Signed-off-by: hsiehjackson <[email protected]>
hsiehjackson · Jun 2, 2023 · 4657e0b · 4657e0b
1 parent f97fd9f
commit 4657e0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/asr/data/audio_to_text.py b/nemo/collections/asr/data/audio_to_text.py
@@ -1195,7 +1195,7 @@ def __init__(self, tokenizer):
                 self._tokenizer = tokenizer
 
             def __call__(self, *args):
-                if isinstance(args[0], Iterable) and self.is_aggregate:
+                if isinstance(args[0], List) and self.is_aggregate:
                     t = []
                     for span in args[0]:
                         t.extend(self._tokenizer.text_to_ids(span['str'], span['lang']))