From f4a30a552ae47ba11f3bb2d7350cf8703a352f30 Mon Sep 17 00:00:00 2001 From: Zoltan Fedor Date: Mon, 6 Feb 2023 04:52:32 -0500 Subject: [PATCH] fix: use correct count of outgoing edges in RayPipeline (#4066) --- haystack/pipelines/ray.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/haystack/pipelines/ray.py b/haystack/pipelines/ray.py index 76585bf2f..02ff0c169 100644 --- a/haystack/pipelines/ray.py +++ b/haystack/pipelines/ray.py @@ -251,7 +251,7 @@ class RayPipeline(Pipeline): if "." in i: [input_node_name, input_edge_name] = i.split(".") assert "output_" in input_edge_name, f"'{input_edge_name}' is not a valid edge name." - outgoing_edges_input_node = self.graph.nodes[input_node_name]["component"].outgoing_edges + outgoing_edges_input_node = self.graph.nodes[input_node_name]["outgoing_edges"] assert int(input_edge_name.split("_")[1]) <= outgoing_edges_input_node, ( f"Cannot connect '{input_edge_name}' from '{input_node_name}' as it only has " f"{outgoing_edges_input_node} outgoing edge(s)."