X

PiperOrigin-RevId: 571265899
google · Oct 6, 2023 · 500e455 · 500e455
1 parent bae5d7e
commit 500e455
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/seqio/helpers.py b/seqio/helpers.py
@@ -297,6 +297,7 @@ def mixture_or_task_with_truncated_data(
         preprocessors=mixture_or_task.preprocessors,
         postprocess_fn=mixture_or_task.postprocessor,
         metric_fns=mixture_or_task.metric_fns,
+        metric_objs=mixture_or_task.metric_objs,
         shuffle_buffer_size=mixture_or_task._shuffle_buffer_size,
     )
     if add_to_seqio_registry: