From fb1efdf7794500036376e79df8fff9e4ec245942 Mon Sep 17 00:00:00 2001 From: Matthew Deng Date: Tue, 28 Nov 2023 10:40:29 -0800 Subject: [PATCH] [train] update XGBoost model format Signed-off-by: Matthew Deng --- python/ray/train/xgboost/xgboost_checkpoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/train/xgboost/xgboost_checkpoint.py b/python/ray/train/xgboost/xgboost_checkpoint.py index 447fb1e32f8e..2c6fbaeb0abf 100644 --- a/python/ray/train/xgboost/xgboost_checkpoint.py +++ b/python/ray/train/xgboost/xgboost_checkpoint.py @@ -15,7 +15,7 @@ class XGBoostCheckpoint(FrameworkCheckpoint): """A :py:class:`~ray.train.Checkpoint` with XGBoost-specific functionality.""" - MODEL_FILENAME = "model.json" + MODEL_FILENAME = "model.ubj" @classmethod def from_model(