Skip to content
Snippets Groups Projects
Commit 1591b8ef authored by Eva Bardou's avatar Eva Bardou :frog:
Browse files

Remove Path.with_suffix() usage when saving the JSON prediction

parent ccce9058
No related branches found
No related tags found
1 merge request!302Remove Path.with_suffix() usage when saving the JSON prediction
Hide whitespace changes
Inline Side-by-side
dan/ocr/predict/inference.py
+ 1
− 1
View file @ 1591b8ef
...@@ -413,7 +413,7 @@ def process_batch( ...@@ -413,7 +413,7 @@ def process_batch(
) )
result["attention_gif"] = gif_filename result["attention_gif"] = gif_filename
json_filename = Path(output, image_path.stem).with_suffix(".json") json_filename = Path(output, f"{image_path.stem}.json")
logger.info(f"Saving JSON prediction in {json_filename}") logger.info(f"Saving JSON prediction in {json_filename}")
json_filename.write_text(json.dumps(result, indent=2)) json_filename.write_text(json.dumps(result, indent=2))
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment