From 5dc2f642f2d4a2a23b903437f716411e122a987c Mon Sep 17 00:00:00 2001 From: David Marx Date: Wed, 3 Jan 2024 00:20:24 -0800 Subject: [PATCH] Update __init__.py --- __init__.py | 25 ++++++++++++++----------- 1 file changed, 14 insertions(+), 11 deletions(-) diff --git a/__init__.py b/__init__.py index ab6f54d..eb87ccc 100644 --- a/__init__.py +++ b/__init__.py @@ -1,14 +1,3 @@ -#from nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS -from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS -#from .nodes.audio_nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS - -from loguru import logger - -logger.info(NODE_CLASS_MAPPINGS) -logger.info(NODE_DISPLAY_NAME_MAPPINGS) - -__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] - import os import subprocess import importlib.util @@ -41,3 +30,17 @@ is_installed("librosa") print(os.environ.get('COMFYUI_DEBUG_MODE')) +############## + +# NB: nodes import needs to be after ensuring dependencies installed + +#from nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS +from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS +#from .nodes.audio_nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS + +from loguru import logger + +logger.info(NODE_CLASS_MAPPINGS) +logger.info(NODE_DISPLAY_NAME_MAPPINGS) + +__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]