diff --git a/bentoml/__init__.py b/bentoml/__init__.py index ee9727b0bc..fcdf83338b 100644 --- a/bentoml/__init__.py +++ b/bentoml/__init__.py @@ -112,8 +112,8 @@ ) xgboost = _LazyLoader("bentoml.xgboost", globals(), "bentoml.xgboost") - io = _LazyLoader("io", globals(), "bentoml.io") - models = _LazyLoader("models", globals(), "bentoml.models") + io = _LazyLoader("bentoml.io", globals(), "bentoml.io") + models = _LazyLoader("bentoml.models", globals(), "bentoml.models") del _LazyLoader