leap-model-parser 0.1.185.dev5__py3-none-any.whl → 0.1.185.dev7__py3-none-any.whl

leap_model_parser/keras_json_model_import.py

@@ -349,6 +349,8 @@ def _add_origin_name_to_nodes(nodes: Dict[str, Node]):
         if "name" in node.data:
             node.data["origin_name"] = node.data["name"]
             node.data.pop("name")
+        elif "output_name" in node.data:
+            node.data["origin_name"] = node.data["output_name"]
 
 
 def _is_input_node(node: Node) -> bool:

leap_model_parser/leap_graph_editor.py

@@ -102,7 +102,8 @@ class LeapGraphEditor:
             #     raise Exception("prediction_type_name is required for loss connection")
 
             new_loss_node_id = self._add_loss_node(node_connection.node.name,
-                                                   node_connection.node.type == NodeMappingType.CustomLoss)
+                                                   node_connection.node.type == NodeMappingType.CustomLoss,
+                                                   node_connection.node.arg_names)
             for input_name, node in node_connection.node_inputs.items():
                 input_node_id = self._find_or_add_input_node(node)
                 # if node.type == NodeMappingType.Layer:
@@ -231,7 +232,7 @@ class LeapGraphEditor:
         self.model_graph[new_node_id] = metric_node
         return new_node_id
 
-    def _add_loss_node(self, loss_name: str, is_custom_loss: bool) -> str:
+    def _add_loss_node(self, loss_name: str, is_custom_loss: bool, arg_names: Optional[List[str]]=None) -> str:
         new_node_id = self._generate_new_node_id()
 
         loss_type = 'CustomLoss' if is_custom_loss else 'Loss'
@@ -250,6 +251,9 @@ class LeapGraphEditor:
             #     f'{new_node_id}-loss': {'connections': []}
             # }
         )
+        if arg_names is not None:
+            loss_node.data['arg_names'] = arg_names
+
 
         self.model_graph[new_node_id] = loss_node
         return new_node_id
@@ -306,7 +310,17 @@ class LeapGraphEditor:
         if input_node.type in (NodeMappingType.Input, NodeMappingType.GroundTruth):
             input_node_id = self._find_encoder_node_id(input_node.name)
             if input_node_id is None:
-                input_node_id = self._add_ground_truth_node(input_node.name)
+                if input_node.type == NodeMappingType.GroundTruth:
+                    input_node_id = self._add_ground_truth_node(input_node.name)
+                else:
+                    raise Exception(f'Couldnt find input node name {input_node.name}')
+        elif 'Input' in input_node.type.value:
+            input_index = int(input_node.type.value.replace('Input', ''))
+            origin_name = self.keras_model.inputs[input_index].node.layer.name
+            input_node_id = self._find_input_node_by_origin_name(origin_name)
+            self.model_graph[input_node_id].data['output_name'] = input_node.name
+            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'):
             input_node_id = self._find_node_by_origin_name(input_node.name).id
         else:

{leap_model_parser-0.1.185.dev5.dist-info → leap_model_parser-0.1.185.dev7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.185.dev5
+Version: 0.1.185.dev7
 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.87.dev5)
+Requires-Dist: code-loader (==1.0.87.dev7)
 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.185.dev5.dist-info → leap_model_parser-0.1.185.dev7.dist-info}/RECORD

@@ -5,8 +5,8 @@ leap_model_parser/contract/graph.py,sha256=c5rSlJJQ3JzLhyeniglzrP5GH0fp8UZDQKZw1
 leap_model_parser/contract/importmodelresponse.py,sha256=GlvnKS8xrebU2Sj0dxqtEhAOOo3DlOtT7AMJ2BlzH9E,145
 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=-mmNrXVWvdNs4-T9vx0OS7QThBQ_wiEvlEaB56-AWqQ,16689
-leap_model_parser/leap_graph_editor.py,sha256=B60ayqfKh-9Slv_WbeaqsN_i4zO3CiRbM0YSomVjBNc,16187
+leap_model_parser/keras_json_model_import.py,sha256=x7HOH6iaASfzJgwMRHgF5SQS-iFOF5j9yCG0mDC9HEA,16794
+leap_model_parser/leap_graph_editor.py,sha256=FHPXJGjhVpdHxcB2EGowW3K-93M31N9oS-LPKhhdA8k,17062
 leap_model_parser/model_parser.py,sha256=fHy3U8oEDCQp_KEpYJKY6DAGf69jCA_EmzToASw8QKQ,7786
 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.dev5.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-leap_model_parser-0.1.185.dev5.dist-info/METADATA,sha256=tprGBPmgqYMoJLkEZVmeyxE0siEhyW1zc5iYONtccaE,1113
-leap_model_parser-0.1.185.dev5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-leap_model_parser-0.1.185.dev5.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
-leap_model_parser-0.1.185.dev5.dist-info/RECORD,,
+leap_model_parser-0.1.185.dev7.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+leap_model_parser-0.1.185.dev7.dist-info/METADATA,sha256=eLKAdEPhS-tPhjBB3qLF-Rtu-wQ5Ht2EuTt3mxisq-Q,1113
+leap_model_parser-0.1.185.dev7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+leap_model_parser-0.1.185.dev7.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
+leap_model_parser-0.1.185.dev7.dist-info/RECORD,,