leap-model-parser 0.1.187__py3-none-any.whl → 0.1.188__py3-none-any.whl

leap_model_parser/leap_graph_editor.py

@@ -207,9 +207,9 @@ class LeapGraphEditor:
     def _handle_input_node_with_index(self, input_node: NodeMapping) -> str:
         input_index = int(input_node.type.value.replace('Input', ''))
         origin_name = self.keras_model.inputs[input_index].node.layer.name
-        input_node = self._find_input_node_by_origin_name(origin_name)
-        assert input_node is not None, f"Input node with origin name {origin_name} not found in model graph"
-        input_node_id = input_node.id
+        input_node_by_origin = self._find_input_node_by_origin_name(origin_name)
+        assert input_node_by_origin is not None, f"Input node with origin name {origin_name} not found in model graph"
+        input_node_id = input_node_by_origin.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:
@@ -246,9 +246,6 @@ class LeapGraphEditor:
 
         return input_node_id
 
-    def _find_prediction_node(self, prediction_index):
-        pass
-
     def _get_all_loss_node_ids(self):
         loss_node_ids = []
         for node_id, node_response in self.model_graph.items():

{leap_model_parser-0.1.187.dist-info → leap_model_parser-0.1.188.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.187
+Version: 0.1.188
 Summary: 
 Home-page: https://github.com/tensorleap/leap-model-parser
 License: MIT
@@ -12,7 +12,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
-Requires-Dist: code-loader (==1.0.89)
+Requires-Dist: code-loader (==1.0.91)
 Requires-Dist: keras-data-format-converter (==0.1.22)
 Requires-Dist: leap-model-rebuilder (==0.1.7)
 Requires-Dist: numpy (>=1.22.3,<2.0.0)

{leap_model_parser-0.1.187.dist-info → leap_model_parser-0.1.188.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=T9IIABlJ9NwAEUGRFlsrsD-c3yeFfUazbbvE6mLXnMA,13022
+leap_model_parser/leap_graph_editor.py,sha256=ToPnBUh31hXrA83BMTo631DLQm36PgFE_uTYB-boFAM,12983
 leap_model_parser/model_parser.py,sha256=kodF6aCPXP5FzZR9To2txqu6YdwklmhdJbyQKO_hghM,6587
 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.187.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-leap_model_parser-0.1.187.dist-info/METADATA,sha256=dTub6iKB7pQaX8UJN1QOz6AgY7cIMVnHNemlYrBax_8,1103
-leap_model_parser-0.1.187.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-leap_model_parser-0.1.187.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
-leap_model_parser-0.1.187.dist-info/RECORD,,
+leap_model_parser-0.1.188.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+leap_model_parser-0.1.188.dist-info/METADATA,sha256=k7_-TXZVkVHf7vcGkSDGi_8sqbXTCD-fsqEoS44UarM,1103
+leap_model_parser-0.1.188.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+leap_model_parser-0.1.188.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
+leap_model_parser-0.1.188.dist-info/RECORD,,