From 289b8ec1709ba327269dad8066c2c0aa37db514b Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Mon, 1 Apr 2024 18:07:54 +0800 Subject: fix(internal): fix get core in infini runtime --- src/infini/internal.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/infini/internal.py b/src/infini/internal.py index 67b77b2a..d97503f2 100644 --- a/src/infini/internal.py +++ b/src/infini/internal.py @@ -30,8 +30,8 @@ def require(name: str, paths: Optional[List] = None) -> Register: def acquire_core() -> Core: caller_frame = inspect.stack()[1][0] - caller_file: str = caller_frame.f_globals["__file__"] - top_name, *_ = caller_file.split(".") + caller_name: str = caller_frame.f_globals["__name__"] + top_name, *_ = caller_name.split(".") try: core = getattr(sys.modules[top_name], "__infini__")["core"] if not isinstance(core, Core): -- cgit v1.2.3-70-g09d2