Fixes issue where benchmark names were lost when converting a Pandas DataFrame to a list of dictionaries. Now includes benchmark names as a key within each dictionary.

PiperOrigin-RevId: 721641743
Change-Id: I0bbc012ce39a936c14ef6cfe507aeb83e136ebf9

Author: liyuying0000

fleetbench/parallel/parallel_bench_lib.py

@@ -462,8 +462,7 @@ def SaveBenchmarkResults(self, df: pd.DataFrame) -> None:
       df = df.rename(columns={"Benchmark": "name"})
       df = df.rename(columns={"WallTimes": "real_time"})
       df = df.rename(columns={"CPUTimes": "cpu_time"})
-
-      data = df.to_dict(orient="records")
+      data = df.reset_index().to_dict(orient="records")
 
       with open(file_name, "w") as json_file:
         json.dump(
@@ -521,7 +520,7 @@ def GeneratePerfCounterDataFrame(
 
     perf_counters_results = (
         perf_counters_results.groupby("Benchmark").agg(**aggregations).round(3)
-    ).reset_index()
+    )
     return perf_counters_results
 
   def ConvertToDataFrame(self) -> pd.DataFrame:

fleetbench/parallel/parallel_bench_lib_test.py

@@ -285,7 +285,7 @@ def test_generate_perf_counter_dataframe(self):
     expected_df = pd.DataFrame([
         {"Benchmark": "test_benchmark1", "instructions": 130.0, "cycles": 3.0},
         {"Benchmark": "test_benchmark2", "instructions": 200.0, "cycles": 2.0},
-    ])
+    ]).set_index("Benchmark")
     pd.testing.assert_frame_equal(df, expected_df)
 
   def test_generate_benchmark_report(self):
@@ -335,8 +335,8 @@ def test_save_benchmark_results(self):
     self.assertEqual(
         data,
         [
-            {"name": "BM_Test1", "Duration": 1, "cpu_time": 1},
-            {"name": "BM_Test2", "Duration": 1, "cpu_time": 2},
+            {"index": 0, "name": "BM_Test1", "Duration": 1, "cpu_time": 1},
+            {"index": 1, "name": "BM_Test2", "Duration": 1, "cpu_time": 2},
         ],
     )