Linting fix

mlcommons · Jul 27, 2023 · 6a18963 · 6a18963
1 parent 89a89b8
commit 6a18963
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/submission_runner.py b/submission_runner.py
@@ -191,9 +191,7 @@ def train_once(
     model_params, model_state = workload.init_model_fn(
         model_init_rng, dropout_rate, aux_dropout_rate)
     if FLAGS.framework == 'pytorch' and FLAGS.torch_compile:
-      compile_error_workloads = [
-          'ogbg', 'librispeech_deepspeech', 'wmt'
-      ]
+      compile_error_workloads = ['ogbg', 'librispeech_deepspeech', 'wmt']
       eager_backend_workloads = ['librispeech_conformer']
       aot_eager_backend_workloads = ['criteo1tb']
       if FLAGS.workload in compile_error_workloads: