Fix in non multiGPU training to have history saved in memory, and not on disk by default

[ori-kron-wis]

No description provided.

[codecov]

Codecov Report

❌ Patch coverage is 42.10526% with 11 lines in your changes missing coverage. Please review.
✅ Project coverage is 77.87%. Comparing base (c00bd8b) to head (51c31e7).
⚠️ Report is 1 commits behind head on main.

Files with missing lines	Patch %	Lines
src/scvi/train/_logger.py	25.00%	9 Missing ⚠️
src/scvi/external/mrvi_jax/_model.py	75.00%	1 Missing ⚠️
src/scvi/train/_trainer.py	66.66%	1 Missing ⚠️

❗ There is a different number of reports uploaded between BASE (c00bd8b) and HEAD (51c31e7). Click for more details.

HEAD has 55 uploads less than BASE

Flag	BASE (c00bd8b)	HEAD (51c31e7)
	59	4

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #3543      +/-   ##
==========================================
- Coverage   85.73%   77.87%   -7.87%     
==========================================
  Files         224      224              
  Lines       20986    20992       +6     
==========================================
- Hits        17992    16347    -1645     
- Misses       2994     4645    +1651     

Files with missing lines	Coverage Δ
src/scvi/external/mrvi_jax/_model.py	61.56% <75.00%> (-28.36%)	⬇️
src/scvi/train/_trainer.py	86.20% <66.66%> (-1.30%)	⬇️
src/scvi/train/_logger.py	83.11% <25.00%> (-8.67%)	⬇️

... and 43 files with indirect coverage changes

🚀 New features to boost your workflow:

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