From fc7eeab292c761f760aa223fada056b0d77e8b49 Mon Sep 17 00:00:00 2001 From: DLPerf <88604684+DLPerf@users.noreply.github.com> Date: Thu, 26 Aug 2021 17:49:26 +0800 Subject: [PATCH] Update dataset_utils_test.py --- tests/data/data/dataset_utils_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/data/data/dataset_utils_test.py b/tests/data/data/dataset_utils_test.py index 841df670..662070eb 100644 --- a/tests/data/data/dataset_utils_test.py +++ b/tests/data/data/dataset_utils_test.py @@ -34,7 +34,7 @@ def _tran_c(data): chained_tran = dsutils.make_chained_transformation( [_tran_a, _tran_b, _tran_c]) - dataset = dataset.map(chained_tran) + dataset = dataset.map(chained_tran,num_parallel_calls=tf.data.experimental.AUTOTUNE) iterator = dataset.make_one_shot_iterator() elem = iterator.get_next()