Skip to content
Snippets Groups Projects
Commit 6c6aa67e authored by vvye's avatar vvye
Browse files

Rename argument

parent 8dddd5d1
No related branches found
No related tags found
No related merge requests found
......@@ -26,7 +26,7 @@ def main(args):
print(f'Topic {topic}, gold timeline {gold_timeline_name}')
by_tokens = args.timeline_length == 'tokens'
by_tokens = args.length_constraint == 'tokens'
timeline = timeline_generation.make_timeline(articles, gold_timeline, keywords, by_tokens)
......@@ -58,7 +58,7 @@ if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--dataset', type=str, choices=['timeline17', 'crisis', 'entities'], help='the dataset to use',
required=True)
parser.add_argument('--timeline_length', type=str, choices=['sentences', 'tokens'], default='sentences',
parser.add_argument('--length_constraint', type=str, choices=['sentences', 'tokens'], default='sentences',
help='what constraint to impose on the length of the generated timeline '
'(number of setences or number of tokens',
required=True)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment