From 59b72d9156319853f344247fc828c4a42f13540f Mon Sep 17 00:00:00 2001 From: ma3oun Date: Thu, 5 Dec 2024 17:31:54 +0100 Subject: [PATCH] bug fix --- graph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/graph.py b/graph.py index 8005e7e..70aad0a 100644 --- a/graph.py +++ b/graph.py @@ -348,9 +348,9 @@ class Graph: to_node_tuple[0], to_node_tuple[1] + 1 ): if from_node_id not in self.nodes: - self.add_node(Node(from_node_id, model=self.model)) + self.add_node(Node(from_node_id, model=self.model,config=self.config)) if to_node_id not in self.nodes: - self.add_node(Node(to_node_id, model=self.model)) + self.add_node(Node(to_node_id, model=self.model,config=self.config)) self.add_edge(from_node_id, to_node_id) self.input_layer = self.get_input_layer() self.output_layer = self.get_output_layer()