The error is raised by a sklearn metric method, which gets unexpected inputs.
Your current code snippet doesn’t show it’s usage, so you would have to check the docs of the method and make sure to pass the right input values.
My best guess is that the metric method expects predictions, not logits.