Enable mo.front.common.extractors module (#5038)

* Enable mo.front.common.extractors module (#5018)

* Enable mo.front.common.extractors module

* Update package_BOM.txt

* Test MO wheel content
This commit is contained in:
Dmitry Kurtaev
2021-04-01 13:54:09 +03:00
committed by GitHub
parent f37e14c614
commit 81b0aec201
4 changed files with 13 additions and 1 deletions

View File

@@ -80,7 +80,17 @@ jobs:
python3 setup.py sdist bdist_wheel
working-directory: model-optimizer
- name: Test
- name: Test package content
run: |
echo "src = open('openvino_mo.egg-info/SOURCES.txt', 'rt').read().split()" | tee -a test_wheel.py
echo "ref = open('automation/package_BOM.txt', 'rt').read().split()" | tee -a test_wheel.py
echo "for name in ref:" | tee -a test_wheel.py
echo " if name.endswith('.py'):" | tee -a test_wheel.py
echo " assert name in src or './' + name in src, name + ' file missed'" | tee -a test_wheel.py
python3 test_wheel.py
working-directory: model-optimizer
- name: Test conversion
run: |
wget -q http://download.tensorflow.org/models/mobilenet_v1_2018_08_02/mobilenet_v1_1.0_224.tgz
tar -xf mobilenet_v1_1.0_224.tgz

View File

@@ -792,6 +792,7 @@ mo/front/caffe/python_layer_extractor.py
mo/front/caffe/register_custom_ops.py
mo/front/common/__init__.py
mo/front/common/custom_replacement_registry.py
mo/front/common/extractors/__init__.py
mo/front/common/extractors/utils.py
mo/front/common/find_unsupported_ops.py
mo/front/common/layout.py
@@ -987,6 +988,7 @@ mo/utils/ir_engine/compare_graphs.py
mo/utils/ir_engine/ir_engine.py
mo/utils/ir_reader/__init__.py
mo/utils/ir_reader/extender.py
mo/utils/ir_reader/extenders/__init__.py
mo/utils/ir_reader/extenders/binary_convolution_extender.py
mo/utils/ir_reader/extenders/bucketize_extender.py
mo/utils/ir_reader/extenders/conv_extender.py