add string constants '0'/'1' to bool normalizer (#4402)

* add string constants '0'/'1' to bool normalizer

* fix bug
This commit is contained in:
Svetlana Dolinina authored and GitHub committed 2021-02-24 13:17:32 +03:00
1 parent abe59efec9
commit 5ee00a989f
1 file changed
+3 -1
+3 -1
View File
@@ -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')))