def add_subparser(subparsers: argparse._SubParsersAction, parents: List[argparse.ArgumentParser]): train_parser = subparsers.add_parser( "train", help="Trains a Rasa model using your NLU data and stories.", parents=parents, formatter_class=argparse.ArgumentDefaultsHelpFormatter, ) train_arguments.set_train_arguments(train_parser) train_subparsers = train_parser.add_subparsers() train_core_parser = train_subparsers.add_parser( "core", parents=parents, conflict_handler="resolve", formatter_class=argparse.ArgumentDefaultsHelpFormatter, help="Trains a Rasa Core model using your stories.", ) train_core_parser.set_defaults(func=train_core) train_nlu_parser = train_subparsers.add_parser( "nlu", parents=parents, formatter_class=argparse.ArgumentDefaultsHelpFormatter, help="Trains a Rasa NLU model using your NLU data.", ) train_nlu_parser.set_defaults(func=train_nlu) train_parser.set_defaults(func=train) train_arguments.set_train_core_arguments(train_core_parser) train_arguments.set_train_nlu_arguments(train_nlu_parser)
def add_subparser( subparsers: SubParsersAction, parents: List[argparse.ArgumentParser] ) -> None: """Add all training parsers. Args: subparsers: subparser we are going to attach to parents: Parent parsers, needed to ensure tree structure in argparse """ train_parser = subparsers.add_parser( "train", help="Trains a Rasa model using your NLU data and stories.", parents=parents, formatter_class=argparse.ArgumentDefaultsHelpFormatter, ) train_arguments.set_train_arguments(train_parser) train_subparsers = train_parser.add_subparsers() train_core_parser = train_subparsers.add_parser( "core", parents=parents, conflict_handler="resolve", formatter_class=argparse.ArgumentDefaultsHelpFormatter, help="Trains a Rasa Core model using your stories.", ) train_core_parser.set_defaults(func=train_core) train_nlu_parser = train_subparsers.add_parser( "nlu", parents=parents, formatter_class=argparse.ArgumentDefaultsHelpFormatter, help="Trains a Rasa NLU model using your NLU data.", ) train_nlu_parser.set_defaults(func=train_nlu) train_parser.set_defaults(func=train) train_arguments.set_train_core_arguments(train_core_parser) train_arguments.set_train_nlu_arguments(train_nlu_parser)