leap-model-parser 0.1.234__tar.gz → 0.1.234.dev0__tar.gz

{leap_model_parser-0.1.234 → leap_model_parser-0.1.234.dev0}/PKG-INFO

@@ -1,9 +1,9 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: leap-model-parser
-Version: 0.1.234
+Version: 0.1.234.dev0
 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,10 +17,11 @@ 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.181)
+Requires-Dist: onnx2kerastl (==0.0.182.dev0)
 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.234 → leap_model_parser-0.1.234.dev0}/leap_model_parser/model_parser.py

@@ -16,7 +16,6 @@ from leap_model_rebuilder import rebuild_model  # type: ignore
 from onnx2kerastl import onnx_to_keras  # type: ignore
 from onnx2kerastl.converter import ConvertedResponse  # type: ignore
 from onnx2kerastl.customonnxlayer import onnx_custom_layers  # type: ignore
-from onnx2kerastl.customonnxlayer import onnx_custom_objects_map  # type: ignore
 from onnx2kerastl.exceptions import OnnxUnsupported  # type: ignore
 from tensorflow.keras.models import load_model  # type: ignore
 
@@ -45,7 +44,7 @@ class ModelParser:
         if custom_layers is None:
             self.custom_layers = {}
 
-        self.custom_layers = {**self.custom_layers, **onnx_custom_layers, **onnx_custom_objects_map}
+        self.custom_layers = {**self.custom_layers, **onnx_custom_layers}
 
         self.mapping_connections = mapping_connections
 
@@ -158,6 +157,7 @@ class ModelParser:
             else:
                 raise e
 
+
     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)
         return self.convert_to_keras_model(imported_model)
@@ -172,6 +172,7 @@ class ModelParser:
 
         _orig_get_json_type = json_utils.get_json_type
 
+
         def _patched_get_json_type(obj):  # type: ignore
             # Handle numpy dtype explicitly
             if isinstance(obj, np.dtype):

{leap_model_parser-0.1.234 → leap_model_parser-0.1.234.dev0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "leap-model-parser"
-version = "0.1.234"
+version = "0.1.234.dev0"
 description = ""
 authors = ["idan <idan.yogev@tensorleap.ai>"]
 license = "MIT"
@@ -18,13 +18,13 @@ 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.181"
+onnx2kerastl = "0.0.182.dev0"
 keras-data-format-converter = "0.1.24"
 leap-model-rebuilder = "0.1.7"
 tensorflow-io-gcs-filesystem = "0.34.0"
 code-loader = ">=1.0.127"
 
-[tool.poetry.dev-dependencies]
+[poetry.group.dev.dependencies]
 pytest = "^7.1.1"
 mypy = "^0.941"
 grappa = "^1.0.1"