diff --git a/src/bentoml/__init__.py b/src/bentoml/__init__.py index 70b5ae50919..df318368c2c 100644 --- a/src/bentoml/__init__.py +++ b/src/bentoml/__init__.py @@ -51,6 +51,7 @@ from ._internal.yatai_client import YataiClient from ._internal.monitoring.api import monitor from ._internal.service.loader import load +from ._internal.runner.strategy import Strategy # Framework specific modules, model management and IO APIs are lazily loaded upon import. if TYPE_CHECKING: @@ -164,6 +165,7 @@ "Model", "Runner", "Runnable", + "Strategy", "YataiClient", # Yatai REST API Client # bento APIs "list",