Skip to content

Commit bc612d1

Browse files
fix: update batch size calculation to use NUM_WORKERS_A2VYBG
1 parent 80d2a0c commit bc612d1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

run_ac.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,9 @@ async def process_llm_record_batch(record_dict_batch: list):
149149
calculated_attribute_by_record_id[record_dict["id"]] = attr_value
150150

151151
async def process_async_llm_calls(record_dict_list):
152-
batch_size = len(record_dict_list) // int(attribute_calculators.NUM_WORKERS)
152+
batch_size = len(record_dict_list) // int(
153+
attribute_calculators.NUM_WORKERS_A2VYBG
154+
)
153155
record_dict_batches = [
154156
record_dict_list[i : i + batch_size]
155157
for i in range(0, len(record_dict_list), batch_size)

0 commit comments

Comments
 (0)