Yes, I have filed a bug about the fact that the test for the one hot routine does not catch all errors.
But for the moment, you need to figure out how to fix your code so that it works not just with that one test but with the other places where the results are used. There are other threads about the “shape” argument you need in that function in order to succeed. E.g. this one.
But I’m hoping from your comment that you have already figured that out! 