f25d9afd2572b98c56c3f597ad8da2648cd7b663,stanza/utils/training/common.py,,main,#Any#Any#Any#,21

Before Change

def main(run_treebank, model_dir, model_name):
    paths = default_paths.get_default_paths()
    args = sys.argv[1:]
    if args[0].startswith("--"):
        mode = Mode[args[0][2:].upper()]
        args = args[1:]
    else:
        mode = Mode.TRAIN
    treebank = args[0]
    extra_args = args[1:]

    if treebank.lower() in ("ud_all", "all_ud"):

After Change

        command_args, extra_args = parser.parse_known_args()

    mode = command_args.mode
    treebanks = []

    for treebank in command_args.treebanks:
        if treebank.lower() in ("ud_all", "all_ud"):
            ud_treebank = common.get_ud_treebanks(paths["UDBASE"])

Instances