From 5ee00a989f116075848c60d789f5f34c999f12bc Mon Sep 17 00:00:00 2001 From: Svetlana Dolinina Date: Wed, 24 Feb 2021 13:17:32 +0300 Subject: [PATCH] add string constants '0'/'1' to bool normalizer (#4402) * add string constants '0'/'1' to bool normalizer * fix bug --- model-optimizer/mo/front/extractor.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/model-optimizer/mo/front/extractor.py b/model-optimizer/mo/front/extractor.py index 210bb2d8552..fde985e6416 100644 --- a/model-optimizer/mo/front/extractor.py +++ b/model-optimizer/mo/front/extractor.py @@ -132,7 +132,7 @@ def attr_getter(node: Node, name: str): def bool_to_str(node: Node, attr: str): - # Function converts 0/1 or bool False/True values to str 'false'/'true' which need to appear in IR + # Function converts 0/1 or bool False/True or '0'/'1' values to str 'false'/'true' which need to appear in IR attribute_name = node.soft_get(attr, None) if attribute_name is None: return None @@ -140,6 +140,8 @@ def bool_to_str(node: Node, attr: str): return str(attribute_name).lower() elif attribute_name in [0, 1]: return str(bool(attribute_name)).lower() + elif attribute_name in ['0', '1']: + return str(bool(int(attribute_name))).lower() else: raise Error('Wrong value {} for boolean attribute {} in node {}'.format( attribute_name, attr, node.soft_get('name')))