diff --git a/python/ray/tune/progress_reporter.py b/python/ray/tune/progress_reporter.py index 832473993..3d6d86b55 100644 --- a/python/ray/tune/progress_reporter.py +++ b/python/ray/tune/progress_reporter.py @@ -52,7 +52,8 @@ class JupyterNotebookReporter(ProgressReporter): messages = [ "== Status ==", memory_debug_str(), - trial_runner.debug_string(delim=delim), + trial_runner.scheduler_alg.debug_string(), + trial_runner.trial_executor.debug_string(), trial_progress_str(trial_runner.get_trials(), fmt="html"), trial_errors_str(trial_runner.get_trials(), fmt="html"), ] @@ -68,7 +69,8 @@ class CLIReporter(ProgressReporter): messages = [ "== Status ==", memory_debug_str(), - trial_runner.debug_string(), + trial_runner.scheduler_alg.debug_string(), + trial_runner.trial_executor.debug_string(), trial_progress_str(trial_runner.get_trials()), trial_errors_str(trial_runner.get_trials()), ] diff --git a/python/ray/tune/trial_runner.py b/python/ray/tune/trial_runner.py index 1f2bc7ca4..80bbcbe57 100644 --- a/python/ray/tune/trial_runner.py +++ b/python/ray/tune/trial_runner.py @@ -186,6 +186,10 @@ class TrialRunner(object): self._local_checkpoint_dir, TrialRunner.CKPT_FILE_TMPL.format(self._session_str)) + @property + def scheduler_alg(self): + return self._scheduler_alg + def _validate_resume(self, resume_type): """Checks whether to resume experiment.