Commit 4498bae7 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Simplify tests [nose]

parent 9e21514f
......@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment