From 2dd27ec3d4c5aad88ef40db261b02abc0669520a Mon Sep 17 00:00:00 2001 From: wyoung1 Date: Sat, 19 Oct 2024 01:10:53 +0800 Subject: [PATCH] fix pylint issue Signed-off-by: wyoung1 --- .../paradigm/multiedge_inference/multiedge_inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/testcasecontroller/algorithm/paradigm/multiedge_inference/multiedge_inference.py b/core/testcasecontroller/algorithm/paradigm/multiedge_inference/multiedge_inference.py index 15a2a76e..4085eafd 100644 --- a/core/testcasecontroller/algorithm/paradigm/multiedge_inference/multiedge_inference.py +++ b/core/testcasecontroller/algorithm/paradigm/multiedge_inference/multiedge_inference.py @@ -105,7 +105,7 @@ def _partition(self, partition_point_list, initial_model_path, sub_model_dir): output_names = point['output_names'] sub_model_path = sub_model_dir + '/' + 'sub_model_' + str(idx+1) + '.onnx' try: - onnx.utils.extract_model(initial_model_path, + onnx.utils.extract_model(initial_model_path, sub_model_path, input_names, output_names) except Exception as e: LOGGER.info(str(e))