Merge pull request #15423 from storyicon/master

feat: ensure the indexability of dynamically imported packages
This commit is contained in:
AUTOMATIC1111 2024-04-06 09:00:35 +03:00 committed by GitHub
commit badb70da48
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 2 deletions

View File

@ -2,13 +2,18 @@ import os
import importlib.util
from modules import errors
import sys
def load_module(path):
module_spec = importlib.util.spec_from_file_location(os.path.basename(path), path)
module = importlib.util.module_from_spec(module_spec)
module_spec.loader.exec_module(module)
if os.path.isfile(path):
sp = os.path.splitext(path)
module_name = sp[0]
else:
module_name = os.path.basename(path)
sys.modules[module_name] = module
return module