From 807391f86368f4489f1a00cafc3a11c43dccb2a6 Mon Sep 17 00:00:00 2001 From: Xiaoyu Date: Thu, 4 May 2023 18:58:23 +0000 Subject: [PATCH 1/2] Skip docker system test for bert example --- examples/test_bert_ptq_cpu.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/examples/test_bert_ptq_cpu.py b/examples/test_bert_ptq_cpu.py index 44b1557e9b..32ce4ef934 100644 --- a/examples/test_bert_ptq_cpu.py +++ b/examples/test_bert_ptq_cpu.py @@ -26,14 +26,15 @@ def check_output(footprint): assert all([value > 0 for value in v.metrics.value.values()]) +# Skip docker_system test until bug is fixed: https://github.com/docker/docker-py/issues/3113 @pytest.mark.parametrize("search_algorithm", ["tpe"]) @pytest.mark.parametrize("execution_order", ["joint"]) -@pytest.mark.parametrize("system", ["local_system", "aml_system", "docker_system"]) +@pytest.mark.parametrize("system", ["local_system", "aml_system"]) @pytest.mark.parametrize("olive_json", ["bert_config.json"]) def test_bert(search_algorithm, execution_order, system, olive_json): # TODO: add gpu e2e test - if system == "docker_system" and platform.system() == "Windows": - pytest.skip("Skip Linux containers on Windows host test case.") + # if system == "docker_system" and platform.system() == "Windows": + # pytest.skip("Skip Linux containers on Windows host test case.") from olive.workflows import run as olive_run @@ -50,7 +51,7 @@ def test_bert(search_algorithm, execution_order, system, olive_json): # set aml_system as dev olive_config["systems"]["aml_system"]["config"]["is_dev"] = True # set docker_system as dev - olive_config["systems"]["docker_system"]["config"]["is_dev"] = True + # olive_config["systems"]["docker_system"]["config"]["is_dev"] = True # update host and target olive_config["engine"]["host"] = system if system != "docker_system" else "local_system" From 33eb68ad2e126b9de34f1ce42b8b656a37a6407f Mon Sep 17 00:00:00 2001 From: Xiaoyu Date: Thu, 4 May 2023 18:59:36 +0000 Subject: [PATCH 2/2] fix format --- examples/test_bert_ptq_cpu.py | 1 - 1 file changed, 1 deletion(-) diff --git a/examples/test_bert_ptq_cpu.py b/examples/test_bert_ptq_cpu.py index 32ce4ef934..2b3f123063 100644 --- a/examples/test_bert_ptq_cpu.py +++ b/examples/test_bert_ptq_cpu.py @@ -4,7 +4,6 @@ # -------------------------------------------------------------------------- import json import os -import platform from pathlib import Path import pytest