leap-model-parser 0.1.226__tar.gz → 0.1.227.dev0__tar.gz

{leap_model_parser-0.1.226 → leap_model_parser-0.1.227.dev0}/PKG-INFO

@@ -1,9 +1,9 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: leap-model-parser
-Version: 0.1.226
+Version: 0.1.227.dev0
 Summary: 
-Home-page: https://github.com/tensorleap/leap-model-parser
 License: MIT
+License-File: LICENSE
 Author: idan
 Author-email: idan.yogev@tensorleap.ai
 Requires-Python: >=3.8,<3.11
@@ -21,6 +21,7 @@ Requires-Dist: onnx2kerastl (==0.0.176)
 Requires-Dist: tensorflow (==2.12.0) ; platform_machine == "x86_64"
 Requires-Dist: tensorflow-io-gcs-filesystem (==0.34.0)
 Requires-Dist: tensorflow-macos (==2.12.0) ; platform_machine == "arm64"
+Project-URL: Homepage, https://github.com/tensorleap/leap-model-parser
 Project-URL: Repository, https://github.com/tensorleap/leap-model-parser
 Description-Content-Type: text/markdown
 

{leap_model_parser-0.1.226 → leap_model_parser-0.1.227.dev0}/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 → leap_model_parser-0.1.227.dev0}/pyproject.toml

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