From 12b95bd960191225e5332597d504a8fbe63939e1 Mon Sep 17 00:00:00 2001 From: JosselinSomervilleRoberts Date: Thu, 14 Mar 2024 21:10:44 -0700 Subject: [PATCH] Add musicsheet to E2E script --- src/image2structure/end2end.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/image2structure/end2end.py b/src/image2structure/end2end.py index 0a6473e..07ab037 100644 --- a/src/image2structure/end2end.py +++ b/src/image2structure/end2end.py @@ -98,6 +98,13 @@ def main(): f"{command_base} --tmp-path {args.destination_path}/tmp/{args.type}/{category} {latex_suffix}" f" >> {args.destination_path}/logs/{args.type}/{category}.log" ) + elif args.type == "musicsheet": + for i, category in enumerate(["music"]): + music_suffix: str = f"musicsheet --subcategory {category}" + collect_commands.append( + f"{command_base} --tmp-path {args.destination_path}/tmp/{args.type}/{category} {music_suffix}" + f" >> {args.destination_path}/logs/{args.type}/{category}.log" + ) else: raise ValueError(f"Invalid type: {args.type}") os.makedirs(f"{args.destination_path}/logs/{args.type}", exist_ok=True)