diff --git a/__init__.py b/__init__.py index f7f8218..c02b5af 100644 --- a/__init__.py +++ b/__init__.py @@ -1,4 +1,4 @@ -from .negative_wildcard_processor import NegativeWildcardProcessor -NODE_CLASS_MAPPINGS = { "neg_wildcard_processor" : NegativeWildcardProcessor } -NODE_DISPLAY_NAME_MAPPINGS = { "neg_wildcard_processor" : "Negative Wildcard Processor" } -__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] \ No newline at end of file +from .negative_wildcard_processor import NegativeWildcardProcessor, CustomTokenPreprocessor +NODE_CLASS_MAPPINGS = { "neg_wildcard_processor" : NegativeWildcardProcessor, "custom_token_processor" : CustomTokenPreprocessor } +NODE_DISPLAY_NAME_MAPPINGS = { "neg_wildcard_processor" : "Negative Wildcard Processor", "custom_token_processor" : "Custom Token Processor" } +__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']