fixed :bug in summary_df

mlcommons · Oct 11, 2024 · 076c1a7 · 076c1a7
1 parent 223b90b
commit 076c1a7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/scoring/score_submissions.py b/scoring/score_submissions.py
@@ -96,8 +96,7 @@ def get_summary_df(workload, workload_df, include_test_split=False):
       axis=1)
   summary_df['time to target on val (s)'] = workload_df.apply(
       lambda x: x['accumulated_submission_time'][int(x[
-          'index to target on val'])]
-      if x['val target reached'] else np.inf,
+          'index to target on val'])] if x['val target reached'] else np.inf,
       axis=1)
 
   # test metrics