Prediction on one sample produces error

[AlexanderZender]

NOTE: ISSUES ARE NOT FOR CODE HELP - Ask for Help at https://stackoverflow.com

Your issue may already be reported!
Also, please search on the issue tracker before creating one.

• I'm submitting a ...
  • bug report
  • feature request
  • support request => Please do not submit support request here, see note at the top of this template.

Issue Description

• When calling the proba_predict method with a single sample the following error is produced:

Prediction sets with more than one work just fine.

Expected Behavior

A single probability is returned.

Current Behavior

This error is returned deep within the predict pipeline:

Possible Solution

Your Code

Error message

Your Local environment

• Operating System, Ubuntu
• Python, version 3.10
• Outputs of pip freeze or conda list

[AlexanderZender]

My current workaround, duplicating the single row and only use the probabilities from one: