leap-model-parser 0.1.223__py3-none-any.whl → 0.1.224__py3-none-any.whl

leap_model_parser/leap_graph_editor.py

@@ -49,6 +49,7 @@ class LeapGraphEditor:
                 self._add_connection_to_node(new_metric_node_id, input_name, input_node_id)
         elif node_connection.node.type in (NodeMappingType.Loss, NodeMappingType.CustomLoss):
             new_loss_node_id = self._add_loss_node(node_connection.node.name,
+                                                   node_connection.node.user_unique_name,
                                                    node_connection.node.type == NodeMappingType.CustomLoss,
                                                    node_connection.node.arg_names)
             for input_name, node in node_connection.node_inputs.items():
@@ -159,7 +160,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, arg_names: Optional[List[str]]=None) -> str:
+    def _add_loss_node(self, loss_name: str, user_unique_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'
@@ -169,7 +170,7 @@ class LeapGraphEditor:
             new_node_id,
             loss_node_name,
             position=[0, 0],
-            data={'type': loss_type, 'selected': loss_name, 'name': loss_name},
+            data={'type': loss_type, 'selected': loss_name, 'name': loss_name, 'user_unique_name': user_unique_name},
             inputs={},
             outputs={
                 f'{new_node_id}-loss': ConnectionOutput([])

{leap_model_parser-0.1.223.dist-info → leap_model_parser-0.1.224.dist-info}/METADATA

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

{leap_model_parser-0.1.223.dist-info → leap_model_parser-0.1.224.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=MD9ksO54VywikEAb2lxaNHq8Nbp8oiJab2Yn8UB-DRI,16841
-leap_model_parser/leap_graph_editor.py,sha256=iiAR8-ynfnq--Q0LK8z6aujJvJrVrSTZXtZWGWHXb44,13409
+leap_model_parser/leap_graph_editor.py,sha256=0SDHNVkMq94PXWaTqZzMlzJMRLWebOj4uyz64c_KCTQ,13559
 leap_model_parser/model_parser.py,sha256=Je37-FjrNgi8w17rBIt_b71xm-BoiopBAvgW30qfb7o,6614
 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.223.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-leap_model_parser-0.1.223.dist-info/METADATA,sha256=bN_EmO_V9rD2BnFK4dpB4XNOIrSwks7Lc-lMkpgun-c,1104
-leap_model_parser-0.1.223.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-leap_model_parser-0.1.223.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
-leap_model_parser-0.1.223.dist-info/RECORD,,
+leap_model_parser-0.1.224.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+leap_model_parser-0.1.224.dist-info/METADATA,sha256=XzCmWdsrO-N07khELKcWtfS5G7e-vVj5rF4hIe9qJ-0,1104
+leap_model_parser-0.1.224.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+leap_model_parser-0.1.224.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
+leap_model_parser-0.1.224.dist-info/RECORD,,