Added tests for segment resizer

nomadkaraoke · Jan 7, 2025 · 98b0d10 · 98b0d10
1 parent 3c247d0
commit 98b0d10
Show file tree

Hide file tree

Showing 4 changed files with 738 additions and 171 deletions.
diff --git a/lyrics_transcriber/output/generator.py b/lyrics_transcriber/output/generator.py
@@ -112,7 +112,7 @@ def generate_outputs(
             outputs.original_txt = self.plain_text.write_original_transcription(transcription_corrected, output_prefix)
 
             # Resize corrected segments to ensure none are longer than max_line_length
-            resized_segments = self.segment_resizer.resize_segments(transcription_corrected.corrected_segments)
+            resized_segments = self.segment_resizer.resize_segments(transcription_corrected.corrected_segments[:2])
             transcription_corrected.resized_segments = resized_segments
             outputs.corrections_json = self.write_corrections_data(transcription_corrected, output_prefix)