[tune] Prevent deletion of checkpoint from user-initiated resto… (#7501)

* Fix restore bug

* Add test

* Lint

* Indent
This commit is contained in:
Ujval Misra
2020-03-09 15:53:10 -07:00
committed by GitHub
parent 08d4cb3822
commit 023d4c02a9
4 changed files with 34 additions and 23 deletions
@@ -239,7 +239,6 @@ class TrialRunnerTest2(unittest.TestCase):
runner.step() # Start trial
self.assertEqual(trials[0].status, Trial.RUNNING)
self.assertEqual(ray.get(trials[0].runner.set_info.remote(1)), 1)
# checkpoint = runner.trial_executor.save(trials[0])
runner.step() # Process result, dispatch save
runner.step() # Process save
runner.trial_executor.stop_trial(trials[0])