leap-model-parser 0.1.242__tar.gz → 0.1.242.dev4__tar.gz

{leap_model_parser-0.1.242 → leap_model_parser-0.1.242.dev4}/PKG-INFO

@@ -1,7 +1,8 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.242
+Version: 0.1.242.dev4
 Summary: 
+Home-page: https://github.com/tensorleap/leap-model-parser
 License: MIT
 Author: idan
 Author-email: idan.yogev@tensorleap.ai
@@ -20,7 +21,6 @@ Requires-Dist: onnx2kerastl (==0.0.186)
 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.242 → leap_model_parser-0.1.242.dev4}/leap_model_parser/model_parser.py

@@ -134,8 +134,10 @@ class ModelParser:
     def convert_onnx_model(self, file_path: str) -> Tuple[Dict[str, Node], Model, Optional[str]]:
         if not onnx_imported:
             raise OnnxUnsupported()
-
-        onnx_model = onnx.load_model(file_path)
+        try:
+            onnx_model = onnx.load_model(file_path)
+        except:
+            raise Exception(f"I caught onnx error: {file_path}")
         input_all = [_input.name for _input in onnx_model.graph.input]
         input_initializer = [
             node.name for node in onnx_model.graph.initializer]
@@ -157,6 +159,8 @@ class ModelParser:
                 raise Exception('Unable to open model file. The model might be corrupted or not a valid.')
             else:
                 raise e
+        except:
+            raise Exception(f"I caught h5 error: {file_path}")
 
     def convert_h5_model(self, file_path: str) -> Tuple[Dict[str, Node], Model, Optional[str]]:
         imported_model = self._load_keras_model_with_custom_layers(file_path)

{leap_model_parser-0.1.242 → leap_model_parser-0.1.242.dev4}/pyproject.toml

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