As a follow up of this pr https://github.com/microsoft/winml-cli/pull/788/changes
As a follow up of this pr https://github.com/microsoft/winml-cli/pull/788/changes