diff --git a/canals/pipeline/pipeline.py b/canals/pipeline/pipeline.py index 60d9a89cb6..756bcc1963 100644 --- a/canals/pipeline/pipeline.py +++ b/canals/pipeline/pipeline.py @@ -89,8 +89,9 @@ def to_dict(self) -> Dict[str, Any]: components[name] = component_to_dict(instance) connections = [] - for sender, receiver, sockets in self.graph.edges: - (sender_socket, receiver_socket) = sockets.split("/") + for sender, receiver, edge_data in self.graph.edges.data(): + sender_socket = edge_data["from_socket"].name + receiver_socket = edge_data["to_socket"].name connections.append( { "sender": f"{sender}.{sender_socket}",