Skip to content

Instantly share code, notes, and snippets.

@magdalini-anastasiadou
Created April 16, 2021 13:22
Show Gist options
  • Save magdalini-anastasiadou/4c3521f7198036e92fced5555ac00c04 to your computer and use it in GitHub Desktop.
Save magdalini-anastasiadou/4c3521f7198036e92fced5555ac00c04 to your computer and use it in GitHub Desktop.
Merge NLU files in the same directory
import argparse
from rasa.shared.nlu.training_data import loading
def merge_nlu_data():
"""Merge Rasa NLU files"""
parser = argparse.ArgumentParser(description="Merge Rasa NLU files")
parser.add_argument(
"--nlu",
type=str,
help="File or folder containing your NLU training data.",
default="data/nlu/",
required=True,
)
parser.add_argument(
"--out",
type=str,
help="Output file name for the result NLU",
required=True,
)
parser.add_argument(
"--format",
type=str,
help="Output format",
required=False,
default="md",
choices=["md", "json", "yaml"],
)
args = parser.parse_args()
train_data = loading.load_data(args.nlu)
with open(args.out, "w") as outfile:
if args.format == "md":
outfile.write(train_data.nlu_as_markdown())
elif args.format == "json":
outfile.write(train_data.nlu_as_json())
else:
outfile.write(train_data.nlg_as_yaml())
if __name__ == "__main__":
merge_nlu_data()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment