diff --git a/__init__.py b/__init__.py index 273cd43..8c5bcf5 100644 --- a/__init__.py +++ b/__init__.py @@ -1,5 +1,9 @@ -from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS -from .spectrum_h3_compat import install_spectrum_h3_compat +if __package__: + from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS + from .spectrum_h3_compat import install_spectrum_h3_compat +else: # pragma: no cover - direct pytest/import fallback + from nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS + from spectrum_h3_compat import install_spectrum_h3_compat install_spectrum_h3_compat()