91db788cc3fa60962a159e9002a20b8930b6e314,dask_xgboost/tests/test_core.py,,test_dask,#Any#Any#Any#Any#,16

Before Change

@gen_cluster(client=True, Worker=Nanny)
def test_dask(c, s, a, b):
    x = np.random.random((1000, 10))
    y = np.random.random(1000)

    dtrain = xgb.DMatrix(x, label=y)

After Change

    bst = xgb.train(param, dtrain)

    ddf = dd.from_pandas(df, npartitions=4)
    dlabels = dd.from_pandas(labels, npartitions=4)
    dbst = yield dxgb._train(c, param, ddf, dlabels)

    result = bst.predict(dtrain)
    dresult = dbst.predict(dtrain)

Instances