ComfyUI loads custom nodes as packages (relative import needed), pytest runs at top level (absolute import needed). Handle both. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
15 lines
338 B
Python
15 lines
338 B
Python
try:
|
|
from .immich_nodes.save_to_immich import SaveToImmich
|
|
except ImportError:
|
|
from immich_nodes.save_to_immich import SaveToImmich
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"SaveToImmich": SaveToImmich,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"SaveToImmich": "Save to Immich",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|