175747717c34502a0d13dfe709853f074ee5be90,magenta/models/shared/events_rnn_graph_test.py,EventSequenceRNNGraphTest,testBuildGraphWithAttention,#EventSequenceRNNGraphTest#,59

Before Change

    g = events_rnn_graph.build_graph(
        "train", self.config,
        sequence_example_file_paths=[self._sequence_file.name])
    self.assertTrue(isinstance(g, tf.Graph))


if __name__ == "__main__":
  tf.test.main()

After Change

  def testBuildGraphWithAttention(self):
    self.config.hparams.attn_length = 10
    with tf.Graph().as_default():
      events_rnn_graph.get_build_graph_fn(
          "train", self.config,
          sequence_example_file_paths=[self._sequence_file.name])()


if __name__ == "__main__":
  tf.test.main()

Instances