Skip to content

Commit 13f07c9

Browse files
committed
feat(codegen): add the progress bar for openai API
1 parent 9d7af54 commit 13f07c9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

bigcodebench/provider/openai.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import os
22
from typing import List
3-
3+
from tqdm import tqdm
44
import openai
55

66
from bigcodebench.gen.util.openai_request import make_auto_request
@@ -38,7 +38,7 @@ def _codegen_api_batch(self, messages: List[str], num_samples: int) -> List[str]
3838
)
3939

4040
all_outputs = []
41-
for message in messages:
41+
for message in tqdm(messages):
4242
ret = make_auto_request(
4343
client,
4444
message=message,

0 commit comments

Comments
 (0)