As discussed in https://github.com/microsoft/MLOS/pull/967#issuecomment-2825025256 We need to - [ ] fix the bug - [ ] add tests for this
As discussed in #967 (comment)
We need to