Bugs for loading scvi

[HelloWorldLTY]

Hi, I notice that there is a bug if I intend to load scVI after the installation of scvi:

ImportError                               Traceback (most recent call last)
Cell In[2], line 1
----> 1 import scvi

File /gpfs/gibbs/project/zhao/tl688/conda_envs/scpert/lib/python3.8/site-packages/scvi/__init__.py:7
      4 import logging
      6 from ._constants import REGISTRY_KEYS
----> 7 from ._settings import settings
      9 # this import needs to come after prior imports to prevent circular import
     10 from . import data, model, external, utils

File /gpfs/gibbs/project/zhao/tl688/conda_envs/scpert/lib/python3.8/site-packages/scvi/_settings.py:6
      3 from pathlib import Path
      4 from typing import Union
----> 6 import pytorch_lightning as pl
      7 import torch
      8 from rich.console import Console

File /gpfs/gibbs/project/zhao/tl688/conda_envs/scpert/lib/python3.8/site-packages/pytorch_lightning/__init__.py:34
     31     _logger.addHandler(logging.StreamHandler())
     32     _logger.propagate = False
---> 34 from pytorch_lightning.callbacks import Callback  # noqa: E402
     35 from pytorch_lightning.core import LightningDataModule, LightningModule  # noqa: E402
     36 from pytorch_lightning.trainer import Trainer  # noqa: E402

File /gpfs/gibbs/project/zhao/tl688/conda_envs/scpert/lib/python3.8/site-packages/pytorch_lightning/callbacks/__init__.py:25
     23 from pytorch_lightning.callbacks.model_summary import ModelSummary
     24 from pytorch_lightning.callbacks.prediction_writer import BasePredictionWriter
---> 25 from pytorch_lightning.callbacks.progress import ProgressBarBase, RichProgressBar, TQDMProgressBar
     26 from pytorch_lightning.callbacks.pruning import ModelPruning
     27 from pytorch_lightning.callbacks.quantization import QuantizationAwareTraining

File /gpfs/gibbs/project/zhao/tl688/conda_envs/scpert/lib/python3.8/site-packages/pytorch_lightning/callbacks/progress/__init__.py:22
     14 """
     15 Progress Bars
     16 =============
   (...)
     19 
     20 """
     21 from pytorch_lightning.callbacks.progress.base import ProgressBarBase  # noqa: F401
---> 22 from pytorch_lightning.callbacks.progress.rich_progress import RichProgressBar  # noqa: F401
     23 from pytorch_lightning.callbacks.progress.tqdm_progress import TQDMProgressBar  # noqa: F401

File /gpfs/gibbs/project/zhao/tl688/conda_envs/scpert/lib/python3.8/site-packages/pytorch_lightning/callbacks/progress/rich_progress.py:20
     17 from datetime import timedelta
     18 from typing import Any, Dict, Optional, Union
---> 20 from torchmetrics.utilities.imports import _compare_version
     22 import pytorch_lightning as pl
     23 from pytorch_lightning.callbacks.progress.base import ProgressBarBase

ImportError: cannot import name '_compare_version' from 'torchmetrics.utilities.imports' (/gpfs/gibbs/project/zhao/tl688/conda_envs/scpert/lib/python3.8/site-packages/torchmetrics/utilities/imports.py)

The solution is:

pip install torchmetrics==0.11.4

Could you please check it in your local environment? Thanks.