Skip to content

Commit

Permalink
It is ok to drop it after scarches categorial covariates fix
Browse files Browse the repository at this point in the history
  • Loading branch information
ori-kron-wis committed Sep 16, 2024
1 parent b6eb2f1 commit 2979ea2
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions src/scvi/model/base/_archesmixin.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
from mudata import MuData
from scipy.sparse import csr_matrix

from scvi import REGISTRY_KEYS, settings
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
Expand Down Expand Up @@ -140,11 +140,6 @@ def load_query_data(

model = _initialize_model(cls, adata, registry, attr_dict)

if model.summary_stats[f"n_{REGISTRY_KEYS.CAT_COVS_KEY}"] > 0:
raise NotImplementedError(
"scArches currently does not support models with extra categorical covariates."
)

version_split = model.registry[_constants._SCVI_VERSION_KEY].split(".")

if int(version_split[1]) < 8 and int(version_split[0]) == 0:
Expand Down

0 comments on commit 2979ea2

Please sign in to comment.