New in 0.17.2 (2022-08-26)#
Changes#
Move
training
argument inJaxVAE
constructor to a keyword argument into the call method. This simplifies theJaxModuleWrapper
logic and avoids the reinstantiation ofJaxVAE
during evaluation #1580.Add a static method on the BaseModelClass to return the AnnDataManger’s full registry #1617.
Clarify docstrings for continuous and categorical covariate keys #1637.
Remove poetry lock, use newer build system #1645.
Bug Fixes#
Breaking Changes#
Fix an issue where
max_epochs
is never determined heuristically for totalvi, instead it would always default to 400 #1639.