Avoid default tokenization in Dask (#10398)
--------- Co-authored-by: Jiaming Yuan <jm.yuan@outlook.com>
This commit is contained in:
committed by
GitHub
parent
01ff2b2c29
commit
dc14f98f40
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user