leap-model-parser 0.1.256.dev3__tar.gz → 0.1.257.dev0__tar.gz

{leap_model_parser-0.1.256.dev3 → leap_model_parser-0.1.257.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.256.dev3
+Version: 0.1.257.dev0
 Summary: 
 Home-page: https://github.com/tensorleap/leap-model-parser
 License: MIT
@@ -17,7 +17,7 @@ Requires-Dist: keras-data-format-converter (==0.1.24)
 Requires-Dist: leap-model-rebuilder (==0.1.7)
 Requires-Dist: numpy (>=1.22.3,<2.0.0)
 Requires-Dist: onnx (==1.13.0)
-Requires-Dist: onnx2kerastl (==0.0.198.dev3)
+Requires-Dist: onnx2kerastl (==0.0.198)
 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"

{leap_model_parser-0.1.256.dev3 → leap_model_parser-0.1.257.dev0}/leap_model_parser/leap_graph_editor.py

@@ -1,3 +1,4 @@
+import logging
 from enum import Enum
 from typing import Optional, Dict, Any, List
 
@@ -55,8 +56,15 @@ class LeapGraphEditor:
             for input_name, node in node_connection.node_inputs.items():
                 input_node_id = self._find_or_add_input_node(node)
                 self._add_connection_to_node(new_loss_node_id, input_name, input_node_id)
+        elif node_connection.node.type.value == 'CustomLatentSpace':
+            new_latent_space_node_id = self._add_latent_space_node(
+                node_connection.node.name,
+                node_connection.node.user_unique_name, 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)
+                self._add_connection_to_node(new_latent_space_node_id, input_name, input_node_id)
         else:
-            raise Exception(f"Can't add node of type {node_connection.node.type.name}")
+            logging.warning(f"Skipping connection of unsupported node type {node_connection.node.type.name}")
 
     def model_graph_dict(self) -> Dict[str, Any]:
         json_model_graph = {}
@@ -176,6 +184,23 @@ class LeapGraphEditor:
         self.model_graph[new_node_id] = metric_node
         return new_node_id
 
+    def _add_latent_space_node(self, latent_space_name: str,
+                             user_unique_name: str, arg_names: List[str]) -> str:
+        new_node_id = self._generate_new_node_id()
+
+        latent_space_node = Node(
+            new_node_id,
+            'CustomLatentSpace',
+            position=[0, 0],
+            data={'latent_space_name': latent_space_name, 'type': 'CustomLatentSpace',
+                  'name': latent_space_name,
+                  'arg_names': arg_names, "user_unique_name": user_unique_name},
+            inputs={},
+            outputs={})
+
+        self.model_graph[new_node_id] = latent_space_node
+        return new_node_id
+
     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()
 

{leap_model_parser-0.1.256.dev3 → leap_model_parser-0.1.257.dev0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "leap-model-parser"
-version = "0.1.256.dev3"
+version = "0.1.257.dev0"
 description = ""
 authors = ["idan <idan.yogev@tensorleap.ai>"]
 license = "MIT"
@@ -18,7 +18,7 @@ tensorflow = {version = "2.12.0", markers = "platform_machine  == 'x86_64'"}
 tensorflow-macos = {version = "2.12.0", markers = "platform_machine  == 'arm64'"}
 numpy = "^1.22.3"
 onnx = "1.13.0"
-onnx2kerastl = "0.0.198.dev3"
+onnx2kerastl = "0.0.198"
 keras-data-format-converter = "0.1.24"
 leap-model-rebuilder = "0.1.7"
 tensorflow-io-gcs-filesystem = "0.34.0"