Merge pull request udacity#117 from Thomas-P/tp_fix

Fix unit test, because the behavior of rnn.MultiRNNCell scope changed.
maxbader · May 9, 2017 · e3d28ce · e3d28ce
2 parents 4258d01 + 5b5bd4a
commit e3d28ce
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tv-script-generation/problem_unittests.py b/tv-script-generation/problem_unittests.py
@@ -208,7 +208,7 @@ def test_build_rnn(build_rnn):
     with tf.Graph().as_default():
         test_rnn_size = 256
         test_rnn_layer_size = 2
-        test_cell = rnn.MultiRNNCell([rnn.BasicLSTMCell(test_rnn_size)] * test_rnn_layer_size)
+        test_cell = rnn.MultiRNNCell([rnn.BasicLSTMCell(test_rnn_size) for _ in range(test_rnn_layer_size)])
 
         test_inputs = tf.placeholder(tf.float32, [None, None, test_rnn_size])
         outputs, final_state = build_rnn(test_cell, test_inputs)
@@ -236,7 +236,7 @@ def test_build_nn(build_nn):
         test_embed_dim = 300
         test_rnn_layer_size = 2
         test_vocab_size = 27
-        test_cell = rnn.MultiRNNCell([rnn.BasicLSTMCell(test_rnn_size)] * test_rnn_layer_size)
+        test_cell = rnn.MultiRNNCell([rnn.BasicLSTMCell(test_rnn_size) for _ in range(test_rnn_layer_size)])
 
         logits, final_state = build_nn(test_cell, test_rnn_size, test_input_data, test_vocab_size, test_embed_dim)