From 7eddbca7d92f133f12e4f198c8f9192fbc1caebf Mon Sep 17 00:00:00 2001
From: Eric Mey <ericmey@hana.mey.house>
Date: Wed, 1 Apr 2026 23:20:11 -0400
Subject: [PATCH] fix: use try/except import for ComfyUI and pytest
 compatibility

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>
---
 __init__.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/__init__.py b/__init__.py
index 187b3f9..b8bbe35 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,4 +1,7 @@
-from immich_nodes.save_to_immich import SaveToImmich
+try:
+    from .immich_nodes.save_to_immich import SaveToImmich
+except ImportError:
+    from immich_nodes.save_to_immich import SaveToImmich
 
 NODE_CLASS_MAPPINGS = {
     "SaveToImmich": SaveToImmich,