From ae317f0cb45b0b21b33c9123c321832c2237b9cf Mon Sep 17 00:00:00 2001 From: Sanchit Raj Date: Thu, 31 Oct 2024 16:31:11 +0530 Subject: [PATCH 1/2] dependency issue --- requirements.txt | 1 - 1 file changed, 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 9985daab..6f839c56 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,4 @@ tensorflow >= 1.10.0, < 2.0.0 -tensorflow-gpu >= 1.10.0, < 2.0.0 tensorflow-probability >= 0.3.0, < 0.8.0 funcsigs >= 1.0.2 regex >= 2018.01.10 From e62843b00b6427fb50b5c4d1405628fd85e62a09 Mon Sep 17 00:00:00 2001 From: Sanchit Raj Date: Thu, 31 Oct 2024 21:42:01 +0530 Subject: [PATCH 2/2] optimize test_make_chained_transformation --- 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..f33b1db0 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()