[tune] Allow fetching pinned objects from trainable functions (#1895)

* updates

* lint

* Update util.py

* Update function_runner.py

* updates
This commit is contained in:
Eric Liang
2018-04-16 15:54:38 -07:00
committed by GitHub
parent ddfc875149
commit ed8c0f1a38
3 changed files with 59 additions and 0 deletions
+2
View File
@@ -9,6 +9,7 @@ import traceback
from ray.tune import TuneError
from ray.tune.trainable import Trainable
from ray.tune.result import TrainingResult
from ray.tune.util import _serve_get_pin_requests
class StatusReporter(object):
@@ -108,6 +109,7 @@ class FunctionRunner(Trainable):
self._default_config["script_min_iter_time_s"]))
result = self._status_reporter._get_and_clear_status()
while result is None:
_serve_get_pin_requests()
time.sleep(1)
result = self._status_reporter._get_and_clear_status()
if result.timesteps_total is None: