[tune] Add on_trial_result to ConcurrencyLimiter (#18766)

ray-project · Sep 21, 2021 · f4666f3 · f4666f3
1 parent ca3fabc
commit f4666f3
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/ray/tune/suggest/suggestion.py b/python/ray/tune/suggest/suggestion.py
@@ -424,6 +424,9 @@ def on_trial_complete(self,
             self.live_trials.remove(trial_id)
             self.num_unfinished_live_trials -= 1
 
+    def on_trial_result(self, trial_id: str, result: Dict) -> None:
+        self.searcher.on_trial_result(trial_id, result)
+
     def add_evaluated_point(self,
                             parameters: Dict,
                             value: float,