[Misc] Add network timeout to eval dataset downloads (#21873)

Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-01 13:16:14 -07:00
parent a19ef3a615
commit 0138708576
6 changed files with 18 additions and 6 deletions
@@ -32,7 +32,10 @@ class GPQAEval(Eval):
        num_threads: int,
        n_repeats: int = 1,
    ):
-        df = pandas.read_csv(filename)
+        if "://" in filename:
+            df = pandas.read_csv(filename, storage_options={"timeout": 30})
+        else:
+            df = pandas.read_csv(filename)
        examples = [row.to_dict() for _, row in df.iterrows()]
        rng = random.Random(0)
        if num_examples: