qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-06-16 09:59:13 +02:00
Code Issues Projects Releases Wiki Activity

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

Browse source 
(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
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/eynollah/predictor.py
View file

@ -184,8 +184,8 @@ class Predictor(mp.context.SpawnProcess):
else: else:
data.append(stack.enter_context(ndarray_shared(shared_data))) data.append(stack.enter_context(ndarray_shared(shared_data)))
if multi_input: if multi_input:
data = tuple(np.concatenate(data0) data = list(np.concatenate(data0)
for data0 in zip(*data)) for data0 in zip(*data))
else: else:
data = np.concatenate(data) data = np.concatenate(data)
#result = self.model.predict(data, verbose=0) #result = self.model.predict(data, verbose=0)