From a5c7a64a5b86656ca39b7a15f40e980078371767 Mon Sep 17 00:00:00 2001 From: jiahangxu Date: Tue, 7 Feb 2023 17:39:21 +0800 Subject: [PATCH] debuging integration test for tf --- .github/workflows/integration-test.yml | 2 +- tests/integration_test/test_latency_predictor.py | 14 +++++++------- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/.github/workflows/integration-test.yml b/.github/workflows/integration-test.yml index 29c92bc..b3d266d 100644 --- a/.github/workflows/integration-test.yml +++ b/.github/workflows/integration-test.yml @@ -32,7 +32,7 @@ jobs: - name: Install dependencies run: | - pip install tensorflow==1.15.0 + pip install tensorflow==2.6.0 pip install onnx==1.9.0 pip install torch==1.9.0 pip install torchvision==0.10.0 diff --git a/tests/integration_test/test_latency_predictor.py b/tests/integration_test/test_latency_predictor.py index 35aa3b1..831412e 100644 --- a/tests/integration_test/test_latency_predictor.py +++ b/tests/integration_test/test_latency_predictor.py @@ -79,13 +79,13 @@ def integration_test(model_type, url, ppath, output_name = "tests/integration_te # start testing for pred_name, pred_version in get_predictors(): - try: - since = time.time() - # print(f'nn-meter --{model_type} {ppath} --predictor {pred_name} --predictor-version {pred_version}') - result = subprocess.check_output(['nn-meter', 'predict', f'--{model_type}', f'{ppath}', '--predictor', f'{pred_name}', '--predictor-version', f'{pred_version}']) - runtime = time.time() - since - except NotImplementedError: - logging.error(f"Meets ERROR when checking --{model_type} {ppath} --predictor {pred_name} --predictor-version {pred_version}") + # try: + since = time.time() + # print(f'nn-meter --{model_type} {ppath} --predictor {pred_name} --predictor-version {pred_version}') + result = subprocess.check_output(['nn-meter', 'predict', f'--{model_type}', f'{ppath}', '--predictor', f'{pred_name}', '--predictor-version', f'{pred_version}']) + runtime = time.time() - since + # except NotImplementedError: + # logging.error(f"Meets ERROR when checking --{model_type} {ppath} --predictor {pred_name} --predictor-version {pred_version}") latency_list = parse_latency_info(result.decode('utf-8')) for model, latency in latency_list: