bugfix: _batch_generator needs to always yield

2024-03-10 16:44:30 -04:00
parent 1af3d710ee
commit d8b816ef43
1 changed files with 1 additions and 1 deletions
@@ -312,7 +312,7 @@ class CLAPWrapper():
            # batch size is bigger than available audio/text items
            if next_batch_idx >= args0_len:
                inputs[0] = input_tmp[dataset_idx:]
-                return func(*tuple(inputs))
+                yield func(*tuple(inputs))
            else:
                inputs[0] = input_tmp[dataset_idx:next_batch_idx]
                yield func(*tuple(inputs))