diff --git a/benchmark/multi_chain_reasoning/bench_other.py b/benchmark/multi_chain_reasoning/bench_other.py index f361496ad..d82375f90 100644 --- a/benchmark/multi_chain_reasoning/bench_other.py +++ b/benchmark/multi_chain_reasoning/bench_other.py @@ -89,7 +89,7 @@ async def multi_chain_gsm8k_async(question, num_chains, call_generate): def main(args): - lines = read_jsonl(args.data_path) + lines = list(read_jsonl(args.data_path)) # Construct prompts k = args.num_shot diff --git a/benchmark/multi_chain_reasoning/bench_sglang.py b/benchmark/multi_chain_reasoning/bench_sglang.py index 1d3129db2..90c011e1a 100644 --- a/benchmark/multi_chain_reasoning/bench_sglang.py +++ b/benchmark/multi_chain_reasoning/bench_sglang.py @@ -37,7 +37,7 @@ prompt_lib = [ def main(args): - lines = read_jsonl(args.data_path) + lines = list(read_jsonl(args.data_path)) # Construct prompts # k = args.num_shot