qurator-spk/dinglehopper
1
0
Fork 0
mirror of https://github.com/qurator-spk/dinglehopper.git synced 2025-06-13 05:39:57 +02:00
Code Issues Releases Wiki Activity

🚧 Use our own removesuffix() as we still support Python 3.8

Browse source
This commit is contained in:
Gerber, Mike 2024-12-12 13:21:49 +01:00
parent 1f98a4eb12
commit f82479eff2
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/dinglehopper/cli_line_dirs.py
View file

@ -14,6 +14,11 @@ from .word_error_rate import word_error_rate_n, words_normalized
def removesuffix(text, suffix): 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): if suffix and text.endswith(suffix):
return text[: -len(suffix)] return text[: -len(suffix)]
return text 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)): for gt_fn in find_all_files(gt_dir, lambda fn: fn.endswith(gt_suffix)):
ocr_fn = os.path.join( ocr_fn = os.path.join(
ocr_dir, 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, + ocr_suffix,
) )
if not os.path.exists(ocr_fn): if not os.path.exists(ocr_fn):