d657fbe414191d55b5e2b77baf3709d7caef3393,magenta/models/rl_tuner/rl_tuner_test.py,RLTunerTest,testTraining,#RLTunerTest#,75

Before Change

    rlt = rl_tuner.RLTuner(self.output_dir, output_every_nth=30)
    rlt.train(num_steps=31, exploration_period=3)

    self.assertTrue(os.path.exists(rlt.save_path + "-30"))
    self.assertTrue(len(rlt.rewards_batched) >= 1)
    self.assertTrue(len(rlt.eval_avg_reward) >= 1)

  def testCompositionStats(self):

After Change

        output_every_nth=30)
    rlt.train(num_steps=31, exploration_period=3)

    checkpoint_dir = os.path.dirname(rlt.save_path)
    checkpoint_files = [
        f for f in os.listdir(checkpoint_dir)
        if os.path.isfile(os.path.join(checkpoint_dir, f))]
    checkpoint_step_30 = [
        f for f in checkpoint_files
        if os.path.basename(rlt.save_path) + "-30" in f]

    self.assertTrue(len(checkpoint_step_30))

Instances