With weights.

dmlc · Feb 25, 2021 · 0579029 · 0579029
1 parent c973fe4
commit 0579029
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/test_with_dask.py b/tests/python/test_with_dask.py
@@ -652,7 +652,7 @@ async def train():
                                 asynchronous=True,
                                 dashboard_address=0) as cluster:
             async with Client(cluster, asynchronous=True) as client:
-                X, y, w = generate_array()
+                X, y, w = generate_array(with_weights=True)
                 dtrain = await DaskDMatrix(client, X, y, weight=w)
                 dvalid = await DaskDMatrix(client, X, y, weight=w)
                 output = await xgb.dask.train(client, {}, dtrain=dtrain)