From e38b25edfb49ffd90107a0f6aca09a0e4feb0106 Mon Sep 17 00:00:00 2001
From: Ujval Misra <misraujval@gmail.com>
Date: Sun, 15 Dec 2019 21:53:24 -0800
Subject: [PATCH] Fix duplicate progress output. (#6497)

---
 python/ray/tune/progress_reporter.py | 6 ++++--
 python/ray/tune/trial_runner.py      | 4 ++++
 2 files changed, 8 insertions(+), 2 deletions(-)

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.