From e952a98103aacbb6cd68abff5492d6fa2ce99cda Mon Sep 17 00:00:00 2001 From: Sauyon Lee Date: Mon, 29 Aug 2022 06:42:49 -0700 Subject: [PATCH] fix: fix context parameter for multi-input IO descriptors --- bentoml/_internal/service/inference_api.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/bentoml/_internal/service/inference_api.py b/bentoml/_internal/service/inference_api.py index 10e505e2f5..01b7322deb 100644 --- a/bentoml/_internal/service/inference_api.py +++ b/bentoml/_internal/service/inference_api.py @@ -57,8 +57,8 @@ def __init__( if isinstance(input_type, dict): # note: in python 3.6 kwarg order was not guaranteed to be preserved, # though it is in practice. - for key in input_type: - if key not in sig.parameters: + for key in sig.parameters: + if key not in input_type: if ( key in ["context", "ctx"] or sig.parameters[key].annotation == Context