From e958eff3a862af334194e0319b40ea66fbc26906 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Fri, 5 Jan 2024 21:15:33 -0600 Subject: [PATCH] misc testing fixes --- api/onnx_web/chain/pipeline.py | 3 ++- api/onnx_web/server/api.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/chain/pipeline.py b/api/onnx_web/chain/pipeline.py index 1a6a597b..c785fce0 100644 --- a/api/onnx_web/chain/pipeline.py +++ b/api/onnx_web/chain/pipeline.py @@ -235,8 +235,9 @@ class ChainPipeline: **kwargs, ) + metadata = stage_sources.metadata stage_sources = StageResult( - images=stage_results, metadata=stage_sources.metadata + images=stage_results, metadata=metadata ) else: logger.debug( diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py index 1166c0dd..26777ce6 100644 --- a/api/onnx_web/server/api.py +++ b/api/onnx_web/server/api.py @@ -129,7 +129,7 @@ def image_reply( } if outputs is not None: - if metadata is not None: + if metadata is None: logger.error("metadata is required with outputs") return error_reply("metadata is required with outputs")