leap-model-parser 0.1.231.dev2__tar.gz → 0.1.232__tar.gz

{leap_model_parser-0.1.231.dev2 → leap_model_parser-0.1.232}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: leap-model-parser
-Version: 0.1.231.dev2
+Version: 0.1.232
 Summary: 
 License: MIT
 License-File: LICENSE
@@ -13,11 +13,11 @@ Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Requires-Dist: code-loader (>=1.0.127)
-Requires-Dist: keras-data-format-converter (==0.1.23.dev0)
+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.179.dev1)
+Requires-Dist: onnx2kerastl (==0.0.180)
 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.231.dev2 → leap_model_parser-0.1.232}/leap_model_parser/model_parser.py

@@ -86,12 +86,19 @@ class ModelParser:
 
         graph, connected_inputs = model_generator.generate_graph(
             model_schema, layer_name_to_inbound_nodes)
-
+        # make sure input order is kept with original model
+        input_list = []
+        for inp in keras_model.inputs:
+            name = inp.name
+            name = name.replace(".", "_")
+            for inp_graph in connected_inputs:
+                if inp_graph.name == name:
+                    input_list.append(inp_graph)
         if self.mapping_connections is not None:
             leap_graph_editor = LeapGraphEditor(graph, keras_model_with_weights)
             leap_graph_editor.add_connections_to_graph(self.mapping_connections)
 
-        return graph, connected_inputs, keras_model_with_weights, error_info
+        return graph, input_list, keras_model_with_weights, error_info
 
     def _get_k_model_from_pb_path(self, file_path: str):
         tar_file = tarfile.open(file_path)

{leap_model_parser-0.1.231.dev2 → leap_model_parser-0.1.232}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "leap-model-parser"
-version = "0.1.231.dev2"
+version = "0.1.232"
 description = ""
 authors = ["idan <idan.yogev@tensorleap.ai>"]
 license = "MIT"
@@ -18,8 +18,8 @@ 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.179.dev1"
-keras-data-format-converter = "0.1.23.dev0"
+onnx2kerastl = "0.0.180"
+keras-data-format-converter = "0.1.24"
 leap-model-rebuilder = "0.1.7"
 tensorflow-io-gcs-filesystem = "0.34.0"
 code-loader = ">=1.0.127"