Skip to content

Commit

Permalink
Remove None value before returning connection names (#336)
Browse files Browse the repository at this point in the history
Remove None value when get connection names.

Co-authored-by: Lina Tang <linatang@microsoft.com>
  • Loading branch information
lumoslnt and Lina Tang committed Sep 8, 2023
1 parent 1a2bd29 commit cc485fd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/promptflow/promptflow/contracts/flow.py
Original file line number Diff line number Diff line change
Expand Up @@ -515,7 +515,7 @@ def get_connection_names(self):
tool = self.get_tool(node.tool) or self._tool_loader.load_tool_for_node(node)
if tool:
connection_names.update(self._get_connection_name_from_tool(tool, node).values())
return connection_names
return set({item for item in connection_names if item})

def get_connection_input_names_for_node(self, node_name):
"""Return connection input names."""
Expand Down

0 comments on commit cc485fd

Please sign in to comment.