fixed bug in xgboost test

madgik · Nov 4, 2024 · 075b351 · 075b351
1 parent e6dd1cf
commit 075b351
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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