From 04be627e7bc9cd9ce1ef76b8a11d4ee3029c213f Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Wed, 3 Jan 2024 23:32:33 -0600 Subject: [PATCH] pass server when serializing metadata --- api/onnx_web/server/api.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py index 7d1fa79d..c9a503c4 100644 --- a/api/onnx_web/server/api.py +++ b/api/onnx_web/server/api.py @@ -102,6 +102,7 @@ def job_reply(name: str): def image_reply( + server: ServerContext, name: str, status: str, job_type: str, @@ -138,7 +139,7 @@ def image_reply( logger.error("metadata and outputs must be the same length") return error_reply("metadata and outputs must be the same length") - data["metadata"] = [m.tojson() for m in metadata] + data["metadata"] = [m.tojson(server, [o]) for m, o in zip(metadata, outputs)] data["outputs"] = outputs return jsonify([data]) @@ -678,6 +679,7 @@ def job_status(server: ServerContext, pool: DevicePoolExecutor): metadata = progress.result.metadata return image_reply( + server, job_name, status, "TODO", @@ -688,7 +690,7 @@ def job_status(server: ServerContext, pool: DevicePoolExecutor): metadata=metadata, ) - return image_reply(job_name, status, "TODO") + return image_reply(server, job_name, status, "TODO") def register_api_routes(app: Flask, server: ServerContext, pool: DevicePoolExecutor):