Update bentoml/xgboost.py

Co-authored-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
sauyon · Sep 14, 2022 · 74cd41b · 74cd41b
1 parent 6574656
commit 74cd41b
Showing 1 changed file with 1 addition and 4 deletions.
diff --git 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'