leap-model-parser 0.1.185.dev7__py3-none-any.whl → 0.1.185.dev9__py3-none-any.whl

leap_model_parser/leap_graph_editor.py

@@ -317,7 +317,7 @@ class LeapGraphEditor:
         elif 'Input' in input_node.type.value:
             input_index = int(input_node.type.value.replace('Input', ''))
             origin_name = self.keras_model.inputs[input_index].node.layer.name
-            input_node_id = self._find_input_node_by_origin_name(origin_name)
+            input_node_id = self._find_input_node_by_origin_name(origin_name).id
             self.model_graph[input_node_id].data['output_name'] = input_node.name
             if input_node_id is None:
                 raise Exception(f"Couldn't find input node by origin name {origin_name}")

leap_model_parser/model_parser.py

@@ -60,48 +60,26 @@ class ModelParser:
                 f"Unable to import external version, {str(model_path)} file format isn't supported")
 
         file_path = str(model_path)
-        try:
-            model_schema, keras_model_with_weights, error_info = model_to_keras_converter(file_path)
-            model_generator = KerasJsonModelImport(self.custom_layers)
-
-            keras_model = keras_model_with_weights
-            if keras_model is None:
-                keras_model = tf.keras.models.model_from_json(json.dumps(model_schema))
-            layer_name_to_inbound_nodes = {
-                layer.name: layer.inbound_nodes
-                for layer in keras_model.layers
-            }
-
-            graph, connected_inputs = model_generator.generate_graph(
-                model_schema, layer_name_to_inbound_nodes)
-
-            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
-        except Exception as e:
-            if model_type.value in [ImportModelTypeEnum.H5_TF2.value, ImportModelTypeEnum.PB_TF2.value]:
-                if model_type.value == ImportModelTypeEnum.H5_TF2.value:
-                    keras_model = self._load_keras_model_with_custom_layers(
-                        file_path)
-                else:
-                    keras_model = self._get_k_model_from_pb_path(file_path)
-
-                rebuilt_model = rebuild_model(keras_model)
-                model_schema, keras_model, error_info = self.convert_to_keras_model(
-                    rebuilt_model)
-
-                layer_name_to_inbound_nodes = {
-                    layer.name: layer.inbound_nodes
-                    for layer in keras_model.layers
-                }
-                model_generator = KerasJsonModelImport(self.custom_layers)
-                graph, connected_inputs = model_generator.generate_graph(
-                    model_schema, layer_name_to_inbound_nodes)
-                return graph, connected_inputs, keras_model, error_info
-            else:
-                raise e
+        model_schema, keras_model_with_weights, error_info = model_to_keras_converter(file_path)
+        model_generator = KerasJsonModelImport(self.custom_layers)
+
+        keras_model = keras_model_with_weights
+        if keras_model is None:
+            keras_model = tf.keras.models.model_from_json(json.dumps(model_schema))
+        layer_name_to_inbound_nodes = {
+            layer.name: layer.inbound_nodes
+            for layer in keras_model.layers
+        }
+
+        graph, connected_inputs = model_generator.generate_graph(
+            model_schema, layer_name_to_inbound_nodes)
+
+        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
+
 
     def _get_k_model_from_pb_path(self, file_path: str):
         tar_file = tarfile.open(file_path)

{leap_model_parser-0.1.185.dev7.dist-info → leap_model_parser-0.1.185.dev9.dist-info}/METADATA

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

{leap_model_parser-0.1.185.dev7.dist-info → leap_model_parser-0.1.185.dev9.dist-info}/RECORD

@@ -6,8 +6,8 @@ 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=x7HOH6iaASfzJgwMRHgF5SQS-iFOF5j9yCG0mDC9HEA,16794
-leap_model_parser/leap_graph_editor.py,sha256=FHPXJGjhVpdHxcB2EGowW3K-93M31N9oS-LPKhhdA8k,17062
-leap_model_parser/model_parser.py,sha256=fHy3U8oEDCQp_KEpYJKY6DAGf69jCA_EmzToASw8QKQ,7786
+leap_model_parser/leap_graph_editor.py,sha256=b2rIpp7KcZHAe1cwg2a6WmQixw3HJZoO_UAWlkgp5C0,17065
+leap_model_parser/model_parser.py,sha256=1EeSwlaznRRMgLeltAcDnlRWZCMpWeXySglYQEYeiRI,6618
 leap_model_parser/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 leap_model_parser/utils/layerpedia/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 leap_model_parser/utils/layerpedia/layerpedia.py,sha256=1syubfXBTB630TVkgcQ-Ge7Qe9Zbr6EtZRreuqCJnQ8,9292
@@ -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.185.dev7.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-leap_model_parser-0.1.185.dev7.dist-info/METADATA,sha256=eLKAdEPhS-tPhjBB3qLF-Rtu-wQ5Ht2EuTt3mxisq-Q,1113
-leap_model_parser-0.1.185.dev7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-leap_model_parser-0.1.185.dev7.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
-leap_model_parser-0.1.185.dev7.dist-info/RECORD,,
+leap_model_parser-0.1.185.dev9.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+leap_model_parser-0.1.185.dev9.dist-info/METADATA,sha256=RMMdB_x3LtzHS-yeS2HxC_ooP8K-8sWK0x2FEb10zHE,1113
+leap_model_parser-0.1.185.dev9.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+leap_model_parser-0.1.185.dev9.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
+leap_model_parser-0.1.185.dev9.dist-info/RECORD,,