Switch to 2 comps

euxhenh · Dec 9, 2023 · 2c7a642 · 2c7a642
1 parent 317dc63
commit 2c7a642
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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}",
         }