decode_wm_lm_speed_test.py

from watermarking.arg_classes.wm_arg_class import WmLMArgs
from watermarking.wm_decode_speed_test import main, HumanWrittenDecodeArgs
from transformers import HfArgumentParser

parser = HfArgumentParser((WmLMArgs, HumanWrittenDecodeArgs))
args: WmLMArgs
human_written_args: HumanWrittenDecodeArgs
args, human_written_args = parser.parse_args_into_dataclasses()
human_written_args.wm_args = args
main(human_written_args)