fixed rnn_decoders_test for tf1.13

asyml · Feb 28, 2019 · 3c114a9 · 3c114a9
1 parent 2a64f49
commit 3c114a9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/texar/modules/decoders/rnn_decoders_test.py b/texar/modules/decoders/rnn_decoders_test.py
@@ -365,7 +365,8 @@ def test_beam_search_cell(self):
                                           tf.float32)
         _ = beam_cell(cell_input, cell_state)
         # Test if beam_cell is sharing variables with decoder cell.
-        self.assertEqual(len(beam_cell.trainable_variables), 0)
+        for tvar in beam_cell.trainable_variables:
+            self.assertTrue(tvar in decoder.trainable_variables)
 
 if __name__ == "__main__":
     tf.test.main()