File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed
src/bentoml/_internal/service Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -84,6 +84,8 @@ def import_service(
84
84
if model_store is not global_model_store :
85
85
BentoMLContainer .model_store .set (model_store )
86
86
87
+ svc : AnyService | Service | None = None
88
+
87
89
def _restore (restore_model_store : bool = False ) -> None :
88
90
"""Undo changes to sys.path, cwd and model store"""
89
91
if restore_model_store and model_store is not global_model_store :
@@ -92,6 +94,8 @@ def _restore(restore_model_store: bool = False) -> None:
92
94
sys .path .remove (working_dir )
93
95
if prev_cwd is not None :
94
96
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 )
95
99
96
100
try :
97
101
svc = _do_import (svc_import_path , working_dir , reload )
You can’t perform that action at this time.
0 commit comments