Update __init__.py

This commit is contained in:
Pheat-AI
2024-09-08 21:59:37 -07:00
committed by GitHub
parent 96b8e9f347
commit 21c95e92ab
+6 -2
View File
@@ -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']