Commit 9e21514f authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

Update test case

parent fe66b694
Pipeline #45533 failed with stage
in 13 minutes and 29 seconds
......@@ -4,7 +4,10 @@ from 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):
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