Fix duplicate progress output. (ray-project#6497)

jsuit · Dec 16, 2019 · e38b25e · e38b25e
1 parent 5719a05
commit e38b25e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/python/ray/tune/progress_reporter.py b/python/ray/tune/progress_reporter.py
@@ -52,7 +52,8 @@ def report(self, trial_runner):
         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 @@ def report(self, trial_runner):
         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
@@ -186,6 +186,10 @@ def __init__(self,
                 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.