IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bug: assert is keyword, not function (#5131)

Browse Source 
* fix bug: assert is keyword, not function

* fix comment
This commit is contained in:
Svetlana Dolinina
2021-04-08 12:02:18 +03:00
committed by GitHub
parent 5d5bf97280
commit f858a62836
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
model-optimizer/extensions/middle/MakeKaldiConstReshapable.py
View File

@@ -97,9 +97,9 @@ class MakeKaldiConstReshapable(MiddleReplacementPattern):
# check that all Parameters have the same batch
for p in params:
assert(p.shape[0] == batch,
"Parameter {} have batch different from the {}".format(p.soft_get('name', p.id),
params[0].soft_get('name', params[0].id)))
assert p.shape[0] == batch, \
"Parameter {} has batch different from the {}".format(p.soft_get('name', p.id),
params[0].soft_get('name', params[0].id))
# make constants for initialization of ReadValue reshapable
for read in graph.get_op_nodes(op='ReadValue'):