wehub-resource-sync
20 lines
635 B
Python
20 lines
635 B
Python
from ray.train.v2._internal.constants import is_v2_enabled
|
|
from ray.train.xgboost._xgboost_utils import RayTrainReportCallback
|
|
from ray.train.xgboost.config import XGBoostConfig
|
|
from ray.train.xgboost.xgboost_checkpoint import XGBoostCheckpoint
|
|
from ray.train.xgboost.xgboost_trainer import XGBoostTrainer
|
|
|
|
if is_v2_enabled():
|
|
from ray.train.v2.xgboost.config import XGBoostConfig # noqa: F811
|
|
from ray.train.v2.xgboost.xgboost_trainer import XGBoostTrainer # noqa: F811
|
|
|
|
__all__ = [
|
|
"RayTrainReportCallback",
|
|
"XGBoostCheckpoint",
|
|
"XGBoostConfig",
|
|
"XGBoostTrainer",
|
|
]
|
|
|
|
|
|
# DO NOT ADD ANYTHING AFTER THIS LINE.
|