From 87ed20738ee90d2639518d91192b06d712346404 Mon Sep 17 00:00:00 2001 From: Antoni Baum Date: Wed, 26 Aug 2020 07:33:17 +0200 Subject: [PATCH] [tune] Add on_pause, on_unpause to ConcurrencyLimiter (#10320) Co-authored-by: Richard Liaw --- python/ray/tune/suggest/suggestion.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/python/ray/tune/suggest/suggestion.py b/python/ray/tune/suggest/suggestion.py index b18680e12..e3cee6dfb 100644 --- a/python/ray/tune/suggest/suggestion.py +++ b/python/ray/tune/suggest/suggestion.py @@ -304,3 +304,9 @@ class ConcurrencyLimiter(Searcher): def set_state(self, state): self.__dict__.update(state) + + def on_pause(self, trial_id): + self.searcher.on_pause(trial_id) + + def on_unpause(self, trial_id): + self.searcher.on_unpause(trial_id)