From c0d984296fdd602f9651aea53b9883c066e7c2d1 Mon Sep 17 00:00:00 2001 From: Michelle Casbon Date: Thu, 14 Dec 2023 12:24:59 -0800 Subject: [PATCH] Internal change. PiperOrigin-RevId: 591018853 --- seqio/dataset_providers_test.py | 1 + 1 file changed, 1 insertion(+) diff --git a/seqio/dataset_providers_test.py b/seqio/dataset_providers_test.py index 6ef62981..3e1b2cac 100644 --- a/seqio/dataset_providers_test.py +++ b/seqio/dataset_providers_test.py @@ -1932,6 +1932,7 @@ def test_get_dataset_enc_dec_sharded_and_packed(self): assert_dataset(output_ds, expected, expected_dtypes=expected_dtypes) + def register_dummy_task( task_name: str, dataset_fn: Callable[[str, str], tf.data.Dataset],