Skip to content

Commit 76b67f1

Browse files
authored
Fixed parse_partial_shape() to work with aligned shape format. (#3598)
1 parent ec8c27b commit 76b67f1

File tree

1 file changed

+2
-0
lines changed
  • tools/accuracy_checker/openvino/tools/accuracy_checker

1 file changed

+2
-0
lines changed

tools/accuracy_checker/openvino/tools/accuracy_checker/utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -910,6 +910,8 @@ def end_telemetry(tm):
910910

911911
def parse_partial_shape(partial_shape):
912912
ps = str(partial_shape)
913+
if ps[0] == '[' and ps[-1] == ']':
914+
ps = ps[1:-1]
913915
preprocessed = ps.replace('{', '(').replace('}', ')').replace('?', '-1')
914916
if '[' not in preprocessed:
915917
preprocessed = preprocessed.replace('(', '').replace(')', '')

0 commit comments

Comments
 (0)