diff --git a/src/bentoml/__init__.py b/src/bentoml/__init__.py index d8fe656eb43..51e608de28c 100644 --- a/src/bentoml/__init__.py +++ b/src/bentoml/__init__.py @@ -30,8 +30,10 @@ from ._internal.context import ServiceContext as Context from ._internal.models import Model from ._internal.monitoring import monitor +from ._internal.resource import Resource from ._internal.runner import Runnable from ._internal.runner import Runner +from ._internal.runner.strategy import Strategy from ._internal.service import Service from ._internal.service.loader import load from ._internal.tag import Tag @@ -219,4 +221,6 @@ "load_config", "save_config", "set_serialization_strategy", + "Strategy", + "Resource", ]