Skip to content

Commit e420b8c

Browse files
authored
fix coompatible pproblem for python3.10 and gradio (#1249)
1 parent 6ae0838 commit e420b8c

File tree

2 files changed

+7
-17
lines changed

2 files changed

+7
-17
lines changed

requirements.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,7 @@ packaging
1212
x2paddle >= 1.4.0
1313
paddle2onnx >= 1.0.5
1414
rarfile
15-
gradio
15+
gradio == 3.11.0
1616
tritonclient[all]
17-
attrdict
1817
psutil
1918
onnx >= 1.6.0

visualdl/component/inference/fastdeploy_client/http_client_manager.py

Lines changed: 6 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -18,13 +18,6 @@
1818
import numpy as np
1919
import requests
2020
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
2821

2922

3023
def prepare_request(inputs_meta, inputs_data, outputs_meta):
@@ -58,7 +51,7 @@ def prepare_request(inputs_meta, inputs_data, outputs_meta):
5851
inputs.append(infer_input)
5952
outputs = []
6053
for output_dict in outputs_meta:
61-
infer_output = httpclient.InferRequestedOutput(output_dict.name)
54+
infer_output = httpclient.InferRequestedOutput(output_dict['name'])
6255
outputs.append(infer_output)
6356
return inputs, outputs
6457

@@ -321,8 +314,8 @@ def infer(self, server_url, model_name, model_version, inputs):
321314

322315
results = {}
323316
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
326319
try:
327320
value = result
328321
if len(result.shape) == 1:
@@ -336,7 +329,7 @@ def infer(self, server_url, model_name, model_version, inputs):
336329
pass
337330
else:
338331
result = result[0]
339-
results[output.name] = result
332+
results[output['name']] = result
340333
return results
341334

342335
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):
353346
except Exception as e:
354347
raise RuntimeError("Failed to retrieve the metadata: " + str(e))
355348

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']
360351
return input_metadata, output_metadata

0 commit comments

Comments
 (0)