remove xgboost

euxhenh · Dec 9, 2023 · 26e1a9e · 26e1a9e
1 parent 2c7a642
commit 26e1a9e
Showing 1 changed file with 1 addition and 10 deletions.
diff --git a/tests/test_predictors.py b/tests/test_predictors.py
@@ -119,8 +119,7 @@ def test_gmix_x(X):
 
 @pytest.mark.parametrize("X", X_mods_no_sparse)
 @pytest.mark.parametrize(
-    "classifier, key", [("LogisticRegression", OBS.LOG_REG),
-                        ("XGBClassifier", OBS.XGB_CLASSIFIER)]
+    "classifier, key", [("LogisticRegression", OBS.LOG_REG)]
 )
 def test_classifiers_x(X, classifier, key):
     adata = AnnData(X)
@@ -148,14 +147,6 @@ def test_classifiers_x(X, classifier, key):
     kmeans = cfg_kmeans.create()
     kmeans(adata)
 
-    if classifier == "XGBClassifier":
-        kwargs = {
-            'n_estimators': 1,
-            'max_depth': 1,
-        }
-    else:
-        kwargs = {}
-
     cfg = OmegaConf.create(
         {
             "_target_": f"src.grinch.{classifier}.Config",