From 26e1a9e7f73e7c4c8ab94014346a23e062f69588 Mon Sep 17 00:00:00 2001 From: Euxhen Hasanaj Date: Sat, 9 Dec 2023 16:19:50 -0500 Subject: [PATCH] remove xgboost --- tests/test_predictors.py | 11 +---------- 1 file changed, 1 insertion(+), 10 deletions(-) diff --git a/tests/test_predictors.py b/tests/test_predictors.py index 80e2745..11d57dd 100644 --- 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",