From bbf5b9ee57620a28b44e45389faaf32ea475f94f Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Sat, 23 Sep 2023 01:28:18 +0800 Subject: [PATCH] [dask] Move dask module into directory. (#9597) --- .../xgboost/{dask.py => dask/__init__.py} | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) rename python-package/xgboost/{dask.py => dask/__init__.py} (99%) diff --git a/python-package/xgboost/dask.py b/python-package/xgboost/dask/__init__.py similarity index 99% rename from python-package/xgboost/dask.py rename to python-package/xgboost/dask/__init__.py index f62a3e5af428..a58c0f22550e 100644 --- a/python-package/xgboost/dask.py +++ b/python-package/xgboost/dask/__init__.py @@ -60,11 +60,11 @@ import numpy -from . import collective, config -from ._typing import _T, FeatureNames, FeatureTypes, ModelIn -from .callback import TrainingCallback -from .compat import DataFrame, LazyLoader, concat, lazy_isinstance -from .core import ( +from xgboost import collective, config +from xgboost._typing import _T, FeatureNames, FeatureTypes, ModelIn +from xgboost.callback import TrainingCallback +from xgboost.compat import DataFrame, LazyLoader, concat, lazy_isinstance +from xgboost.core import ( Booster, DataIter, DMatrix, @@ -75,8 +75,8 @@ _deprecate_positional_args, _expect, ) -from .data import _is_cudf_ser, _is_cupy_array -from .sklearn import ( +from xgboost.data import _is_cudf_ser, _is_cupy_array +from xgboost.sklearn import ( XGBClassifier, XGBClassifierBase, XGBClassifierMixIn, @@ -91,8 +91,8 @@ _wrap_evaluation_matrices, xgboost_model_doc, ) -from .tracker import RabitTracker, get_host_ip -from .training import train as worker_train +from xgboost.tracker import RabitTracker, get_host_ip +from xgboost.training import train as worker_train if TYPE_CHECKING: import dask