feat: enhance __init__.py with version tracking, web directory config and improved documentation

- Add module docstring
- Add version tracking (__version__)
- Add web directory config for UI components
- Add explicit __all__ exports list
- Improve code organization
This commit is contained in:
limbicnation
2024-11-29 18:43:09 +01:00
parent 248ae4e64a
commit 87a4afd108
+21 -2
View File
@@ -1,11 +1,30 @@
"""
ComfyUI Depth Estimation Node
A custom node for depth map estimation using transformer models.
"""
from .depth_estimation_node import DepthEstimationNode
# Version info
__version__ = "1.0.0"
# Node class mapping for ComfyUI registration
NODE_CLASS_MAPPINGS = {
"DepthEstimationNode": DepthEstimationNode
}
# Display names for UI
NODE_DISPLAY_NAME_MAPPINGS = {
"DepthEstimationNode": "Depth Estimation Node"
"DepthEstimationNode": "Depth Estimation"
}
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
# Web extension info for ComfyUI
WEB_DIRECTORY = "./js"
# Module exports
__all__ = [
"NODE_CLASS_MAPPINGS",
"NODE_DISPLAY_NAME_MAPPINGS",
"__version__",
"WEB_DIRECTORY"
]