feat: Adding support for Annbatch

[ori-kron-wis]

No description provided.

[codecov]

Codecov Report

❌ Patch coverage is 81.50289% with 32 lines in your changes missing coverage. Please review.
✅ Project coverage is 72.69%. Comparing base (204a03c) to head (5745e31).
⚠️ Report is 1 commits behind head on main.

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

❌ Your project check has failed because the head coverage (72.69%) is below the target coverage (75.00%). You can increase the head coverage or adjust the target coverage.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #3620      +/-   ##
==========================================
+ Coverage   71.08%   72.69%   +1.61%     
==========================================
  Files         229      229              
  Lines       22617    22788     +171     
==========================================
+ Hits        16077    16566     +489     
+ Misses       6540     6222     -318     

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

... 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.

[chatgpt-codex-connector]

💡 Codex Review

Here are some automated review suggestions for this pull request.

Reviewed commit: 6ee2098c75

ℹ️ About Codex in GitHub

Your team has set up Codex to review pull requests in this repo. Reviews are triggered when you

• Open a pull request for review
• Mark a draft as ready
• Comment "@codex review".

If Codex has suggestions, it will comment; otherwise it will react with 👍.

Codex can also answer questions or update the PR. Try commenting "@codex address that feedback".