From 2c7a642babbea25db70fe35a4a3f5eac4d41a529 Mon Sep 17 00:00:00 2001 From: Euxhen Hasanaj Date: Sat, 9 Dec 2023 16:07:08 -0500 Subject: [PATCH] Switch to 2 comps --- src/grinch/processors/predictors.py | 2 +- tests/test_predictors.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/grinch/processors/predictors.py b/src/grinch/processors/predictors.py index 13ba31f..9dfc594 100644 --- a/src/grinch/processors/predictors.py +++ b/src/grinch/processors/predictors.py @@ -215,7 +215,7 @@ def _post_process(self, adata: AnnData) -> None: class BaseSupervisedPredictor(BasePredictor, abc.ABC): - """A base class for unsupervised predictors, e.g., clustering.""" + """A base class for supervised predictors, e.g., logistic regression.""" __processor_reqs__ = ['fit'] class Config(BasePredictor.Config): diff --git a/tests/test_predictors.py b/tests/test_predictors.py index 8311a7b..80e2745 100644 --- a/tests/test_predictors.py +++ b/tests/test_predictors.py @@ -127,7 +127,7 @@ def test_classifiers_x(X, classifier, key): cfg_pca = OmegaConf.create( { "_target_": "src.grinch.PCA.Config", - "n_components": 4, + "n_components": 2, "seed": 42, "write_key": f"obsm.{OBSM.X_PCA}", }