fix(scheduling): expose Strategy class

bentoml · Jun 26, 2023 · c9ba5fa · c9ba5fa
1 parent 91ed3da
commit c9ba5fa
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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",