@@ -24,9 +24,8 @@ def __init__(self, target_model: str):
2424 )
2525 self .client = httpclient .InferenceServerClient (service_url )
2626 if not self .client .is_model_ready (self .model_name , self .model_version ):
27- raise RuntimeError (
28- f"Requested model: { self .model_name } , version: { self .model_version } is not accessible"
29- )
27+ msg = f"Requested model: { self .model_name } , version: { self .model_version } is not accessible"
28+ raise RuntimeError (msg )
3029
3130 self .metadata = self .client .get_model_metadata (
3231 model_name = self .model_name ,
@@ -58,7 +57,9 @@ def get_output_layers(self):
5857 def infer_sync (self , dict_data ):
5958 inputs = _prepare_inputs (dict_data , self .inputs )
6059 raw_result = self .client .infer (
61- model_name = self .model_name , model_version = self .model_version , inputs = inputs
60+ model_name = self .model_name ,
61+ model_version = self .model_version ,
62+ inputs = inputs ,
6263 )
6364
6465 inference_results = {}
@@ -70,7 +71,9 @@ def infer_sync(self, dict_data):
7071 def infer_async (self , dict_data , callback_data ):
7172 inputs = _prepare_inputs (dict_data , self .inputs )
7273 raw_result = self .client .infer (
73- model_name = self .model_name , model_version = self .model_version , inputs = inputs
74+ model_name = self .model_name ,
75+ model_version = self .model_version ,
76+ inputs = inputs ,
7477 )
7578 inference_results = {}
7679 for output in self .metadata ["outputs" ]:
@@ -161,7 +164,8 @@ def _parse_model_arg(target_model: str):
161164 return service_url , model_spec [0 ], ""
162165 if len (model_spec ) == 2 :
163166 return service_url , model_spec [0 ], model_spec [1 ]
164- raise ValueError ("invalid target_model format" )
167+ msg = "Invalid target_model format"
168+ raise ValueError (msg )
165169
166170
167171def _prepare_inputs (dict_data , inputs_meta ):
@@ -180,7 +184,9 @@ def _prepare_inputs(dict_data, inputs_meta):
180184 input_data = np .array (input_data , dtype = model_precision )
181185
182186 infer_input = httpclient .InferInput (
183- input_name , input_data .shape , input_info .precision
187+ input_name ,
188+ input_data .shape ,
189+ input_info .precision ,
184190 )
185191 infer_input .set_data_from_numpy (input_data )
186192 inputs .append (infer_input )
0 commit comments