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