From b25606b04bc7694eaba0e97fb9aa7961e0c03774 Mon Sep 17 00:00:00 2001 From: Rajas Bansal Date: Thu, 15 Jun 2023 17:46:59 -0700 Subject: [PATCH] Fix the labeled stats shown for labeled examples (#334) * Fix the labeled stats shown for labeled examples * Changed the text --------- Co-authored-by: Rajas Bansal --- src/autolabel/labeler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/autolabel/labeler.py b/src/autolabel/labeler.py index 1d5f7785..63799549 100644 --- a/src/autolabel/labeler.py +++ b/src/autolabel/labeler.py @@ -127,7 +127,7 @@ def run( for current_index in track_with_stats( indices, postfix_dict, - total=len(inputs), + total=len(inputs) - current_index, advance=self.CHUNK_SIZE, console=console, ): @@ -388,7 +388,7 @@ def handle_existing_task_run( else: print(f"Metric: {m.name}: {m.value}") print_table(table, console=console, default_style=METRIC_TABLE_STYLE) - pprint(f"{len(llm_labels)} examples have been labeled so far.") + pprint(f"{task_run.current_index} examples labeled so far.") if len(llm_labels) > 0: console.rule("Last Annotated Example") pprint("[bold blue]Prompt[/bold blue]: ", end="")