feat: Adding support for Annbatch

[ori-kron-wis]

No description provided.

[codecov]

Codecov Report

❌ Patch coverage is 85.29412% with 5 lines in your changes missing coverage. Please review.
✅ Project coverage is 76.33%. Comparing base (370c642) to head (7e1a210).

Files with missing lines	Patch %	Lines
src/scvi/dataloaders/_custom_dataloaders.py	84.84%	5 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #3620      +/-   ##
==========================================
+ Coverage   74.70%   76.33%   +1.63%     
==========================================
  Files         225      225              
  Lines       21631    21664      +33     
==========================================
+ Hits        16159    16537     +378     
+ Misses       5472     5127     -345     

Files with missing lines	Coverage Δ
src/scvi/dataloaders/__init__.py	100.00% <100.00%> (ø)
src/scvi/dataloaders/_custom_dataloaders.py	88.02% <84.84%> (+61.71%)	⬆️

... and 10 files with indirect coverage changes

🚀 New features to boost your workflow:

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