MLServer
MLServer copied to clipboard
Control the output name when using `decode_args` decorator
Hi, I've started using the decode_args decorator as I find it super useful! However, is there a way to specify the output name? I see that it's set to output-0.
I'm trying to chain some models together and would like to control the output name. Perhaps by passing it as an argument to decode_args or specifying it in model metadata.
Hey @AndriiG13 ,
Thanks for that feedback!
Unfortunately, at the moment it's hardcoded to output-X (where X is the index within the outputs list). However, our intention is to read these from the model metadata. This metadata info can be specified at the moment within the model-settings.json file (i.e. available inputs and outputs), but it's not used by the decode_args decorator yet.
Awesome! Glad to know it's on the roadmap :)