Merge pull request #15423 from storyicon/master

feat: ensure the indexability of dynamically imported packages
AUTOMATIC1111 · Apr 6, 2024 · badb70d · badb70d
2 parents 447198f + e73a7e4
commit badb70d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/modules/script_loading.py b/modules/script_loading.py
@@ -2,13 +2,18 @@
 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