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.51%. Comparing base (bf57711) to head (990e375).

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

❗ There is a different number of reports uploaded between BASE (bf57711) and HEAD (990e375). Click for more details.

HEAD has 45 uploads less than BASE

Flag	BASE (bf57711)	HEAD (990e375)
	49	4

Additional details and impacted files

@@             Coverage Diff             @@
##             main    #3620       +/-   ##
===========================================
- Coverage   86.51%   76.51%   -10.01%     
===========================================
  Files         225      225               
  Lines       21637    21670       +33     
===========================================
- Hits        18719    16580     -2139     
- Misses       2918     5090     +2172     

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

... and 54 files with indirect coverage changes

🚀 New features to boost your workflow:

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