fix: loading service from an absolute path (#4415)

bentoml · Jan 18, 2024 · fc95886 · fc95886
1 parent 84edd85
commit fc95886
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/_bentoml_sdk/service/factory.py b/src/_bentoml_sdk/service/factory.py
@@ -4,6 +4,7 @@
 import logging
 import math
 import os
+import pathlib
 import sys
 import typing as t
 from functools import lru_cache
@@ -168,6 +169,18 @@ def import_string(self) -> str:
             if import_module == "__main__":
                 if hasattr(sys.modules["__main__"], "__file__"):
                     import_module = sys.modules["__main__"].__file__
+                    assert isinstance(import_module, str)
+                    try:
+                        import_module_path = pathlib.Path(import_module).relative_to(
+                            pathlib.Path(self.working_dir)
+                        )
+                    except ValueError:
+                        raise BentoMLException(
+                            "Failed to get service import origin, service object defined in __main__ module is not supported"
+                        )
+                    import_module = str(import_module_path.with_suffix("")).replace(
+                        os.path.sep, "."
+                    )
                 else:
                     raise BentoMLException(
                         "Failed to get service import origin, service object defined interactively in console or notebook is not supported"