Skip to content

Commit 1ef9e4f

Browse files
committed
fix code style
1 parent 74fda8b commit 1ef9e4f

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

ppstructure/kie/predict_kie_token_ser.py

+6-2
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,9 @@ def __call__(self, img):
115115
else:
116116
data[idx] = [data[idx]]
117117
if self.args.use_onnx:
118-
input_tensor = {name: data[idx] for idx, name in enumerate(self.input_tensor)}
118+
input_tensor = {
119+
name: data[idx] for idx, name in enumerate(self.input_tensor)
120+
}
119121
self.output_tensors = self.predictor.run(None, input_tensor)
120122
else:
121123
for idx in range(len(self.input_tensor)):
@@ -125,7 +127,9 @@ def __call__(self, img):
125127

126128
outputs = []
127129
for output_tensor in self.output_tensors:
128-
output = output_tensor if self.args.use_onnx else output_tensor.copy_to_cpu()
130+
output = (
131+
output_tensor if self.args.use_onnx else output_tensor.copy_to_cpu()
132+
)
129133
outputs.append(output)
130134
preds = outputs[0]
131135

0 commit comments

Comments
 (0)