diff --git a/.github/workflows/mo.yml b/.github/workflows/mo.yml index 7dcb851466a..2b9e786f421 100644 --- a/.github/workflows/mo.yml +++ b/.github/workflows/mo.yml @@ -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 diff --git a/model-optimizer/automation/package_BOM.txt b/model-optimizer/automation/package_BOM.txt index cf76f4e150d..69fbe70b421 100644 --- a/model-optimizer/automation/package_BOM.txt +++ b/model-optimizer/automation/package_BOM.txt @@ -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 diff --git a/model-optimizer/mo/front/common/extractors/__init__.py b/model-optimizer/mo/front/common/extractors/__init__.py new file mode 100644 index 00000000000..e69de29bb2d diff --git a/model-optimizer/mo/utils/ir_reader/extenders/__init__.py b/model-optimizer/mo/utils/ir_reader/extenders/__init__.py new file mode 100644 index 00000000000..e69de29bb2d