leap-model-parser 0.1.185.dev9__tar.gz → 0.1.185.dev10__tar.gz

{leap_model_parser-0.1.185.dev9 → leap_model_parser-0.1.185.dev10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.185.dev9
+Version: 0.1.185.dev10
 Summary: 
 Home-page: https://github.com/tensorleap/leap-model-parser
 License: MIT

{leap_model_parser-0.1.185.dev9 → leap_model_parser-0.1.185.dev10}/leap_model_parser/leap_graph_editor.py

@@ -319,6 +319,18 @@ class LeapGraphEditor:
             origin_name = self.keras_model.inputs[input_index].node.layer.name
             input_node_id = self._find_input_node_by_origin_name(origin_name).id
             self.model_graph[input_node_id].data['output_name'] = input_node.name
+            output_keys = list(self.model_graph[input_node_id].outputs.keys())
+            for output_key in output_keys:
+                new_output_key = f'{input_node_id}-{input_node.name}'
+                if output_key == new_output_key:
+                    continue
+                self.model_graph[input_node_id].outputs[new_output_key] = self.model_graph[input_node_id].outputs[output_key]
+                del self.model_graph[input_node_id].outputs[output_key]
+                for connection in self.model_graph[input_node_id].outputs[new_output_key].connections:
+                    for connection_input in self.model_graph[connection.node].inputs[connection.input].connections:
+                        if connection.output == output_key:
+                            connection.output = new_output_key
+
             if input_node_id is None:
                 raise Exception(f"Couldn't find input node by origin name {origin_name}")
         elif input_node.type.value.startswith('Prediction'):

{leap_model_parser-0.1.185.dev9 → leap_model_parser-0.1.185.dev10}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "leap-model-parser"
-version = "0.1.185.dev9"
+version = "0.1.185.dev10"
 description = ""
 authors = ["idan <idan.yogev@tensorleap.ai>"]
 license = "MIT"