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 <[email protected]>
refuel-ai · Jun 16, 2023 · b25606b · b25606b
1 parent ed31cc5
commit b25606b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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="")