From 33894a19c4af6207f7cfdda75f84570f04836de5 Mon Sep 17 00:00:00 2001
From: Tom Jablin <tjablin@google.com>
Date: Thu, 31 Mar 2022 11:45:16 -0700
Subject: [PATCH] Fix spreadsheet coloring. Fix power unit.

---
 tools/submission/generate-final-report.py | 5 ++++-
 tools/submission/submission-checker.py    | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/tools/submission/generate-final-report.py b/tools/submission/generate-final-report.py
index beb7c4e7a..88743b25f 100644
--- 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
index ba1a262e2..308262f97 100755
--- 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",
     }