[tune][minor] Fixes (#1383)

This commit is contained in:
Richard Liaw
2018-01-11 18:14:20 -08:00
committed by GitHub
parent 1290072764
commit d4592382a4
3 changed files with 45 additions and 22 deletions
+9 -1
View File
@@ -11,7 +11,7 @@ import os
from collections import namedtuple
from ray.tune import TuneError
from ray.tune.logger import NoopLogger, UnifiedLogger
from ray.tune.result import TrainingResult, DEFAULT_RESULTS_DIR
from ray.tune.result import TrainingResult, DEFAULT_RESULTS_DIR, pretty_print
from ray.tune.registry import _default_registry, get_registry, TRAINABLE_CLASS
@@ -285,6 +285,14 @@ class Trial(object):
print("Error restoring runner:", traceback.format_exc())
self.status = Trial.ERROR
def update_last_result(self, result, terminate=False):
if terminate:
result = result._replace(done=True)
print("TrainingResult for {}:".format(self))
print(" {}".format(pretty_print(result).replace("\n", "\n ")))
self.last_result = result
self.result_logger.on_result(self.last_result)
def _setup_runner(self):
self.status = Trial.RUNNING
trainable_cls = get_registry().get(