leap-model-parser 0.1.159__tar.gz → 0.1.161__tar.gz

{leap_model_parser-0.1.159 → leap_model_parser-0.1.161}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.159
+Version: 0.1.161
 Summary: 
 Home-page: https://github.com/tensorleap/leap-model-parser
 License: MIT
@@ -16,7 +16,7 @@ Requires-Dist: keras-data-format-converter (==0.1.17)
 Requires-Dist: leap-model-rebuilder (==0.1.5)
 Requires-Dist: numpy (>=1.22.3,<2.0.0)
 Requires-Dist: onnx (==1.10.1)
-Requires-Dist: onnx2kerastl (==0.0.145)
+Requires-Dist: onnx2kerastl (==0.0.148)
 Requires-Dist: tensorflow (==2.11.0) ; platform_machine == "x86_64"
 Requires-Dist: tensorflow-io-gcs-filesystem (==0.30.0)
 Requires-Dist: tensorflow-macos (==2.11.0) ; platform_machine == "arm64"

{leap_model_parser-0.1.159 → leap_model_parser-0.1.161}/leap_model_parser/model_parser.py

@@ -13,6 +13,7 @@ from keras_data_format_converter import convert_channels_first_to_last  # type:
 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
 from onnx2kerastl.exceptions import OnnxUnsupported  # type: ignore
 from tensorflow.keras.models import load_model  # type: ignore
 
@@ -37,6 +38,8 @@ class ModelParser:
         if custom_layers is None:
             self.custom_layers = {}
 
+        self.custom_layers = {**self.custom_layers, **onnx_custom_layers}
+
         self._model_types_converter = {
             ImportModelTypeEnum.JSON_TF2.value: self.convert_json_model,
             ImportModelTypeEnum.H5_TF2.value: self.convert_h5_model,

{leap_model_parser-0.1.159 → leap_model_parser-0.1.161}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "leap-model-parser"
-version = "0.1.159"
+version = "0.1.161"
 description = ""
 authors = ["idan <idan.yogev@tensorleap.ai>"]
 license = "MIT"
@@ -18,7 +18,7 @@ tensorflow = {version = "2.11.0", markers = "platform_machine  == 'x86_64'"}
 tensorflow-macos = {version = "2.11.0", markers = "platform_machine  == 'arm64'"}
 numpy = "^1.22.3"
 onnx = "1.10.1"
-onnx2kerastl = "0.0.145"
+onnx2kerastl = "0.0.148"
 keras-data-format-converter = "0.1.17"
 leap-model-rebuilder = "0.1.5"
 tensorflow-io-gcs-filesystem = "0.30.0"