MLServer icon indicating copy to clipboard operation
MLServer copied to clipboard

Control the output name when using `decode_args` decorator

Open AndriiG13 opened this issue 2 years ago • 2 comments

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.

AndriiG13 avatar Jan 19 '23 12:01 AndriiG13

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.

adriangonz avatar Jan 19 '23 15:01 adriangonz

Awesome! Glad to know it's on the roadmap :)

AndriiG13 avatar Jan 19 '23 15:01 AndriiG13