[CI] Fix custom metric test with empty dataset. (#6917)

2021-04-30 09:00:05 +08:00 · 2021-04-30 09:00:05 +08:00 · b31d37eac5
commit b31d37eac5
parent db6285fb55
1 changed files with 2 additions and 0 deletions
--- a/tests/python/testing.py
+++ b/tests/python/testing.py
@ -272,6 +272,8 @@ def eval_error_metric(predt, dtrain: xgb.DMatrix):
    label = dtrain.get_label()
    r = np.zeros(predt.shape)
    gt = predt > 0.5
+    if predt.size == 0:
+        return "CustomErr", 0
    r[gt] = 1 - label[gt]
    le = predt <= 0.5
    r[le] = label[le]