Skip to content

Commit

Permalink
style(rapidocr_onnxruntime): Code wrapping with black
Browse files Browse the repository at this point in the history
  • Loading branch information
SWHL committed May 21, 2024
1 parent a291ca1 commit cb14cc8
Showing 1 changed file with 10 additions and 2 deletions.
12 changes: 10 additions & 2 deletions python/rapidocr_onnxruntime/utils/parse_parameters.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,15 +119,23 @@ def __call__(self, config, **kwargs):
global_dict, det_dict, cls_dict, rec_dict = self.parse_kwargs(**kwargs)
new_config = {
"Global": self.update_global_params(config["Global"], global_dict),
"Det": self.update_params(config["Det"], det_dict, "det_", ["det_model_path", "det_use_cuda", "det_use_dml"]),
"Det": self.update_params(
config["Det"],
det_dict,
"det_",
["det_model_path", "det_use_cuda", "det_use_dml"],
),
"Cls": self.update_params(
config["Cls"],
cls_dict,
"cls_",
["cls_label_list", "cls_model_path", "cls_use_cuda", "cls_use_dml"],
),
"Rec": self.update_params(
config["Rec"], rec_dict, "rec_", ["rec_model_path", "rec_use_cuda", "rec_use_dml"]
config["Rec"],
rec_dict,
"rec_",
["rec_model_path", "rec_use_cuda", "rec_use_dml"],
),
}

Expand Down

0 comments on commit cb14cc8

Please sign in to comment.