From 353a39c39a31417e444d5a20d0035674e439a66a Mon Sep 17 00:00:00 2001 From: p-wysocki Date: Wed, 22 May 2024 10:47:03 +0200 Subject: [PATCH] test fix --- .../python/src/openvino/runtime/opset15/ops.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/src/bindings/python/src/openvino/runtime/opset15/ops.py b/src/bindings/python/src/openvino/runtime/opset15/ops.py index 9f2a4ab9ceac7d..2b7406099c71ba 100644 --- a/src/bindings/python/src/openvino/runtime/opset15/ops.py +++ b/src/bindings/python/src/openvino/runtime/opset15/ops.py @@ -65,12 +65,14 @@ def col2im( :return: The new node performing Col2Im operation. """ - for param in [strides, dilations]: - if param is None: - param = [1, 1] - for param in [pads_begin, pads_end]: - if param is None: - param = [0, 0] + if strides is None: + strides = [1, 1] + if dilations is None: + dilations = [1, 1] + if pads_begin is None: + pads_begin = [0, 0] + if pads_end is None: + pads_end = [0, 0] return _get_node_factory_opset15().create( "Col2Im", as_nodes(data, output_size, kernel_size, name=name),