leap-model-parser 0.1.245.dev4__tar.gz → 0.1.245.dev6__tar.gz

{leap_model_parser-0.1.245.dev4 → leap_model_parser-0.1.245.dev6}/PKG-INFO

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

{leap_model_parser-0.1.245.dev4 → leap_model_parser-0.1.245.dev6}/leap_model_parser/model_parser.py

@@ -67,24 +67,43 @@ def _is_empty_file(path: str) -> bool:
         return False
 
 
-def _raise_model_file_error(path: str, kind: str, original_exc: Exception) -> None:
+def _validate_model_file(
+    path: str,
+    kind: str,
+    onnx_model: Optional["onnx.ModelProto"] = None,
+    post_load_check: bool = False,
+) -> None:
     suffix = Path(path).suffix
     if _is_empty_file(path):
         raise InvalidModelFile(
             f"The provided {kind} model file is empty (0 bytes). "
             "Please provide a real model binary."
-        ) from original_exc
+        )
     if _is_git_lfs_pointer(path):
         raise InvalidModelFile(
             f"This {suffix} file is a Git LFS pointer, not the actual model binary. "
             "Run `git lfs pull` (or re-download the artifact) to fetch the real file."
-        ) from original_exc
+        )
     snippet = _text_snippet(path)
     if snippet is not None:
         raise InvalidModelFile(
             f"The provided model is not a real {kind} model binary although its suffix is {suffix}. "
             f"File's content starts with:\n{snippet}."
-        ) from original_exc
+        )
+    if post_load_check and onnx_model is not None:
+        if (
+            len(onnx_model.graph.node) == 0
+            and len(onnx_model.graph.input) == 0
+            and len(onnx_model.graph.output) == 0
+        ):
+            raise InvalidModelFile(
+                "The provided onnx model file appears to be empty or invalid "
+                "(no graph nodes, inputs, or outputs)."
+            )
+
+
+def _raise_model_file_error(path: str, kind: str, original_exc: Exception) -> None:
+    _validate_model_file(path, kind)
     raise InvalidModelFile(f"Failed to open {kind} model file: {original_exc}") from original_exc
 
 class ModelParser:
@@ -190,6 +209,12 @@ class ModelParser:
             onnx_model = onnx.load_model(file_path)
         except Exception as e:
             _raise_model_file_error(file_path, "onnx", e)
+        _validate_model_file(
+            file_path,
+            "onnx",
+            onnx_model=onnx_model,
+            post_load_check=True,
+        )
         input_all = [_input.name for _input in onnx_model.graph.input]
         input_initializer = [
             node.name for node in onnx_model.graph.initializer]

{leap_model_parser-0.1.245.dev4 → leap_model_parser-0.1.245.dev6}/pyproject.toml

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