leap-model-parser 0.1.226__py3-none-any.whl → 0.1.227__py3-none-any.whl

leap_model_parser/leap_graph_editor.py

@@ -127,6 +127,22 @@ class LeapGraphEditor:
         self.model_graph[new_node_id] = ground_truth_node
         return new_node_id
 
+    def _add_input_encoder_not_connected_to_the_model_node(self, input_name: str) -> str:
+        new_node_id = self._generate_new_node_id()
+        ground_truth_node = Node(
+            new_node_id,
+            'Input',
+            position=[0, 0],
+            data={'name': input_name, 'output_name': input_name,
+                  'type': 'Input', "selected": input_name},
+            inputs={},
+            outputs={
+                f'{new_node_id}-{input_name}': ConnectionOutput([])
+            }
+        )
+        self.model_graph[new_node_id] = ground_truth_node
+        return new_node_id
+
     def _add_visualizer_node(self, visualizer_name: str, visualizer_type: str,
                              user_unique_name: str, arg_names: List[str]) -> str:
         new_node_id = self._generate_new_node_id()
@@ -240,6 +256,8 @@ class LeapGraphEditor:
             if input_node_id is None:
                 if input_node.type == NodeMappingType.GroundTruth:
                     input_node_id = self._add_ground_truth_node(input_node.name)
+                elif input_node.type == NodeMappingType.Input:
+                    input_node_id = self._add_input_encoder_not_connected_to_the_model_node(input_node.name)
                 else:
                     raise Exception(f'Couldnt find input node name {input_node.name}')
         elif 'Input' in input_node.type.value:

{leap_model_parser-0.1.226.dist-info → leap_model_parser-0.1.227.dist-info}/METADATA

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

{leap_model_parser-0.1.226.dist-info → leap_model_parser-0.1.227.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=h9-eb41W67goutRV5c3VkStxvntgb6EmPFdPeshhtsY,17058
-leap_model_parser/leap_graph_editor.py,sha256=0SDHNVkMq94PXWaTqZzMlzJMRLWebOj4uyz64c_KCTQ,13559
+leap_model_parser/leap_graph_editor.py,sha256=s6iVO06xM6LJ8ZjSxkr1srYxsHhGl2KnQjQoc9g4_iE,14329
 leap_model_parser/model_parser.py,sha256=W8FGKCBOBClIl0sdBlTdoSRT8sIdVrlbO8veNMouJ7o,8169
 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.226.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-leap_model_parser-0.1.226.dist-info/METADATA,sha256=GtWZ8_i_RR1EKOGGcdMfvk-H8mMGtB0hMJTLppRTF44,1104
-leap_model_parser-0.1.226.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-leap_model_parser-0.1.226.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
-leap_model_parser-0.1.226.dist-info/RECORD,,
+leap_model_parser-0.1.227.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+leap_model_parser-0.1.227.dist-info/METADATA,sha256=ceu2bCLoaj3r8fBM4gUmXZr_Xh0_cElbXuSw5UqldAQ,1104
+leap_model_parser-0.1.227.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+leap_model_parser-0.1.227.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
+leap_model_parser-0.1.227.dist-info/RECORD,,