leap-model-parser 0.1.232__py3-none-any.whl → 0.1.234__py3-none-any.whl

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.234.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2021 TensorLeap
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

{leap_model_parser-0.1.232.dist-info → leap_model_parser-0.1.234.dist-info}/METADATA

@@ -1,9 +1,9 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.232
+Version: 0.1.234
 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.180)
+Requires-Dist: onnx2kerastl (==0.0.181)
 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.232.dist-info → leap_model_parser-0.1.234.dist-info}/RECORD

@@ -1,3 +1,4 @@
+LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
 leap_model_parser/__init__.py,sha256=OAU7rFHAVVWUM-cDtQ4Ohum567KN8M-YTkHZp5KiYbo,132
 leap_model_parser/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 leap_model_parser/contract/graph.py,sha256=lbs_IUnlkvoVjiARBMJee-aNbqeNXNPsUeOEwQliKgU,1170
@@ -6,7 +7,7 @@ leap_model_parser/contract/nodedata.py,sha256=1_ML0nzp3QUZ0_9mGSLhfO4_hqjYMwi0DW
 leap_model_parser/contract/ui_components.json,sha256=0lsxwOLElW1E-imCcdh3zKPWgzFuQ_bApG6aHvYfTvo,410591
 leap_model_parser/keras_json_model_import.py,sha256=h9-eb41W67goutRV5c3VkStxvntgb6EmPFdPeshhtsY,17058
 leap_model_parser/leap_graph_editor.py,sha256=s6iVO06xM6LJ8ZjSxkr1srYxsHhGl2KnQjQoc9g4_iE,14329
-leap_model_parser/model_parser.py,sha256=672YTDpQP38439juRr22UbrLG8YQng8HVmMmLVOy1mI,8510
+leap_model_parser/model_parser.py,sha256=Q2r1eaRX_vr0YtaQThATRbNCG52szWQ4gKolLRF5UCY,8616
 leap_model_parser/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 leap_model_parser/utils/layerpedia/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 leap_model_parser/utils/layerpedia/layerpedia.py,sha256=1syubfXBTB630TVkgcQ-Ge7Qe9Zbr6EtZRreuqCJnQ8,9292
@@ -17,8 +18,8 @@ leap_model_parser/utils/uicomponents/generatenodedata.py,sha256=LRaPlO5jJ9pUtkvL
 leap_model_parser/utils/uicomponents/tensorflowinscpection.py,sha256=ym613z9iQKPDBpr0RYD35bTABdm1L-Ez86G47BYT7qw,6775
 leap_model_parser/utils/uicomponents/ui_components.json,sha256=0lsxwOLElW1E-imCcdh3zKPWgzFuQ_bApG6aHvYfTvo,410591
 leap_model_parser/utils/uicomponents/ui_components_config.yaml,sha256=cRH8T-c3TAL0nfefRvt9pFsjbTWNEg38NRyHR7RpJsk,19534
-leap_model_parser-0.1.232.dist-info/METADATA,sha256=3jnqLtGeBomuwwUqg-9mV4mVHvl-h3fOTuaoKTN5EOE,1138
-leap_model_parser-0.1.232.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-leap_model_parser-0.1.232.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
-leap_model_parser-0.1.232.dist-info/licenses/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-leap_model_parser-0.1.232.dist-info/RECORD,,
+leap_model_parser-0.1.234.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+leap_model_parser-0.1.234.dist-info/METADATA,sha256=HzL7AugLBCHzqTODKDe4jO2UX2G3Gb8wSkcDRGc1cBE,1104
+leap_model_parser-0.1.234.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+leap_model_parser-0.1.234.dist-info/entry_points.txt,sha256=ZvV6EuQt1uAqwapNg5Lo2qjJM9ZG5g2wfzZoLh_Ztyk,77
+leap_model_parser-0.1.234.dist-info/RECORD,,

{leap_model_parser-0.1.232.dist-info → leap_model_parser-0.1.234.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.2.1
+Generator: poetry-core 1.9.0
 Root-Is-Purelib: true
 Tag: py3-none-any