diff --git a/scripts/benchmark_streaming_chunk_overhead.py b/scripts/benchmark_streaming_chunk_overhead.py index a1ad41ac77d..11fbea6a6a3 100644 --- a/scripts/benchmark_streaming_chunk_overhead.py +++ b/scripts/benchmark_streaming_chunk_overhead.py @@ -169,11 +169,13 @@ def _make_wrapper( custom_llm_provider=provider, ) + @dataclass class TimingSample: wall_s: float cpu_s: float + def drive_sync( provider_key: str, chunks_per_stream: int, n_streams: int ) -> TimingSample: