leap-model-parser 0.1.222__tar.gz → 0.1.224__tar.gz

{leap_model_parser-0.1.222 → leap_model_parser-0.1.224}/PKG-INFO

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

{leap_model_parser-0.1.222 → leap_model_parser-0.1.224}/leap_model_parser/leap_graph_editor.py

@@ -31,6 +31,8 @@ class LeapGraphEditor:
                 f"Prediction node with name {prediction_node.name} not found in model graph"
             prediction_node_id = prediction_mapping_node.id
             self.model_graph[prediction_node_id].data['prediction_type'] = prediction_labels_name
+        elif 'Input' in node_connection.node.type.value:
+            self._find_or_add_input_node(node_connection.node)
         elif node_connection.node.type == NodeMappingType.Visualizer:
             new_visualizer_node_id = self._add_visualizer_node(
                 node_connection.node.name, node_connection.node.sub_type,
@@ -47,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():
@@ -157,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'
@@ -167,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.222 → leap_model_parser-0.1.224}/pyproject.toml

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