leap-model-parser 0.1.168.dev1__tar.gz → 0.1.169__tar.gz

{leap_model_parser-0.1.168.dev1 → leap_model_parser-0.1.169}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: leap-model-parser
-Version: 0.1.168.dev1
+Version: 0.1.169
 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.154)
+Requires-Dist: onnx2kerastl (==0.0.157)
 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.168.dev1 → leap_model_parser-0.1.169}/leap_model_parser/keras_json_model_import.py

@@ -191,6 +191,8 @@ class KerasJsonModelImport:
             return [KerasJsonModelImport._serialize_arg(a) for a in arg]
         if isinstance(arg, dict):
             return {k: KerasJsonModelImport._serialize_arg(v) for k, v in arg.items()}
+        if isinstance(arg, tf.TensorShape):
+            return f'{nojsonable_arg_serialization_prefix}tf.TensorShape({str([v for v in arg])})'
         raise Exception('Unsupported call arg')
 
     @staticmethod

{leap_model_parser-0.1.168.dev1 → leap_model_parser-0.1.169}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "leap-model-parser"
-version = "0.1.168.dev1"
+version = "0.1.169"
 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.154"
+onnx2kerastl = "0.0.157"
 keras-data-format-converter = "0.1.17"
 leap-model-rebuilder = "0.1.5"
 tensorflow-io-gcs-filesystem = "0.30.0"