Skip to content

Commit 943b2b3

Browse files
committed
fix: clear module from sys.modules before importing service
Signed-off-by: Frost Ming <[email protected]>
1 parent 0a26bfa commit 943b2b3

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/bentoml/_internal/service/loader.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,8 @@ def import_service(
8484
if model_store is not global_model_store:
8585
BentoMLContainer.model_store.set(model_store)
8686

87+
svc: AnyService | Service | None = None
88+
8789
def _restore(restore_model_store: bool = False) -> None:
8890
"""Undo changes to sys.path, cwd and model store"""
8991
if restore_model_store and model_store is not global_model_store:
@@ -92,6 +94,8 @@ def _restore(restore_model_store: bool = False) -> None:
9294
sys.path.remove(working_dir)
9395
if prev_cwd is not None:
9496
os.chdir(prev_cwd)
97+
if svc is not None and (import_string := getattr(svc, "_import_str", None)):
98+
sys.modules.pop(import_string.split(":", 1)[0], None)
9599

96100
try:
97101
svc = _do_import(svc_import_path, working_dir, reload)

0 commit comments

Comments
 (0)