Fix cls.__module__

This commit is contained in:
huchenlei 2024-04-15 22:14:19 -04:00
parent ff6f4680c4
commit 0f82948e4f
2 changed files with 6 additions and 8 deletions

View File

@ -1,7 +1,7 @@
from modules import scripts
from modules.shared import opts
xyz_grid = [x for x in scripts.scripts_data if x.script_class.__module__ == "xyz_grid.py"][0].module
xyz_grid = [x for x in scripts.scripts_data if x.script_class.__module__ == "scripts.xyz_grid"][0].module
def int_applier(value_name:str, min_range:int = -1, max_range:int = -1):
"""

View File

@ -9,15 +9,13 @@ loaded_scripts = {}
def load_module(path):
module_spec = importlib.util.spec_from_file_location(os.path.basename(path), path)
module_name, _ = os.path.splitext(os.path.basename(path))
full_module_name = "scripts." + module_name
module_spec = importlib.util.spec_from_file_location(full_module_name, path)
module = importlib.util.module_from_spec(module_spec)
module_spec.loader.exec_module(module)
loaded_scripts[path] = module
module_name, _ = os.path.splitext(os.path.basename(path))
sys.modules["scripts." + module_name] = module
loaded_scripts[full_module_name] = module
sys.modules[full_module_name] = module
return module