Improve type annotation and styles (#2926)

2025-01-16 12:51:11 -08:00
parent a883f0790d
commit bc6915e3b9
7 changed files with 78 additions and 26 deletions
--- a/benchmark/tree_of_thought_deep/bench_sglang.py
+++ b/benchmark/tree_of_thought_deep/bench_sglang.py
@@ -103,6 +103,7 @@ def tree_search(s, question, num_branches):

 def main(args):
    lines = read_jsonl(args.data_path)
+    lines = list(lines)

    # Construct prompts
    num_branches = 2