[tune] Allow nested values in trial runner (#5346)

This commit is contained in:
Richard Liaw
2019-08-06 14:36:17 -07:00
committed by GitHub
parent e8d9cfc1f1
commit 094ec7adbc
10 changed files with 163 additions and 106 deletions
+2
View File
@@ -134,6 +134,7 @@ class _MockSuggestionAlgorithm(SuggestionAlgorithm):
self.live_trials = {}
self.counter = {"result": 0, "complete": 0}
self.stall = False
self.results = []
super(_MockSuggestionAlgorithm, self).__init__(**kwargs)
def _suggest(self, trial_id):
@@ -144,6 +145,7 @@ class _MockSuggestionAlgorithm(SuggestionAlgorithm):
def on_trial_result(self, trial_id, result):
self.counter["result"] += 1
self.results += [result]
def on_trial_complete(self, trial_id, **kwargs):
self.counter["complete"] += 1