Add warning for unexpected model output in batched prediction

[grumpyp]

Before submitting

• Was this discussed/agreed via a Github issue? (no need for typos and docs improvements)
  unexpected output for HF model with matching #294
• Did you read the contributor guideline, Pull Request section?
• Did you make sure to update the docs?
  is this needed?
• Did you write any new necessary tests?

What does this PR do?

Fixes #294 .

PR review

Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in GitHub issues there's a high chance it will not be merged.

Did you have fun?

Yes, the Debugger is my new friend :D

Question:

Do I need to add this warning to the docs?

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 95%. Comparing base (6e9dca6) to head (4bbc92f).
Report is 1 commits behind head on main.

Additional details and impacted files

@@         Coverage Diff         @@
##           main   #300   +/-   ##
===================================
- Coverage    95%    95%   -0%     
===================================
  Files        19     19           
  Lines      1244   1237    -7     
===================================
- Hits       1182   1170   -12     
- Misses       62     67    +5     

[aniketmaurya]

Thanks for updating the PR @grumpyp. Looks great! added minor comment and then we can merge.