From 15ad494bdf29810c7b41eed10135b6576bcba6fb Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Fri, 15 Sep 2017 16:20:10 -0700 Subject: [PATCH] Fit output --- serve/main.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/serve/main.py b/serve/main.py index abdaf5d..abf2742 100644 --- a/serve/main.py +++ b/serve/main.py @@ -75,7 +75,10 @@ def worker(): trace = traceback.format_exc() recvQ.put((False, trace)) continue - recvQ.put((True, r.tolist())) + if isinstance(r, list): + recvQ.put((True, [elem.tolist() for elem in r])) + else: + recvQ.put((True, r.tolist())) if __name__ == '__main__': -- GitLab