wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-27 23:08:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] fix flaky callback test (#11102)

Browse Source
This commit is contained in:
Kai Fricke
2020-09-29 17:44:32 +01:00
committed by GitHub
parent 1db83764bf
commit 10015e60fb
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/tests/test_trial_runner_callbacks.py
+3
View File
@@ -89,6 +89,7 @@ class TrialRunnerCallbacks(unittest.TestCase):
for t in trials:
self.trial_runner.add_trial(t)
self.executor.next_trial = trials[0]
self.trial_runner.step()
# Trial 1 has been started
@@ -103,6 +104,7 @@ class TrialRunnerCallbacks(unittest.TestCase):
"trial_complete", "trial_fail"
]))
self.executor.next_trial = trials[1]
self.trial_runner.step()
# Iteration not increased yet
@@ -120,6 +122,7 @@ class TrialRunnerCallbacks(unittest.TestCase):
{TRAINING_ITERATION: 0})
# Let the first trial save a checkpoint
self.executor.next_trial = trials[0]
trials[0].saving_to = cp
self.trial_runner.step()
self.assertEqual(self.callback.state["trial_save"]["iteration"], 2)