From 46146cc886e8e2f7ca967e38cc44250352bf8716 Mon Sep 17 00:00:00 2001 From: Matthias Bisping Date: Mon, 13 Feb 2023 18:44:55 +0100 Subject: [PATCH] Change default value of number of sentences to generate --- synthesis/segment/segments.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/synthesis/segment/segments.py b/synthesis/segment/segments.py index 910a27b..02e9fc6 100644 --- a/synthesis/segment/segments.py +++ b/synthesis/segment/segments.py @@ -65,7 +65,7 @@ def split_into_figure_and_caption(rectangle: Rectangle): return figure_box, caption_box -def generate_random_text_block(rectangle: Rectangle, n_sentences=3000) -> ContentRectangle: +def generate_random_text_block(rectangle: Rectangle, n_sentences=None) -> ContentRectangle: block = TextBlock( *rectangle.coords, font=pick_random_font_available_on_system( @@ -87,7 +87,7 @@ def generate_random_table_caption(rectangle: Rectangle) -> ContentRectangle: return generate_random_caption(rectangle, f"Tabl {rnd.randint(1, 20)}") -def generate_random_caption(rectangle: Rectangle, caption_start, n_sentences=1000) -> ContentRectangle: +def generate_random_caption(rectangle: Rectangle, caption_start, n_sentences=None) -> ContentRectangle: block = TextBlock( *rectangle.coords, text_generator=CaptionGenerator(caption_start=caption_start),