Skip to content
Snippets Groups Projects
Commit 4498bae7 authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Simplify tests [nose]

parent 9e21514f
No related branches found
No related tags found
1 merge request!89Cleanup
Hide whitespace changes
Inline Side-by-side
bob/learn/tensorflow/tests/data/db_to_tfrecords_config.py
+ 1
− 4
View file @ 4498bae7
...@@ -4,10 +4,7 @@ from bob.learn.tensorflow.data import dataset_using_generator ...@@ -4,10 +4,7 @@ from bob.learn.tensorflow.data import dataset_using_generator
mnist = tf.keras.datasets.mnist mnist = tf.keras.datasets.mnist
(x_train, y_train), (_, _) = mnist.load_data() (x_train, y_train), (_, _) = mnist.load_data()
samples = [ samples = zip(tf.keras.backend.arange(len(x_train)), x_train, y_train)
[a, b, c]
for a, b, c in zip(tf.keras.backend.arange(len(x_train)), x_train, y_train)
]
def reader(sample): def reader(sample):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment