moved to type checking blocks beucase of ruff updates

src/scvi/model/base/_archesmixin.py

@@ -3,6 +3,7 @@
 import logging
 import warnings
 from copy import deepcopy
+from typing import TYPE_CHECKING
 
 import anndata
 import numpy as np
@@ -13,7 +14,6 @@
 from scipy.sparse import csr_matrix
 
 from scvi import settings
-from scvi._types import AnnOrMuData
 from scvi.data import _constants
 from scvi.data._constants import _MODEL_NAME_KEY, _SETUP_ARGS_KEY, _SETUP_METHOD_NAME
 from scvi.model._utils import parse_device_args
@@ -25,7 +25,10 @@
 from scvi.nn import FCLayers
 from scvi.utils._docstrings import devices_dsp
 
-from ._base_model import BaseModelClass
+if TYPE_CHECKING:
+    from scvi._types import AnnOrMuData
+
+    from ._base_model import BaseModelClass
 
 logger = logging.getLogger(__name__)
 

src/scvi/model/base/_base_model.py

@@ -11,7 +11,6 @@
 from uuid import uuid4
 
 import numpy as np
-import pandas as pd
 import rich
 import torch
 from anndata import AnnData
@@ -54,6 +53,8 @@
 if TYPE_CHECKING:
     from collections.abc import Sequence
 
+    import pandas as pd
+
     from scvi._types import AnnOrMuData, MinifiedDataType
 
 logger = logging.getLogger(__name__)