From 285f7a8183c8c80d217e727fa47e31b3aa61389c Mon Sep 17 00:00:00 2001 From: BVH <82035780+bvhari@users.noreply.github.com> Date: Wed, 6 Sep 2023 18:59:36 +0530 Subject: [PATCH] Revert formatting changes --- __init__.py | 47 +++++++++++++++++++++++------------------------ 1 file changed, 23 insertions(+), 24 deletions(-) diff --git a/__init__.py b/__init__.py index 18dd91e..c6273d1 100644 --- a/__init__.py +++ b/__init__.py @@ -1,24 +1,23 @@ -import importlib -import os - - -node_list = [ # Add list of .py files containing nodes here - "advanced_model_merging", - "reference_only", - "sampler_rescalecfg", - "sampler_tonemap", - "sampler_tonemap_rescalecfg", - "sdxl_model_merging" -] - -NODE_CLASS_MAPPINGS = {} -NODE_DISPLAY_NAME_MAPPINGS = {} - -for module_name in node_list: - imported_module = importlib.import_module(".{}".format(module_name), __name__) - - NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **imported_module.NODE_CLASS_MAPPINGS} - if hasattr(imported_module, "NODE_DISPLAY_NAME_MAPPINGS"): - NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **imported_module.NODE_DISPLAY_NAME_MAPPINGS} - -__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] +import importlib +import os + +node_list = [ #Add list of .py files containing nodes here + "advanced_model_merging", + "reference_only", + "sampler_rescalecfg", + "sampler_tonemap", + "sampler_tonemap_rescalecfg", + "sdxl_model_merging" +] + +NODE_CLASS_MAPPINGS = {} +NODE_DISPLAY_NAME_MAPPINGS = {} + +for module_name in node_list: + imported_module = importlib.import_module(".{}".format(module_name), __name__) + + NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **imported_module.NODE_CLASS_MAPPINGS} + if hasattr(imported_module, "NODE_DISPLAY_NAME_MAPPINGS"): + NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **imported_module.NODE_DISPLAY_NAME_MAPPINGS} + +__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']