18
18
import numpy as np
19
19
import requests
20
20
import tritonclient .http as httpclient
21
- from attrdict import AttrDict
22
-
23
-
24
- def convert_http_metadata_config (metadata ):
25
- metadata = AttrDict (metadata )
26
-
27
- return metadata
28
21
29
22
30
23
def prepare_request (inputs_meta , inputs_data , outputs_meta ):
@@ -58,7 +51,7 @@ def prepare_request(inputs_meta, inputs_data, outputs_meta):
58
51
inputs .append (infer_input )
59
52
outputs = []
60
53
for output_dict in outputs_meta :
61
- infer_output = httpclient .InferRequestedOutput (output_dict . name )
54
+ infer_output = httpclient .InferRequestedOutput (output_dict [ ' name' ] )
62
55
outputs .append (infer_output )
63
56
return inputs , outputs
64
57
@@ -321,8 +314,8 @@ def infer(self, server_url, model_name, model_version, inputs):
321
314
322
315
results = {}
323
316
for output in output_metadata :
324
- result = response .as_numpy (output . name ) # datatype: numpy
325
- if output . datatype == 'BYTES' : # datatype: bytes
317
+ result = response .as_numpy (output [ ' name' ] ) # datatype: numpy
318
+ if output [ ' datatype' ] == 'BYTES' : # datatype: bytes
326
319
try :
327
320
value = result
328
321
if len (result .shape ) == 1 :
@@ -336,7 +329,7 @@ def infer(self, server_url, model_name, model_version, inputs):
336
329
pass
337
330
else :
338
331
result = result [0 ]
339
- results [output . name ] = result
332
+ results [output [ ' name' ] ] = result
340
333
return results
341
334
342
335
def raw_infer (self , server_url , model_name , model_version , raw_input ):
@@ -353,8 +346,6 @@ def get_model_meta(self, server_url, model_name, model_version):
353
346
except Exception as e :
354
347
raise RuntimeError ("Failed to retrieve the metadata: " + str (e ))
355
348
356
- model_metadata = convert_http_metadata_config (model_metadata )
357
-
358
- input_metadata = model_metadata .inputs
359
- output_metadata = model_metadata .outputs
349
+ input_metadata = model_metadata ['inputs' ]
350
+ output_metadata = model_metadata ['outputs' ]
360
351
return input_metadata , output_metadata
0 commit comments