diff --git a/__init__.py b/__init__.py index fd50801f..e30041e8 100644 --- a/__init__.py +++ b/__init__.py @@ -32,8 +32,8 @@ def serialize(obj): if not file.endswith(".py"): continue name = os.path.splitext(file)[0] - imported_module = importlib.import_module(".py.{}".format(name), __name__) try: + imported_module = importlib.import_module(".py.{}".format(name), __name__) NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **imported_module.NODE_CLASS_MAPPINGS} NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **imported_module.NODE_DISPLAY_NAME_MAPPINGS} serialized_CLASS_MAPPINGS = {k: serialize(v) for k, v in imported_module.NODE_CLASS_MAPPINGS.items()}