Skip to content

Commit

Permalink
fix pylint issue
Browse files Browse the repository at this point in the history
Signed-off-by: wyoung1 <[email protected]>
  • Loading branch information
wyoung1 committed Oct 18, 2024
1 parent 647a7f6 commit 2dd27ec
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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))
Expand Down

0 comments on commit 2dd27ec

Please sign in to comment.