hunk.code_window:         self.addCleanup(lambda: shutil.rmtree(tmpdir))
           def train(config):
               module = _MockModule(10, 20)
               trainer = pl.Trainer(
                   max_epochs=1,
                   callbacks=[
                       TuneReportCheckpointCallback(
  
hunk.labels: keep keep keep replace keep keep keep keep
hunk.after_edit:             module = _MockModule(10., 20.)
  
hunk.file_path: python/ray/tune/tests/test_integration_pytorch_lightning.py
hunk.type: replace
file: /* CSS for binder integration */
  
  div.binder-badge {
    margin: 1em auto;
    vertical-align: middle;
  }
  