Signed-off-by: rshaw@neuralmagic.com <robertgshaw2@gmail.com>
This commit is contained in:
rshaw@neuralmagic.com 2025-03-26 01:44:11 +00:00
parent ae4f3e2aeb
commit 7bb88b2edc

View File

@ -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)