Run pyformat on submission related scripts

mlcommons · Aug 26, 2022 · 1749731 · 1749731
1 parent 99e65fe
commit 1749731
Show file tree

Hide file tree

Showing 2 changed files with 757 additions and 313 deletions.
diff --git a/tools/submission/generate-final-report.py b/tools/submission/generate-final-report.py
@@ -92,9 +92,8 @@ def main():
   ]
   columns_order = [['Result'],
                    [
-                       'resnet', 'retinanet', '3d-unet-99',
-                       '3d-unet-99.9', 'rnnt', 'bert-99', 'bert-99.9',
-                       'dlrm-99', 'dlrm-99.9'
+                       'resnet', 'retinanet', '3d-unet-99', '3d-unet-99.9',
+                       'rnnt', 'bert-99', 'bert-99.9', 'dlrm-99', 'dlrm-99.9'
                    ], ['SingleStream', 'MultiStream', 'Server', 'Offline'],
                    [
                        'Latency (ms)',
@@ -105,28 +104,28 @@ def main():
                    ]]
 
   filter_scenarios = {
-    'datacenter': {
-      'resnet': ['Server', 'Offline'],
-      'retinanet': ['Server', 'Offline'],
-      'rnnt': ['Server', 'Offline'],
-      'bert-99': ['Server', 'Offline'],
-      'bert-99.9': ['Server', 'Offline'],
-      'dlrm-99': ['Server', 'Offline'],
-      'dlrm-99.9': ['Server', 'Offline'],
-      '3d-unet-99': ['Offline'],
-      '3d-unet-99.9': ['Offline'],
-    },
-    'edge': {
-      'resnet': ['SingleStream', 'MultiStream', 'Offline'],
-      'retinanet': ['SingleStream', 'MultiStream', 'Offline'],
-      'rnnt': ['SingleStream', 'Offline'],
-      'bert-99': ['SingleStream', 'Offline'],
-      'bert-99.9': [],
-      'dlrm-99': [],
-      'dlrm-99.9': [],
-      '3d-unet-99': ['SingleStream', 'Offline'],
-      '3d-unet-99.9': ['SingleStream', 'Offline'],
-    }
+      'datacenter': {
+          'resnet': ['Server', 'Offline'],
+          'retinanet': ['Server', 'Offline'],
+          'rnnt': ['Server', 'Offline'],
+          'bert-99': ['Server', 'Offline'],
+          'bert-99.9': ['Server', 'Offline'],
+          'dlrm-99': ['Server', 'Offline'],
+          'dlrm-99.9': ['Server', 'Offline'],
+          '3d-unet-99': ['Offline'],
+          '3d-unet-99.9': ['Offline'],
+      },
+      'edge': {
+          'resnet': ['SingleStream', 'MultiStream', 'Offline'],
+          'retinanet': ['SingleStream', 'MultiStream', 'Offline'],
+          'rnnt': ['SingleStream', 'Offline'],
+          'bert-99': ['SingleStream', 'Offline'],
+          'bert-99.9': [],
+          'dlrm-99': [],
+          'dlrm-99.9': [],
+          '3d-unet-99': ['SingleStream', 'Offline'],
+          '3d-unet-99.9': ['SingleStream', 'Offline'],
+      }
   }
 
   def MakeWorksheet(df, index, filter_dict, sheet_name):
@@ -158,7 +157,8 @@ def Apply(f, *args):
     return lambda x: f(x, *args)
 
   def FilterScenario(x, suite):
-    return x.apply(lambda y: y['Scenario'] in filter_scenarios[suite][y['Model']], axis = 1)
+    return x.apply(
+        lambda y: y['Scenario'] in filter_scenarios[suite][y['Model']], axis=1)
 
   def MakeUniqueID(x):
     key_list = ['Suite', 'Category', 'Submitter', 'Platform']