hunk.code_window:         def train(config):
               module = _MockModule(10, 20)
               trainer = pl.Trainer(
                   max_epochs=1,
                   callbacks=[
                       TuneReportCallback(
                           {
  
hunk.labels: keep replace keep keep keep keep keep
hunk.after_edit:             module = _MockModule(10., 20.)
               trainer = pl.Trainer(
                   max_epochs=1,
                   callbacks=[TuneReportCallback(on="validation_end")])
               trainer.fit(module)
           analysis = tune.run(train, stop={TRAINING_ITERATION: 1})
           self.assertEqual(analysis.trials[0].last_result["avg_val_loss"],
                            10. * 1.1)
       def testReportCallbackNamed(self):
           def train(config):
               module = _MockModule(10., 20.)
  
hunk.file_path: python/ray/tune/tests/test_integration_pytorch_lightning.py
hunk.type: replace
file: cartpole-dqn-w-param-noise:
      env: CartPole-v0
      run: DQN
      stop:
          episode_reward_mean: 150
          timesteps_total: 300000
      config:
          # Works for both torch and tf.
          framework: tf
          exploration_config:
              type: ParameterNoise
              random_timesteps: 10000
              initial_stddev: 1.0
          batch_mode: complete_episodes
          lr: 0.0008
          num_workers: 0
          model:
              fcnet_hiddens: [32, 32]
              fcnet_activation: tanh
  