Skip to content

Commit 598c931

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: Remove redundant progress bar update commands within runnable inference
PiperOrigin-RevId: 700151822
1 parent d7dff72 commit 598c931

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

vertexai/preview/evaluation/_evaluation.py

-3
Original file line numberDiff line numberDiff line change
@@ -574,7 +574,6 @@ def _run_runnable_inference(
574574
tasks.append(task)
575575
for task in tasks:
576576
response_dict, latency, failure = task.result()
577-
pbar.update(1)
578577
responses.append(response_dict["output"])
579578
latency_list.append(latency)
580579
failure_list.append(failure)
@@ -602,7 +601,6 @@ def _run_runnable_inference(
602601
"the pre-existing `response` column provided "
603602
"in the evaluation dataset is not used."
604603
)
605-
pbar.update(1)
606604
elif callable(runnable):
607605
with tqdm(total=len(evaluation_run_config.dataset)) as pbar:
608606
with futures.ThreadPoolExecutor(
@@ -645,7 +643,6 @@ def _run_runnable_inference(
645643
" column provided in the evaluation dataset"
646644
" is not used."
647645
)
648-
pbar.update(1)
649646
else:
650647
raise ValueError(f"Unsupported runnable type: {type(runnable)}")
651648

0 commit comments

Comments
 (0)