bugfix: _batch_generator needs to always yield

microsoft · Mar 10, 2024 · d8b816e · d8b816e
1 parent 1af3d71
commit d8b816e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/msclap/CLAPWrapper.py b/msclap/CLAPWrapper.py
@@ -312,7 +312,7 @@ def _generic_batch_inference(self, func, *args):
             # 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))