From 20047bd9eabbaf95d2e6830c3a8c13eb59aab0b0 Mon Sep 17 00:00:00 2001
From: bojiang <bojiang_@outlook.com>
Date: Wed, 6 Nov 2024 18:52:08 +0800
Subject: [PATCH] chore: add return names

---
 nodes/nodes.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/nodes/nodes.py b/nodes/nodes.py
index b7b1b4d..89896af 100644
--- a/nodes/nodes.py
+++ b/nodes/nodes.py
@@ -136,6 +136,7 @@ class LoadImage:
     CATEGORY = "bentoml/io"
     BENTOML_NODE = True
     RETURN_TYPES = ("IMAGE", "MASK")
+    RETUEN_NAMES = ("image", "mask")
     FUNCTION = "load_image"
 
     def load_image(self, image):
@@ -208,6 +209,7 @@ class StringInput:
         }
 
     RETURN_TYPES = ("STRING",)
+    RETUEN_NAMES = ("string",)
     BENTOML_NODE = True
     FUNCTION = "string_input"
     CATEGORY = "bentoml/io"
@@ -231,6 +233,7 @@ class IntegerInput:
         }
 
     RETURN_TYPES = ("INT",)
+    RETUEN_NAMES = ("int",)
     FUNCTION = "identity"
     BENTOML_NODE = True
     CATEGORY = "bentoml/io"
@@ -254,6 +257,7 @@ class FloatInput:
         }
 
     RETURN_TYPES = ("FLOAT",)
+    RETUEN_NAMES = ("float",)
     FUNCTION = "identity"
     BENTOML_NODE = True
     CATEGORY = "bentoml/io"
@@ -277,6 +281,7 @@ class BooleanInput:
         }
 
     RETURN_TYPES = ("BOOLEAN",)
+    RETUEN_NAMES = ("bool",)
     FUNCTION = "identity"
     CATEGORY = "bentoml/io"
     BENTOML_NODE = True
@@ -300,6 +305,7 @@ class PathInput:
         }
 
     RETURN_TYPES = (anytype,)
+    RETUEN_NAMES = ("path",)
     FUNCTION = "identity"
     BENTOML_NODE = True
     CATEGORY = "bentoml/io"