Skip to content

Commit

Permalink
Merge pull request #817 from mlcommons/formatting_fixes
Browse files Browse the repository at this point in the history
Formatting fixes
  • Loading branch information
priyakasimbeg authored Nov 21, 2024
2 parents adc5ea9 + bb2b361 commit 3c61cc4
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 2 deletions.
2 changes: 1 addition & 1 deletion scoring/performance_profile.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
3 changes: 2 additions & 1 deletion scoring/score_submissions.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 3c61cc4

Please sign in to comment.