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
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()