From 81b0aec2019d80812d1e426af7cbdd2e1b8917ea Mon Sep 17 00:00:00 2001 From: Dmitry Kurtaev Date: Thu, 1 Apr 2021 13:54:09 +0300 Subject: [PATCH] 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 --- .github/workflows/mo.yml | 12 +++++++++++- model-optimizer/automation/package_BOM.txt | 2 ++ .../mo/front/common/extractors/__init__.py | 0 .../mo/utils/ir_reader/extenders/__init__.py | 0 4 files changed, 13 insertions(+), 1 deletion(-) create mode 100644 model-optimizer/mo/front/common/extractors/__init__.py create mode 100644 model-optimizer/mo/utils/ir_reader/extenders/__init__.py 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