From 4498bae74e0ca0f61546aac88572c83630cbb0db Mon Sep 17 00:00:00 2001 From: Tiago Freitas Pereira <tiagofrepereira@gmail.com> Date: Fri, 13 Nov 2020 11:31:48 +0100 Subject: [PATCH] Simplify tests [nose] --- bob/learn/tensorflow/tests/data/db_to_tfrecords_config.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/bob/learn/tensorflow/tests/data/db_to_tfrecords_config.py b/bob/learn/tensorflow/tests/data/db_to_tfrecords_config.py index 637d8c55..e711f9a4 100644 --- a/bob/learn/tensorflow/tests/data/db_to_tfrecords_config.py +++ b/bob/learn/tensorflow/tests/data/db_to_tfrecords_config.py @@ -4,10 +4,7 @@ from bob.learn.tensorflow.data import dataset_using_generator mnist = tf.keras.datasets.mnist (x_train, y_train), (_, _) = mnist.load_data() -samples = [ - [a, b, c] - for a, b, c in zip(tf.keras.backend.arange(len(x_train)), x_train, y_train) -] +samples = zip(tf.keras.backend.arange(len(x_train)), x_train, y_train) def reader(sample): -- GitLab