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

{leap_model_parser-0.1.233 → leap_model_parser-0.1.234}/PKG-INFO

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

{leap_model_parser-0.1.233 → leap_model_parser-0.1.234}/leap_model_parser/model_parser.py

@@ -16,6 +16,7 @@ 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
 
@@ -44,7 +45,7 @@ class ModelParser:
         if custom_layers is None:
             self.custom_layers = {}
 
-        self.custom_layers = {**self.custom_layers, **onnx_custom_layers}
+        self.custom_layers = {**self.custom_layers, **onnx_custom_layers, **onnx_custom_objects_map}
 
         self.mapping_connections = mapping_connections
 
@@ -157,7 +158,6 @@ 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,7 +172,6 @@ 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.233 → leap_model_parser-0.1.234}/pyproject.toml

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