Skip to content

Commit 8a26435

Browse files
authored
allow Tuple type in join_engine_outputs (#1120)
1 parent 4158a25 commit 8a26435

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/deepsparse/utils/data.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ def join_engine_outputs(
243243
244244
This is the opposite of `split_engine_inputs` and is meant to be used in tandem.
245245
"""
246-
assert all(isinstance(item, List) for item in batch_outputs)
246+
assert all(isinstance(item, (List, Tuple)) for item in batch_outputs)
247247

248248
candidate_output = list(map(numpy.concatenate, zip(*batch_outputs)))
249249

0 commit comments

Comments
 (0)