mirror of
https://github.com/qurator-spk/dinglehopper.git
synced 2025-06-10 20:29:57 +02:00
🚧 Use our own removesuffix() as we still support Python 3.8
This commit is contained in:
parent
2bf2529c38
commit
6980d7a252
1 changed files with 6 additions and 1 deletions
|
@ -14,6 +14,11 @@ from .word_error_rate import word_error_rate_n, words_normalized
|
|||
|
||||
|
||||
def removesuffix(text, suffix):
|
||||
"""
|
||||
Remove suffix from text.
|
||||
|
||||
Can be replaced with str.removesuffix when we only support Python >= 3.9.
|
||||
"""
|
||||
if suffix and text.endswith(suffix):
|
||||
return text[: -len(suffix)]
|
||||
return text
|
||||
|
@ -59,7 +64,7 @@ def find_gt_and_ocr_files(gt_dir, gt_suffix, ocr_dir, ocr_suffix) -> Iterator[Tu
|
|||
for gt_fn in find_all_files(gt_dir, lambda fn: fn.endswith(gt_suffix)):
|
||||
ocr_fn = os.path.join(
|
||||
ocr_dir,
|
||||
os.path.relpath(gt_fn, start=gt_dir).removesuffix(gt_suffix)
|
||||
removesuffix(os.path.relpath(gt_fn, start=gt_dir), gt_suffix)
|
||||
+ ocr_suffix,
|
||||
)
|
||||
if not os.path.exists(ocr_fn):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue