Added mo.py to wheel packages (#4731)

This commit is contained in:
Mikhail Ryzhov 2021-03-12 15:57:12 +03:00 committed by GitHub
parent 882e377ef9
commit 0100810dd6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions

View File

@ -1,5 +1,6 @@
[options] [options]
py_modules = py_modules =
mo
mo_tf mo_tf
mo_caffe mo_caffe
mo_mxnet mo_mxnet

View File

@ -37,7 +37,7 @@ py_modules = []
for name in os.listdir(): for name in os.listdir():
if re.match('requirements(.*)\.txt', name): if re.match('requirements(.*)\.txt', name):
requirements_txt.append(name) requirements_txt.append(name)
if re.match('mo_(.*)\.py', name): if re.match('mo(.*)\.py', name):
py_modules.append(name.split('.')[0]) py_modules.append(name.split('.')[0])
# Minimal set of dependencies # Minimal set of dependencies