From c05c9b552bde2bef12137479072cb647e0ed0e40 Mon Sep 17 00:00:00 2001
From: Chenlei Hu <hcl@comfy.org>
Date: Tue, 25 Mar 2025 16:47:42 -0400
Subject: [PATCH] nit

---
 comfy_execution/graph.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/comfy_execution/graph.py b/comfy_execution/graph.py
index 35342557..94425a9a 100644
--- a/comfy_execution/graph.py
+++ b/comfy_execution/graph.py
@@ -65,6 +65,11 @@ def get_input_info(
     """Get the input type, category, and extra info for a given input name.
     If valid_inputs is provided, it will be used instead of the class_def.INPUT_TYPES().
 
+    Arguments:
+        class_def: The class definition of the node.
+        input_name: The name of the input to get info for.
+        valid_inputs: The valid inputs for the node, or None to use the class_def.INPUT_TYPES().
+
     Returns:
         tuple[str, str, dict] | tuple[None, None, None]: The input type, category, and extra info for the input name.
     """