SCVI view_anndata_setup causes AttributeError with rich.pretty

Hello, I was trying to look at my scvi model using view_anndata_setup, but got the error below that says ‘rich has no attribute pretty’. I was actually able to do everything else, like train my model completely fine. Eventually, I was able to solve the error by calling importing rich.pretty. Looking at the source code of view_anndata_setup, it seems rich is imported, but rich.pretty is not, but I think maybe it should be at least in my case, given my versions (see below).

Error:
AttributeError: module 'rich' has no attribute 'pretty'

Version Info:

python 13.3.0, scvi-tools 1.4.1, rich 15.0.0

Installation steps:

mamba install conda-forge::scvi-tools
pip install --upgrade "jax[cuda12]"
pip install --upgrade torch --extra-index-url https://download.pytorch.org/whl/cu126
pip install --upgrade "scvi-tools[cuda]"
pip install --upgrade torchvision --extra-index-url https://download.pytorch.org/whl/cu126

The fix you made was already included in scvi-tools 1.4.3 (rich updated their API), so you can just upgrade your scvi-tools version.