From 6c6aa67e3ed4d60c9b655002406e0e3770da0edb Mon Sep 17 00:00:00 2001
From: vvye <ekaiser.hellwege@gmail.com>
Date: Wed, 6 Oct 2021 21:25:50 +0200
Subject: [PATCH] Rename argument
---
run.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/run.py b/run.py
index 2fa4d1c..3ad471d 100644
--- a/run.py
+++ b/run.py
@@ -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)
--
GitLab