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

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

@@ -1,9 +1,9 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.241
+Version: 0.1.242.dev4
 Summary: 
+Home-page: https://github.com/tensorleap/leap-model-parser
 License: MIT
-License-File: LICENSE
 Author: idan
 Author-email: idan.yogev@tensorleap.ai
 Requires-Python: >=3.8,<3.11
@@ -17,11 +17,10 @@ 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.185)
+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.241 → 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.241 → leap_model_parser-0.1.242.dev4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "leap-model-parser"
-version = "0.1.241"
+version = "0.1.242.dev4"
 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.185"
+onnx2kerastl = "0.0.186"
 keras-data-format-converter = "0.1.24"
 leap-model-rebuilder = "0.1.7"
 tensorflow-io-gcs-filesystem = "0.34.0"