Predictor for multi-input models: present as list instead of tuple…

(because TF-Serving expects that and cannot cast)
This commit is contained in:
Robert Sachunsky 2026-06-03 20:56:00 +02:00
parent 4e7e1c06b9
commit 38fe4d33ad

View file

@ -184,7 +184,7 @@ class Predictor(mp.context.SpawnProcess):
else:
data.append(stack.enter_context(ndarray_shared(shared_data)))
if multi_input:
data = tuple(np.concatenate(data0)
data = list(np.concatenate(data0)
for data0 in zip(*data))
else:
data = np.concatenate(data)