From a2d97f46f325410b9b773431de39feffd8f42190 Mon Sep 17 00:00:00 2001 From: ThomasMeissnerDS Date: Sat, 14 Oct 2023 17:52:16 +0200 Subject: [PATCH] Fix some unit tests and bugs --- bluecast/ml_modelling/xgboost.py | 2 +- bluecast/tests/test_cast_cv.py | 1 - bluecast/tests/test_shap_explanations.py | 1 - 3 files changed, 1 insertion(+), 3 deletions(-) diff --git a/bluecast/ml_modelling/xgboost.py b/bluecast/ml_modelling/xgboost.py index a3cde1ef..613d0013 100644 --- a/bluecast/ml_modelling/xgboost.py +++ b/bluecast/ml_modelling/xgboost.py @@ -168,7 +168,7 @@ def autotune( if ( not self.conf_params_xgboost or not self.conf_training - or self.experiment_tracker + or not self.experiment_tracker ): raise ValueError( "conf_params_xgboost, conf_training or experiment_tracker is None" diff --git a/bluecast/tests/test_cast_cv.py b/bluecast/tests/test_cast_cv.py index b40f84d4..31755d93 100644 --- a/bluecast/tests/test_cast_cv.py +++ b/bluecast/tests/test_cast_cv.py @@ -23,7 +23,6 @@ def test_blueprint_cv_xgboost(synthetic_train_test_data): df_val = synthetic_train_test_data[1] xgboost_param_config = XgboostTuneParamsConfig() xgboost_param_config.steps_max = 100 - xgboost_param_config.num_leaves_max = 16 train_config = TrainingConfig() train_config.hyperparameter_tuning_rounds = 10 diff --git a/bluecast/tests/test_shap_explanations.py b/bluecast/tests/test_shap_explanations.py index 38307c82..07ddecf5 100644 --- a/bluecast/tests/test_shap_explanations.py +++ b/bluecast/tests/test_shap_explanations.py @@ -27,7 +27,6 @@ def test_shap_explanations(): df_val = create_synthetic_dataframe(100, random_state=21) xgboost_param_config = XgboostTuneParamsConfig() xgboost_param_config.steps_max = 100 - xgboost_param_config.num_leaves_max = 16 train_config = TrainingConfig() train_config.hyperparameter_tuning_rounds = 10 train_config.enable_feature_selection = False