From bb2b361c3bda4085c45134de79a4ebf25f1f64f3 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Thu, 21 Nov 2024 23:38:41 +0000 Subject: [PATCH] formatting --- scoring/performance_profile.py | 2 +- scoring/score_submissions.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/scoring/performance_profile.py b/scoring/performance_profile.py index 0d5ca9770..f4f2d5679 100644 --- a/scoring/performance_profile.py +++ b/scoring/performance_profile.py @@ -321,7 +321,7 @@ def compute_performance_profiles(submissions, df = df[BASE_WORKLOADS + HELDOUT_WORKLOADS] # Sort workloads alphabetically (for better display) df = df.reindex(sorted(df.columns), axis=1) - + # Save time to target dataframe df.to_csv(os.path.join(output_dir, 'time_to_targets.csv')) # For each held-out workload set to inf if the base workload is inf or nan diff --git a/scoring/score_submissions.py b/scoring/score_submissions.py index 59295b686..8cc06b15f 100644 --- a/scoring/score_submissions.py +++ b/scoring/score_submissions.py @@ -211,7 +211,8 @@ def main(_): verbosity=0, self_tuning_ruleset=FLAGS.self_tuning_ruleset, strict=FLAGS.strict, - output_dir=FLAGS.output_dir,) + output_dir=FLAGS.output_dir, + ) if not os.path.exists(FLAGS.output_dir): os.mkdir(FLAGS.output_dir) performance_profile.plot_performance_profiles(