[Serving]update np.object to np.object_ (#1021)

np.object to np.object_
This commit is contained in:
heliqi
2022-12-30 16:43:47 +08:00
committed by GitHub
parent 665f2e549a
commit 6310ddc8d6
5 changed files with 5 additions and 5 deletions

View File

@@ -141,7 +141,7 @@ class TritonPythonModel:
self.uie_model_.set_schema(schema)
results = self.uie_model_.predict(texts, return_dict=True)
results = np.array(results, dtype=np.object)
results = np.array(results, dtype=np.object_)
out_tensor = pb_utils.Tensor(self.output_names[0], results)
inference_response = pb_utils.InferenceResponse(
output_tensors=[out_tensor, ])