leap-model-parser 0.1.185.dev9__py3-none-any.whl → 0.1.185.dev11__py3-none-any.whl

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_input.output == output_key:
+                            connection_input.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.dist-info → leap_model_parser-0.1.185.dev11.dist-info}/METADATA

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

{leap_model_parser-0.1.185.dev9.dist-info → leap_model_parser-0.1.185.dev11.dist-info}/RECORD

@@ -6,7 +6,7 @@ leap_model_parser/contract/importmodelresponse.py,sha256=GlvnKS8xrebU2Sj0dxqtEhA
 leap_model_parser/contract/nodedata.py,sha256=1_ML0nzp3QUZ0_9mGSLhfO4_hqjYMwi0DWLwymUnWEs,43326
 leap_model_parser/contract/ui_components.json,sha256=0lsxwOLElW1E-imCcdh3zKPWgzFuQ_bApG6aHvYfTvo,410591
 leap_model_parser/keras_json_model_import.py,sha256=x7HOH6iaASfzJgwMRHgF5SQS-iFOF5j9yCG0mDC9HEA,16794
-leap_model_parser/leap_graph_editor.py,sha256=b2rIpp7KcZHAe1cwg2a6WmQixw3HJZoO_UAWlkgp5C0,17065
+leap_model_parser/leap_graph_editor.py,sha256=wbvM9zDo-nLn-MPRDLJM9C7EKtE43D1KqciwsjvjDc0,17888
 leap_model_parser/model_parser.py,sha256=1EeSwlaznRRMgLeltAcDnlRWZCMpWeXySglYQEYeiRI,6618
 leap_model_parser/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 leap_model_parser/utils/layerpedia/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -18,8 +18,8 @@ leap_model_parser/utils/uicomponents/generatenodedata.py,sha256=LRaPlO5jJ9pUtkvL
 leap_model_parser/utils/uicomponents/tensorflowinscpection.py,sha256=ym613z9iQKPDBpr0RYD35bTABdm1L-Ez86G47BYT7qw,6775
 leap_model_parser/utils/uicomponents/ui_components.json,sha256=0lsxwOLElW1E-imCcdh3zKPWgzFuQ_bApG6aHvYfTvo,410591
 leap_model_parser/utils/uicomponents/ui_components_config.yaml,sha256=cRH8T-c3TAL0nfefRvt9pFsjbTWNEg38NRyHR7RpJsk,19534
-leap_model_parser-0.1.185.dev9.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-leap_model_parser-0.1.185.dev9.dist-info/METADATA,sha256=RMMdB_x3LtzHS-yeS2HxC_ooP8K-8sWK0x2FEb10zHE,1113
-leap_model_parser-0.1.185.dev9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-leap_model_parser-0.1.185.dev9.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
-leap_model_parser-0.1.185.dev9.dist-info/RECORD,,
+leap_model_parser-0.1.185.dev11.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+leap_model_parser-0.1.185.dev11.dist-info/METADATA,sha256=OO79gPOX78m2BeUv5ulVMARwxmgL918c_Nvfu7qmxd8,1114
+leap_model_parser-0.1.185.dev11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+leap_model_parser-0.1.185.dev11.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
+leap_model_parser-0.1.185.dev11.dist-info/RECORD,,