From 2ec78b3f85300d6df9e94406ac51fe3f00111dea Mon Sep 17 00:00:00 2001
From: vvye <ekaiser.hellwege@gmail.com>
Date: Sun, 12 Sep 2021 13:48:32 +0200
Subject: [PATCH] Add argument for printing timelines
---
run.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/run.py b/run.py
index 1b1354b..f4944db 100644
--- a/run.py
+++ b/run.py
@@ -20,10 +20,11 @@ def main(args):
for gold_timeline_name, gold_timeline in gold_timelines.items():
print(f'Topic {topic}, gold timeline {gold_timeline_name}')
- print()
timeline = timeline_generation.make_timeline(articles, gold_timeline, keywords)
- timeline_generation.print_timeline(timeline)
+
+ if args.print_timelines:
+ timeline_generation.print_timeline(timeline)
if args.timeline_output_path:
filename = Path(args.timeline_output_path) / f'{topic}_{gold_timeline_name}.txt'
@@ -43,6 +44,9 @@ def main(args):
if __name__ == '__main__':
parser = argparse.ArgumentParser()
+ parser.add_argument('--print_timelines',
+ action='store_true',
+ help='whether to print the timelines to the console after generating them')
parser.add_argument('--timeline_output_path',
type=str,
help='a directory to save the generated timeline files '
--
GitLab