Skip to content

Commit

Permalink
moved to type checking blocks beucase of ruff updates
Browse files Browse the repository at this point in the history
  • Loading branch information
ori-kron-wis committed Sep 17, 2024
1 parent 4a648ff commit e3831cb
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 3 deletions.
7 changes: 5 additions & 2 deletions src/scvi/model/base/_archesmixin.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
import logging
import warnings
from copy import deepcopy
from typing import TYPE_CHECKING

import anndata
import numpy as np
Expand All @@ -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
Expand All @@ -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__)

Expand Down
3 changes: 2 additions & 1 deletion src/scvi/model/base/_base_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@
from uuid import uuid4

import numpy as np
import pandas as pd
import rich
import torch
from anndata import AnnData
Expand Down Expand Up @@ -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__)
Expand Down

0 comments on commit e3831cb

Please sign in to comment.