From bd59d89c7faf7148ef66a6aaddf615b405fc539c Mon Sep 17 00:00:00 2001 From: thinkwee Date: Tue, 26 Nov 2024 10:09:55 +0800 Subject: [PATCH] fix multi-model support --- graph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/graph.py b/graph.py index c34ec9ef..c217e12e 100644 --- a/graph.py +++ b/graph.py @@ -273,9 +273,9 @@ class Graph: for to_node_tuple in to_node_list: for to_node_id in range(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.config.get("Model")) + self.add_node(Node(from_node_id, model=self.config.get("Model"))) if to_node_id not in self.nodes: - self.add_node(Node(to_node_id), model=self.config.get("Model")) + self.add_node(Node(to_node_id, model=self.config.get("Model"))) self.add_edge(from_node_id, to_node_id) self.input_layer = self.get_input_layer() self.output_layer = self.get_output_layer()