diff --git a/bentoml/xgboost.py b/bentoml/xgboost.py index 10f572d6164..53e3ed369e9 100644 --- a/bentoml/xgboost.py +++ b/bentoml/xgboost.py @@ -4,14 +4,11 @@ from ._internal.frameworks.xgboost import load_model from ._internal.frameworks.xgboost import save_model from ._internal.frameworks.xgboost import get_runnable -from ._internal.frameworks.xgboost import XGBoostOptions +from ._internal.frameworks.xgboost import XGBoostOptions as ModelOptions logger = logging.getLogger(__name__) -ModelOptions = XGBoostOptions - - def save(tag, *args, **kwargs): logger.warning( f'The "{__name__}.save" method is being deprecated. Use "{__name__}.save_model" instead'