Added mo.py to wheel packages (#4731)

2021-03-12 15:57:12 +03:00 · 2021-03-12 15:57:12 +03:00 · 0100810dd6
commit 0100810dd6
parent 882e377ef9
2 changed files with 2 additions and 1 deletions
--- a/inference-engine/ie_bridges/python/wheel/meta/openvino-dev.setup.cfg
+++ b/inference-engine/ie_bridges/python/wheel/meta/openvino-dev.setup.cfg
@ -1,5 +1,6 @@
 [options]
 py_modules = 
+	mo
 	mo_tf
 	mo_caffe
 	mo_mxnet
--- a/model-optimizer/setup.py
+++ b/model-optimizer/setup.py
@ -37,7 +37,7 @@ py_modules = []
 for name in os.listdir():
    if re.match('requirements(.*)\.txt', name):
        requirements_txt.append(name)
-    if re.match('mo_(.*)\.py', name):
+    if re.match('mo(.*)\.py', name):
        py_modules.append(name.split('.')[0])

 # Minimal set of dependencies