Dask зависает при использовании метода поезда dask_xgboost

Question

Я пытаюсь воспроизвести пример dask xgboost из документации dask-ml на http://ml.dask.org/examples/xgboost.html. К сожалению, похоже, что Dask не заканчивает обучение, и мне трудно отследить значениеошибки и предупреждения.Вот мой код:

def main():
    cluster = LocalCluster()
    dask_client = Client(cluster)

    x, y = make_classification(n_samples=1000, n_features=10, chunks=10, n_informative=4, random_state=0)
    X_train, X_test, y_train, y_test = train_test_split(x, y, test_size=0.15)
    print("classification training...")
    params = {'objective': 'binary:logistic',
              'max_depth': 4,
              'eta': 0.01,
              'subsample': 0.5,
              'min_child_weight': 0.5}
    bst = dask_xgboost.train(dask_client, params, X_train, y_train, num_boost_round=10)

    dask_client.close()


if __name__ == '__main__':
    main()

Вот вывод:

classification training...
Socket Poll Error:WSAError-code=10022
Socket Poll Error:WSAError-code=10022
Socket Poll Error:WSAError-code=10022
Socket Poll Error:WSAError-code=10022
distributed.nanny - WARNING - Worker process 12376 exited with status 4294967295
distributed.nanny - WARNING - Worker process 3540 exited with status 4294967295
distributed.nanny - WARNING - Worker process 9424 exited with status 4294967295
distributed.nanny - WARNING - Worker process 12952 exited with status 4294967295
distributed.nanny - WARNING - Restarting worker
distributed.nanny - WARNING - Restarting worker
distributed.nanny - WARNING - Restarting worker
distributed.nanny - WARNING - Restarting worker

Есть мысли?