ONNX Runner Only return first of the outputs!

[cristozz]

BentoML/bentoml/_internal/frameworks/onnx.py

Line 415 in 28f0bc3

return self.predict_fns[method_name](output_names, input_names)[0]

I have a onnx model that have two outputs ['label', 'probabilities'] but when i use RUN only return 'label' values, and i dont see any configuration to return all outputs.

[parano]

Hi @cristozz, I believe this issue is addressed in #2912, it will be available in the upcoming release version 1.1.

@larme could you confirm if this is the same issue that #2912 fixed?

[petcai]

Hi guys, I am wondering whether we have any progress on this issue. Thanks

[parano]

Hi @petcai sorry about the delay in getting back to you here, this issue has been fixed in #2912 and released since 1.0.7