From fa0da607ea8e1e4fbc680410e091469a712e90f0 Mon Sep 17 00:00:00 2001 From: zhangyang2057 Date: Mon, 29 Jan 2024 14:19:48 +0800 Subject: [PATCH] Support multiple kmodels for accuracy_test. --- tests/accuracy_test/nuc_proxy_accuracy_test.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/accuracy_test/nuc_proxy_accuracy_test.py b/tests/accuracy_test/nuc_proxy_accuracy_test.py index b5f5b45207..910b946199 100644 --- a/tests/accuracy_test/nuc_proxy_accuracy_test.py +++ b/tests/accuracy_test/nuc_proxy_accuracy_test.py @@ -176,8 +176,12 @@ def recv_worker(conn, target): if dict['app'] == 1: dict['app'] = recv_file(conn, case_dir, target.logger) - if dict['kmodel'] == 1: - dict['kmodel'] = recv_file(conn, case_dir, target.logger) + kmodel_num = dict['kmodel'] + if kmodel_num > 0: + file_names = [] + for i in range(kmodel_num): + file_names.append(recv_file(conn, case_dir, target.logger)) + dict['kmodel'] = ' '.join(file_names) if dict['description'] == 1: dict['description'] = recv_file(conn, case_dir, target.logger)