apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

misc testing fixes

This commit is contained in:
Sean Sube 2024-01-05 21:15:33 -06:00
parent 7d782842ed
commit e958eff3a8
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/onnx_web/chain/pipeline.py
View File

@ -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(

2
api/onnx_web/server/api.py
View File

@ -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")