Avoid default tokenization in Dask (#10398)

---------

Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
This commit is contained in:
Richard (Rick) Zamora
2024-06-14 06:44:54 -05:00
committed by GitHub
parent 01ff2b2c29
commit dc14f98f40
5 changed files with 16 additions and 9 deletions

View File

@@ -7,6 +7,7 @@ import json
import os
import re
import sys
import uuid
import warnings
import weakref
from abc import ABC, abstractmethod
@@ -3143,3 +3144,9 @@ class Booster:
UserWarning,
)
return nph_stacked
def __dask_tokenize__(self) -> uuid.UUID:
# TODO: Implement proper tokenization to avoid unnecessary re-computation in
# Dask. However, default tokenzation causes problems after
# https://github.com/dask/dask/pull/10883
return uuid.uuid4()