Skip to content

Commit

Permalink
Support custom repo branch and owner for final report generation
Browse files Browse the repository at this point in the history
  • Loading branch information
arjunsuresh committed Oct 11, 2024
1 parent 1dafdfc commit fa0eeca
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions tools/submission/generate_final_report.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,10 @@ def get_args():
"""Parse commandline."""
parser = argparse.ArgumentParser()
parser.add_argument('--input', required=True, help='results csv from checker')
parser.add_argument('--version', default='4.0', help='mlperf version')
parser.add_argument('--repository', default='submissions_inference_4.0', help='mlperf repository')
parser.add_argument('--version', default='4.1', help='mlperf version')
parser.add_argument('--repository', default='submissions_inference_4.1', help='mlperf repository')
parser.add_argument('--repository-owner', default='mlcommons', help='mlperf repository owner in GitHub')
parser.add_argument('--repository-branch', default='main', help='mlperf repository branch')
args = parser.parse_args()
return args

Expand Down Expand Up @@ -59,7 +61,7 @@ def main():
df['p#'] = df.apply(lambda x: int(x['host_processors_per_node']), axis=1)

# details url
base_url = f'https://github.com/mlcommons/{args.repository}/tree/main'
base_url = f'https://github.com/{args.repository_owner}/{args.repository}/tree/{args.repository_branch}'
df['Details'] = df.apply(
lambda x: '=HYPERLINK("{}","details")'.format('/'.join(
[base_url, x['Category'], x['Submitter'], 'results', x['Platform']])),
Expand Down

0 comments on commit fa0eeca

Please sign in to comment.