formatting

mlcommons · Jan 27, 2024 · d6e2a36 · d6e2a36
1 parent 84c87b9
commit d6e2a36
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/scoring/performance_profile.py b/scoring/performance_profile.py
@@ -190,16 +190,16 @@ def get_times_for_submission(submission,
     if num_studies != NUM_STUDIES:
       if strict:
         raise ValueError(f'Expecting {NUM_STUDIES} trials for workload '
-                          f'{workload} but found {num_studies} trials.')
+                         f'{workload} but found {num_studies} trials.')
       else:
         logging.warning(f'Expecting {NUM_STUDIES} trials for workload '
-                          f'{workload} but found {num_studies} trials.')
+                        f'{workload} but found {num_studies} trials.')
     for study, group in group.groupby('study'):
       num_trials = len(group)
       if num_trials != NUM_TRIALS and not self_tuning_ruleset:
         if strict:
           raise ValueError(f'Expecting {NUM_TRIALS} trials for workload '
-                          f'{workload} but found {num_trials} trials.')
+                           f'{workload} but found {num_trials} trials.')
         else:
           logging.warning(f'Expecting {NUM_TRIALS} trials for workload '
                           f'{workload} but found {num_trials} trials.')
@@ -211,7 +211,6 @@ def get_times_for_submission(submission,
       else:
         time_val = float('inf')
       time_vals_per_study.append(time_val)
-
 
     workloads.append({
         'submission': submission_name,

diff --git a/scoring/scoring_utils.py b/scoring/scoring_utils.py
@@ -179,7 +179,8 @@ def get_experiment_df(experiment_dir):
         }
         logging.info(os.path.join(experiment_dir, study_dir, workload))
         trial_dirs = [
-            t for t in os.listdir(os.path.join(experiment_dir, study_dir, workload))
+            t for t in os.listdir(
+                os.path.join(experiment_dir, study_dir, workload))
             if re.match(TRIAL_DIR_REGEX, t)
         ]
         for trial in trial_dirs: