Skip to content

Commit 1b0708e

Browse files
committed
cleaning up handler
1 parent c7ec28f commit 1b0708e

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

vetiver/handlers/xgboost.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,12 +11,12 @@
1111

1212

1313
class XGBoostHandler(BaseHandler):
14-
"""Handler class for creating VetiverModels with statsmodels.
14+
"""Handler class for creating VetiverModels with xgboost.
1515
1616
Parameters
1717
----------
18-
model : statsmodels
19-
a trained and fit statsmodels model
18+
model :
19+
a trained and fit xgboost model
2020
"""
2121

2222
model_class = staticmethod(lambda: xgboost.Booster)
@@ -35,7 +35,7 @@ def create_meta(
3535
url: str = None,
3636
required_pkgs: list = [],
3737
):
38-
"""Create metadata for statsmodel"""
38+
"""Create metadata for xgboost"""
3939
required_pkgs = required_pkgs + ["xgboost"]
4040
meta = _model_meta(user, version, url, required_pkgs)
4141

0 commit comments

Comments
 (0)