Fix spreadsheet coloring. Fix power unit.

mlcommons · Mar 31, 2022 · 33894a1 · 33894a1
1 parent ca18f5c
commit 33894a1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/tools/submission/generate-final-report.py b/tools/submission/generate-final-report.py
@@ -68,6 +68,7 @@ def main():
   indices['closed'] = [
       'ID',
       'Unique ID (e.g. for Audit)',
+      'ColorKey',
       'Submitter',
       'Availability',
       'System',
@@ -99,7 +100,6 @@ def main():
                        'Samples/s',
                        'Queries/s',
                        'Joules',
-                       'Joules/Stream',
                        'Watts',
                    ]]
 
@@ -168,6 +168,8 @@ def MakeUniqueID(x):
     return '/'.join(x[key_list])
 
   df['Unique ID (e.g. for Audit)'] = df.apply(MakeUniqueID, axis=1)
+  df['ColorKey'] = df.apply(
+      lambda x: ''.join(x[['Availability', 'Submitter']]), axis=1)
   df.sort_values(
       by=[
           'Category', 'Availability', 'Submitter', 'Unique ID (e.g. for Audit)'
@@ -209,6 +211,7 @@ def MakeUniqueID(x):
   bg_format = writer.book.add_format({'bg_color': '#efefef'})
   for ws in writer.book.worksheets():
     ws.set_column(1, 1, None, None, {'hidden': 1})
+    ws.set_column(2, 2, None, None, {'hidden': 1})
     ws.set_column(len(indices['closed']), 100, None, score_format)
     ws.conditional_format(
         2 + len(columns), 0, 200, 100, {

diff --git a/tools/submission/submission-checker.py b/tools/submission/submission-checker.py
@@ -1243,7 +1243,7 @@ def log_result(submitter,
     }
     power_unit_dict = {
         "SingleStream": "Joules",
-        "MultiStream": "Joules/Stream",
+        "MultiStream": "Joules",
         "Offline": "Watts",
         "Server": "Watts",
     }