From c24f515d60ae155727a3be8c576a3149a8e08693 Mon Sep 17 00:00:00 2001 From: harry-humanloop <91058617+harry-humanloop@users.noreply.github.com> Date: Tue, 19 Nov 2024 19:20:50 +0000 Subject: [PATCH] fix eval run progress printing by clearing line fix the leftover "0s" after done [########################################] 2/2 (100.00%) | DONE0s --- src/humanloop/eval_utils/run.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/humanloop/eval_utils/run.py b/src/humanloop/eval_utils/run.py index 5fd25e7d..b886947d 100644 --- a/src/humanloop/eval_utils/run.py +++ b/src/humanloop/eval_utils/run.py @@ -196,7 +196,7 @@ def increment(self): time_per_item = elapsed_time / self._progress if self._progress > 0 else 0 eta = (self._total - self._progress) * time_per_item - progress_display = f"\r[{bar}] {self._progress}/{self._total}" + progress_display = f"[{bar}] {self._progress}/{self._total}" progress_display += f" ({percentage:.2f}%)" if self._progress < self._total: @@ -204,6 +204,8 @@ def increment(self): else: progress_display += " | DONE" + sys.stderr.write("\r") # Move the cursor to the beginning of the line + sys.stderr.write("\033[K") # Clear the line from the cursor to the end sys.stderr.write(progress_display) if self._progress >= self._total: