Fixed tests

tcotte · Dec 20, 2017 · 1511461 · 1511461
1 parent 9bf680d
commit 1511461
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/project_tests.py b/project_tests.py
@@ -99,7 +99,7 @@ def test_optimize(optimize):
     layers_output = tf.Variable(tf.zeros(shape))
     correct_label = tf.placeholder(tf.float32, [None, None, None, num_classes])
     learning_rate = tf.placeholder(tf.float32)
-    logits, train_op, cross_entropy_loss = optimize(layers_output, correct_label, learning_rate, num_classes)
+    logits, train_op, cross_entropy_loss, accuracy_op = optimize(layers_output, correct_label, learning_rate, num_classes)
 
     _assert_tensor_shape(logits, [2*3*4, num_classes], 'Logits')