fix: handle different input data types

[ilan-gold]

No description provided.

[codecov]

Codecov Report

❌ Patch coverage is 97.72727% with 2 lines in your changes missing coverage. Please review.
✅ Project coverage is 91.91%. Comparing base (9833c40) to head (1a12389).

Files with missing lines	Patch %	Lines
src/annbatch/loader.py	97.72%	2 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #233      +/-   ##
==========================================
- Coverage   93.12%   91.91%   -1.22%     
==========================================
  Files          15       15              
  Lines        1324     1397      +73     
==========================================
+ Hits         1233     1284      +51     
- Misses         91      113      +22     

Files with missing lines	Coverage Δ
src/annbatch/loader.py	87.73% <97.72%> (-1.16%)	⬇️

... and 2 files with indirect coverage changes

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.