Update unit tests

ThomasMeissnerDS · Dec 5, 2024 · ae71917 · ae71917
1 parent e40d92d
commit ae71917
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 0 deletions.
diff --git a/bluecast/tests/test_cast_regression.py b/bluecast/tests/test_cast_regression.py
@@ -39,6 +39,7 @@ def test_blueprint_xgboost(synthetic_train_test_data, synthetic_calibration_data
     df_val = synthetic_train_test_data[1]
     df_calibration = synthetic_calibration_data
     xgboost_param_config = XgboostTuneParamsRegressionConfig()
+    xgboost_param_config.steps_min = 2
     xgboost_param_config.steps_max = 100
     xgboost_param_config.max_depth_max = 3
 

diff --git a/bluecast/tests/test_shap_explanations.py b/bluecast/tests/test_shap_explanations.py
@@ -33,6 +33,7 @@ def test_shap_explanations():
     df_train = create_synthetic_dataframe(200, random_state=20)
     df_val = create_synthetic_dataframe(100, random_state=21)
     xgboost_param_config = XgboostTuneParamsConfig()
+    xgboost_param_config.steps_min = 2
     xgboost_param_config.steps_max = 100
     train_config = TrainingConfig()
     train_config.hyperparameter_tuning_rounds = 10

diff --git a/dist/bluecast-1.6.4-py3-none-any.whl b/dist/bluecast-1.6.4-py3-none-any.whl
diff --git a/dist/bluecast-1.6.4.tar.gz b/dist/bluecast-1.6.4.tar.gz