From 74cd41b68f7c345d5729fc6d8402b40c87c9f643 Mon Sep 17 00:00:00 2001
From: Sauyon Lee <2347889+sauyon@users.noreply.github.com>
Date: Wed, 14 Sep 2022 16:17:35 -0700
Subject: [PATCH] Update bentoml/xgboost.py

Co-authored-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
---
 bentoml/xgboost.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

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'