Merge pull request udacity#169 from kkaneda/kaneda/fix_test_get_init_…

…cell tv-script-generation: Fix test_get_init_cell for TensorFlow 1.2 (#334)
sebastianstock · Jul 21, 2017 · df38789 · df38789
2 parents 3bf801a + f08b9cf
commit df38789
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tv-script-generation/problem_unittests.py b/tv-script-generation/problem_unittests.py
@@ -182,7 +182,7 @@ def test_get_inputs(get_inputs):
 
 def test_get_init_cell(get_init_cell):
     with tf.Graph().as_default():
-        test_batch_size_ph = tf.placeholder(tf.int32)
+        test_batch_size_ph = tf.placeholder(tf.int32, [])
         test_rnn_size = 256
 
         cell, init_state = get_init_cell(test_batch_size_ph, test_rnn_size)