Avoid overwritten existing custom fields in dnsmos annotation.

lhotse-speech · Jan 22, 2025 · de6ff96 · de6ff96
1 parent 509f4a9
commit de6ff96
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lhotse/workflows/dnsmos.py b/lhotse/workflows/dnsmos.py
@@ -209,5 +209,8 @@ def _annotate_cuts(
 
         for future in tqdm(futures, desc="Processing"):
             cut, result = future.result()
-            new_cut = fastcopy(cut, custom=result)
-            yield new_cut
+            if cut.custom is not None:
+                cut.custom.update(result)
+            else:
+                cut.custom = result
+            yield cut