chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,19 @@
|
||||
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.
|
||||
Reference in New Issue
Block a user