Skip to content

Commit 2816cf6

Browse files
committed
chore: pr review
1 parent 9d9897e commit 2816cf6

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

run_ac.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -180,16 +180,15 @@ def make_batches(
180180
async def process_async_llm_calls_a2vybg(
181181
record_dict_list: List[Dict[str, Any]]
182182
) -> None:
183-
184183
batch_size = max(amount_a2vybg // int(attribute_calculators.NUM_WORKERS_A2VYBG), 1)
185184
tasks = [
186185
process_llm_record_batch_a2vybg(batch)
187186
for batch in make_batches(record_dict_list, size=batch_size)
188187
]
189188
await asyncio.gather(*tasks)
189+
send_cache_to_object_storage_a2vybg()
190190
if check_abort_status_a2vybg():
191191
raise ValueError("Encountered error during LLM processing.")
192-
send_cache_to_object_storage_a2vybg()
193192

194193

195194
if __name__ == "__main__":

0 commit comments

Comments
 (0)