Skip to content

Commit 06a8066

Browse files
author
Julia Kamelina
authored
update mo import (#2972)
1 parent 976e8ea commit 06a8066

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tools/model_tools/src/openvino/model_zoo/omz_converter.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -225,9 +225,11 @@ def main():
225225
if str(mo_path).lower().endswith('.py'):
226226
mo_dir = mo_path.parent
227227
else:
228-
mo_package_path, stderr = _common.get_package_path(args.python, 'mo')
228+
mo_package_path, stderr = _common.get_package_path(args.python, 'openvino.tools.mo')
229229
if mo_package_path is None:
230-
sys.exit('Unable to load Model Optimizer. Errors occurred: {}'.format(stderr))
230+
mo_package_path, stderr = _common.get_package_path(args.python, 'mo')
231+
if mo_package_path is None:
232+
sys.exit('Unable to load Model Optimizer. Errors occurred: {}'.format(stderr))
231233
mo_dir = mo_package_path.parent
232234

233235
output_dir = args.download_dir if args.output_dir is None else args.output_dir

0 commit comments

Comments
 (0)