diff --git a/__init__.py b/__init__.py index c6c1f69..0f9bb8e 100644 --- a/__init__.py +++ b/__init__.py @@ -1,9 +1,11 @@ -from .compass_image_loader import CompassImageLoader +from .compass_image_loader import CompassImageLoader, CompassImageSaver NODE_CLASS_MAPPINGS = { "CompassImageLoader": CompassImageLoader, + "CompassImageSaver": CompassImageSaver, } NODE_DISPLAY_NAME_MAPPINGS = { "CompassImageLoader": "Compass Image Loader", + "CompassImageSaver": "Save Compass Images", }