From 075b35101393f884d4df6ebc30129c8504345650 Mon Sep 17 00:00:00 2001 From: Apostolos Glenis Date: Mon, 4 Nov 2024 14:33:31 +0200 Subject: [PATCH] fixed bug in xgboost test --- tests/algorithm_validation_tests/flower/test_xgboost.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/algorithm_validation_tests/flower/test_xgboost.py b/tests/algorithm_validation_tests/flower/test_xgboost.py index f359abd74..82aeca582 100644 --- a/tests/algorithm_validation_tests/flower/test_xgboost.py +++ b/tests/algorithm_validation_tests/flower/test_xgboost.py @@ -26,5 +26,5 @@ def test_xgboost(get_algorithm_result): algorithm_result = get_algorithm_result("xgboost", input) # {'metrics_aggregated': {'AUC': 0.7575790087463558}} print(algorithm_result) - auc_aggregated = algorithm_result["metrics_aggregated"]["AUC"] + auc_aggregated = algorithm_result["metrics_aggregated"][1]["AUC"] assert auc_aggregated > 0.0