From 7bb88b2edcdc2375a7f4218dca52af9ae68c96b3 Mon Sep 17 00:00:00 2001 From: "rshaw@neuralmagic.com" Date: Wed, 26 Mar 2025 01:44:11 +0000 Subject: [PATCH] updated Signed-off-by: rshaw@neuralmagic.com --- benchmarks/convert_to_csv.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/benchmarks/convert_to_csv.py b/benchmarks/convert_to_csv.py index c845164fc3088..587727c8aaccd 100644 --- a/benchmarks/convert_to_csv.py +++ b/benchmarks/convert_to_csv.py @@ -1,7 +1,11 @@ +from argparse import ArgumentParser import pandas as pd -df_sgl = pd.read_json("sgl-results.json", lines=True) -df_sgl.to_csv("sgl-results.csv") +parser = ArgumentParser() +parser.add_argument("--input-path", type=str, required=True) +parser.add_argument("--output-path", type=str, required=True) -df_vllm = pd.read_json("vllm-results.json", lines=True) -df_vllm.to_csv("vllm-results.csv") +if __name__ == "__main__": + args = parser.parse_args() + df = pd.read_json(args.input_path, lines=True) + df.to_csv(args.output_path)