diff --git a/python/sglang/test/simple_eval_common.py b/python/sglang/test/simple_eval_common.py index e3b96ef81363..c667098b7b06 100644 --- a/python/sglang/test/simple_eval_common.py +++ b/python/sglang/test/simple_eval_common.py @@ -519,7 +519,7 @@ def make_report_from_example_htmls(htmls: List[str]): def download_dataset(path, url): print(f"Downloading dataset {path} from {url}") try: - response = requests.get(url, stream=True, timeout=120) + response = requests.get(url, stream=True, timeout=30) response.raise_for_status() total_size = int(response.headers.get("content-length", 0)) diff --git a/python/sglang/test/simple_eval_gpqa.py b/python/sglang/test/simple_eval_gpqa.py index b39366ef5df8..3ad37a604432 100644 --- a/python/sglang/test/simple_eval_gpqa.py +++ b/python/sglang/test/simple_eval_gpqa.py @@ -32,7 +32,10 @@ def __init__( 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: diff --git a/python/sglang/test/simple_eval_math.py b/python/sglang/test/simple_eval_math.py index 37d4b120b930..6cb5658bbdbb 100644 --- a/python/sglang/test/simple_eval_math.py +++ b/python/sglang/test/simple_eval_math.py @@ -40,7 +40,10 @@ def __init__( num_examples: Optional[int], num_threads: int, ): - 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()] if num_examples: examples = random.Random(0).sample(examples, num_examples) diff --git a/python/sglang/test/simple_eval_mmlu.py b/python/sglang/test/simple_eval_mmlu.py index a68dbb935a21..281da9e802f0 100644 --- a/python/sglang/test/simple_eval_mmlu.py +++ b/python/sglang/test/simple_eval_mmlu.py @@ -86,7 +86,10 @@ class MMLUEval(Eval): def __init__(self, filename: str, num_examples: Optional[int], num_threads: int): - 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()] if num_examples: examples = random.Random(0).sample(examples, num_examples) diff --git a/sgl-model-gateway/e2e_test/infra/simple_eval_common.py b/sgl-model-gateway/e2e_test/infra/simple_eval_common.py index 92e72937d9e9..7be4358172c7 100644 --- a/sgl-model-gateway/e2e_test/infra/simple_eval_common.py +++ b/sgl-model-gateway/e2e_test/infra/simple_eval_common.py @@ -457,7 +457,7 @@ def download_dataset(path: str, url: str) -> None: """Download a dataset from URL to path.""" logger.info("Downloading dataset from %s", url) try: - response = requests.get(url, stream=True) + response = requests.get(url, stream=True, timeout=30) response.raise_for_status() total_size = int(response.headers.get("content-length", 0)) diff --git a/sgl-model-gateway/e2e_test/infra/simple_eval_mmlu.py b/sgl-model-gateway/e2e_test/infra/simple_eval_mmlu.py index 1083e56ca60c..a83ed1d2eaaf 100644 --- a/sgl-model-gateway/e2e_test/infra/simple_eval_mmlu.py +++ b/sgl-model-gateway/e2e_test/infra/simple_eval_mmlu.py @@ -93,7 +93,10 @@ class MMLUEval(Eval): """MMLU benchmark evaluation.""" def __init__(self, filename: str, num_examples: int | None, num_threads: int): - 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()] if num_examples: examples = random.Random(0).sample(examples, num_examples)