save dataframe

Edenzzzz · Edenzzzz · commit e3fc0053933b · 2025-07-29T23:22:36.000Z
diff --git a/benchmarks/bench_batch_attention.py b/benchmarks/bench_batch_attention.py
@@ -196,6 +196,7 @@ def main() -> None:
         ],
     )
     print(df.to_markdown(index=False, floatfmt=".2f"))
+    df.to_csv("bench_batch_attention.csv", index=False)
 
 
 if __name__ == "__main__":

Original file line number	Diff line number	Diff line change
`@@ -196,6 +196,7 @@ def main() -> None:`
`196`	`196`	`],`
`197`	`197`	`)`
`198`	`198`	`print(df.to_markdown(index=False, floatfmt=".2f"))`
	`199`	`+ df.to_csv("bench_batch_attention.csv", index=False)`
`199`	`200`
`200`	`201`
`201`	`202`	`if __name__ == "__main__":`