From 9e21514fb1ca29f9c930fba9b1fde5f3a202caa4 Mon Sep 17 00:00:00 2001
From: Tiago Freitas Pereira <tiagofrepereira@gmail.com>
Date: Fri, 13 Nov 2020 11:01:50 +0100
Subject: [PATCH] Update test case

---
 bob/learn/tensorflow/tests/data/db_to_tfrecords_config.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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 52799ddd..637d8c55 100644
--- a/bob/learn/tensorflow/tests/data/db_to_tfrecords_config.py
+++ b/bob/learn/tensorflow/tests/data/db_to_tfrecords_config.py
@@ -4,7 +4,10 @@ from bob.learn.tensorflow.data import dataset_using_generator
 mnist = tf.keras.datasets.mnist
 
 (x_train, y_train), (_, _) = mnist.load_data()
-samples = (tf.keras.backend.arange(len(x_train)), x_train, y_train)
+samples = [
+    [a, b, c]
+    for a, b, c in zip(tf.keras.backend.arange(len(x_train)), x_train, y_train)
+]
 
 
 def reader(sample):
-- 
GitLab