Custom dataloader registry support

[ori-kron-wis]

No description provided.

[codecov]

Codecov Report

Attention: Patch coverage is 45.04505% with 305 lines in your changes missing coverage. Please review.

Project coverage is 79.05%. Comparing base (d0abcdc) to head (4e1dcaf).

Files with missing lines	Patch %	Lines
src/scvi/dataloaders/_custom_dataloders.py	40.06%	190 Missing ⚠️
src/scvi/model/base/_base_model.py	37.28%	74 Missing ⚠️
src/scvi/model/_scanvi.py	56.09%	18 Missing ⚠️
src/scvi/model/_scvi.py	50.00%	12 Missing ⚠️
src/scvi/model/base/_archesmixin.py	77.41%	7 Missing ⚠️
src/scvi/model/base/_training_mixin.py	76.92%	3 Missing ⚠️
src/scvi/model/base/_save_load.py	90.00%	1 Missing ⚠️

❗ There is a different number of reports uploaded between BASE (d0abcdc) and HEAD (4e1dcaf). Click for more details.

HEAD has 9 uploads less than BASE

Flag	BASE (d0abcdc)	HEAD (4e1dcaf)
	12	3

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #2932      +/-   ##
==========================================
- Coverage   86.69%   79.05%   -7.64%     
==========================================
  Files         193      194       +1     
  Lines       17115    17601     +486     
==========================================
- Hits        14838    13915     -923     
- Misses       2277     3686    +1409     

Files with missing lines	Coverage Δ
src/scvi/dataloaders/__init__.py	100.00% <100.00%> (ø)
src/scvi/dataloaders/_data_splitting.py	95.47% <ø> (ø)
src/scvi/model/base/_save_load.py	82.17% <90.00%> (-4.14%)	⬇️
src/scvi/model/base/_training_mixin.py	88.38% <76.92%> (-1.42%)	⬇️
src/scvi/model/base/_archesmixin.py	76.36% <77.41%> (-0.53%)	⬇️
src/scvi/model/_scvi.py	80.00% <50.00%> (-15.92%)	⬇️
src/scvi/model/_scanvi.py	79.82% <56.09%> (-13.20%)	⬇️
src/scvi/model/base/_base_model.py	79.38% <37.28%> (-13.49%)	⬇️
src/scvi/dataloaders/_custom_dataloders.py	40.06% <40.06%> (ø)

... and 25 files with indirect coverage changes

🚀 New features to boost your workflow:

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

[canergen]

does this work for AnnData? Can we remove the if statement here?