[Benchmark] Remove re-exports from bench_serving.py (#19130)

scripts/playground/bench_speculative.py

@@ -19,12 +19,8 @@ import numpy as np
 import requests
 from transformers import AutoTokenizer
 
-from sglang.bench_serving import (
-    DatasetRow,
-    benchmark,
-    sample_mmmu_requests,
-    set_global_args,
-)
+from sglang.bench_serving import benchmark, set_global_args
+from sglang.benchmark.datasets import DatasetRow, sample_mmmu_requests
 from sglang.srt.server_args import ServerArgs
 from sglang.test.test_utils import (
     DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH,

scripts/playground/replay_request_dump.py

@@ -18,7 +18,7 @@ from datetime import datetime
 
 import requests
 
-from sglang.bench_serving import set_ulimit
+from sglang.benchmark.utils import set_ulimit
 from sglang.utils import get_exception_traceback