Skip to content

Commit 0696506

Browse files
committed
[TESTS] Fix ONNX serialization test failures: setuptools bump + converter bug fix
### Details: - Bump setuptools upper bound from <80.10 to <84 to allow 80.10.x patch releases and 83.x - Fix wrong method call in onnx_prototxt_converter.py: serialize_proto -> deserialize_proto in the bytes branch of save_model ### Tickets: - N/A
1 parent a41b4dd commit 0696506

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

src/bindings/python/constraints.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ pytest-timeout==2.4.0
1010
# Python bindings
1111
build<1.6
1212
pygments>=2.8.1
13-
setuptools>=77,<80.10
13+
setuptools>=77,<84
1414
sympy>=1.10
1515
wheel>=0.38.1
1616
patchelf<=0.17.2.4

src/frontends/onnx/tests/onnx_prototxt_converter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def _get_output_file_path(path, extension):
5858

5959
def save_model(proto, f, format=None, save_as_external_data=False, all_tensors_to_one_file=True, location=None, size_threshold=1024, convert_attribute=False):
6060
if isinstance(proto, bytes):
61-
proto = onnx.serialization.registry.get("protobuf").serialize_proto(proto, onnx.ModelProto())
61+
proto = onnx.serialization.registry.get("protobuf").deserialize_proto(proto, onnx.ModelProto())
6262

6363
if save_as_external_data:
6464
convert_model_to_external_data(proto, all_tensors_to_one_file, location, size_threshold, convert_attribute)

0 commit comments

Comments
 (0)