leap-model-parser 0.1.231.dev2__tar.gz → 0.1.231.dev3__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: leap-model-parser
-Version: 0.1.231.dev2
+Version: 0.1.231.dev3
 Summary: 
 License: MIT
 License-File: LICENSE

{leap_model_parser-0.1.231.dev2 → leap_model_parser-0.1.231.dev3}/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.231.dev3}/pyproject.toml

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