Update __init__.py
This commit is contained in:
+6
-2
@@ -3,6 +3,7 @@ from .batch_image_overlay import NODE_CLASS_MAPPINGS as OVERLAY_CLASS_MAPPINGS,
|
||||
from .remove_black_to_transparent import NODE_CLASS_MAPPINGS as REMOVE_BLACK_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as REMOVE_BLACK_DISPLAY_NAMES
|
||||
from .batch_enlarged_overlay import NODE_CLASS_MAPPINGS as ENLARGED_OVERLAY_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as ENLARGED_OVERLAY_DISPLAY_NAMES
|
||||
from .canny_shrink_recenter import NODE_CLASS_MAPPINGS as CANNY_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as CANNY_DISPLAY_NAMES
|
||||
from .calculate_center_bounding_box import NODE_CLASS_MAPPINGS as CENTER_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as CENTER_DISPLAY_NAMES
|
||||
|
||||
# Merge all node mappings into one
|
||||
NODE_CLASS_MAPPINGS = {
|
||||
@@ -10,7 +11,8 @@ NODE_CLASS_MAPPINGS = {
|
||||
**OVERLAY_CLASS_MAPPINGS,
|
||||
**REMOVE_BLACK_CLASS_MAPPINGS,
|
||||
**ENLARGED_OVERLAY_CLASS_MAPPINGS,
|
||||
**CANNY_CLASS_MAPPINGS
|
||||
**CANNY_CLASS_MAPPINGS,
|
||||
**CENTER_CLASS_MAPPINGS
|
||||
}
|
||||
|
||||
NODE_DISPLAY_NAME_MAPPINGS = {
|
||||
@@ -18,8 +20,10 @@ NODE_DISPLAY_NAME_MAPPINGS = {
|
||||
**OVERLAY_DISPLAY_NAMES,
|
||||
**REMOVE_BLACK_DISPLAY_NAMES,
|
||||
**ENLARGED_OVERLAY_DISPLAY_NAMES,
|
||||
**CANNY_DISPLAY_NAMES
|
||||
**CANNY_DISPLAY_NAMES,
|
||||
**CENTER_DISPLAY_NAMES
|
||||
}
|
||||
|
||||
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user