fix: loading mechanism for new mrvi

[ori-kron-wis]

No description provided.

[codecov]

Codecov Report

❌ Patch coverage is 63.15789% with 7 lines in your changes missing coverage. Please review.
✅ Project coverage is 74.82%. Comparing base (3e275ff) to head (7327094).

Files with missing lines	Patch %	Lines
src/scvi/external/mrvi/_model.py	58.82%	7 Missing ⚠️

❗ There is a different number of reports uploaded between BASE (3e275ff) and HEAD (7327094). Click for more details.

HEAD has 64 uploads less than BASE

Flag	BASE (3e275ff)	HEAD (7327094)
	67	3

Additional details and impacted files

@@             Coverage Diff             @@
##             main    #3615       +/-   ##
===========================================
- Coverage   86.95%   74.82%   -12.13%     
===========================================
  Files         225      225               
  Lines       21273    21274        +1     
===========================================
- Hits        18498    15919     -2579     
- Misses       2775     5355     +2580     

Files with missing lines	Coverage Δ
src/scvi/external/mrvi_torch/_model.py	87.48% <100.00%> (-0.78%)	⬇️
src/scvi/external/mrvi_torch/_module.py	96.98% <ø> (+0.37%)	⬆️
src/scvi/external/mrvi/_model.py	72.13% <58.82%> (-22.96%)	⬇️

... and 58 files with indirect coverage changes

🚀 New features to boost your workflow:

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