Added mo.py to wheel packages

mryzhov · Mar 10, 2021 · 74b308d · 74b308d
1 parent d7e6e47
commit 74b308d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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

diff --git a/model-optimizer/setup.py b/model-optimizer/setup.py
@@ -37,7 +37,7 @@
 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