Minor fixes

k2-fsa · Aug 4, 2023 · 37db8a1 · 37db8a1
1 parent e04b0c8
commit 37db8a1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 9 deletions.
diff --git a/examples/libriheavy/matching.py b/examples/libriheavy/matching.py
@@ -361,14 +361,12 @@ def write(
         for seg in segments:
             id = f"{current_cut.id}_{cut_segment_index[current_cut.id]}"
             cut_segment_index[current_cut.id] += 1
-            custom = (
-                {
-                    "texts": [seg["ref"], seg["hyp"]],
-                    "pre_texts": [seg["pre_ref"], seg["pre_hyp"]],
-                    "begin_byte": seg["begin_byte"],
-                    "end_byte": seg["end_byte"],
-                },
-            )
+            custom = {
+                "texts": [seg["ref"], seg["hyp"]],
+                "pre_texts": [seg["pre_ref"], seg["pre_hyp"]],
+                "begin_byte": seg["begin_byte"],
+                "end_byte": seg["end_byte"],
+            }
             if params.output_post_texts:
                 custom["post_texts"] = [seg["post_ref"], seg["post_hyp"]]
             supervision = SupervisionSegment(

diff --git a/examples/libriheavy/matching_parallel.py b/examples/libriheavy/matching_parallel.py
@@ -51,7 +51,7 @@ def get_args():
     parser.add_argument(
         "--batch-size",
         type=int,
-        default=100,
+        default=50,
         help="""The number of cuts in a batch.
         """,
     )