[POT] Fix precommit tests (#14972)

Co-authored-by: Alexander Kozlov <alexander.kozlov@intel.com>
This commit is contained in:
Liubov Talamanova 2023-01-06 13:44:54 +00:00 committed by GitHub
parent 7fab95fef2
commit ddd4f050c7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 10 additions and 1 deletions

View File

@ -80,6 +80,9 @@ def test_default_quantization(model_params, tmp_path, models, algorithm, preset)
}}]})
model_name, model_framework, model_precision, expected_accuracy_dict = model_params
if model_framework == 'mxnet':
pytest.skip('Skipped due to conflict with numpy version in mxnet #99501.')
run_quantization(models=models,
model_name=model_name,
model_framework=model_framework,

View File

@ -76,6 +76,9 @@ def _params(request):
def test_compression(_params, tmp_path, models):
model_name, model_framework, algorithm, preset, subset_size, expected_accuracy, additional_params, device = _params
if model_framework == 'mxnet':
pytest.skip('Skipped due to conflict with numpy version in mxnet #99501.')
algorithm_config = make_algo_config(algorithm, preset, subset_size, additional_params, device)
if model_name in CASCADE_MAP:

View File

@ -19,7 +19,7 @@ from openvino.tools.pot.statistics.collector import StatisticsCollector
from .utils.data_helper import dump_intermediate_data, load_json
EPS = 1e-6
EPS = 2e-5
def get_fq_nodes_stats_algo(model, preset, bits, is_weights, clipping_value=None):

View File

@ -36,6 +36,9 @@ def _params(request):
def test_unify_scales(_params, tmp_path, models):
model_name, model_framework, algorithm, preset, device = _params
if model_framework == 'mxnet':
pytest.skip('Skipped due to conflict with numpy version in mxnet #99501.')
algorithm_config = Dict({
'algorithms': [{
'name': algorithm,