From 25b3fbe47727381eabd761ed33111b5daf458275 Mon Sep 17 00:00:00 2001
From: Priya Kasimbeg <kasimbeg@google.com>
Date: Tue, 28 Nov 2023 00:42:10 +0000
Subject: [PATCH] push

---
 submission_runner.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/submission_runner.py b/submission_runner.py
index ae4360a6f..2e642c8ce 100644
--- a/submission_runner.py
+++ b/submission_runner.py
@@ -221,8 +221,9 @@ def train_once(
         model_init_rng, dropout_rate, aux_dropout_rate)
     if FLAGS.framework == 'pytorch' and FLAGS.torch_compile:
       compile_error_workloads = [
-          'librispeech_conformer', 
-          'ogbg', 'criteo1tb', 
+          'librispeech_conformer',
+          'ogbg',
+          'criteo1tb',
           'imagenet_vit',
       ]
       eager_backend_workloads = ['librispeech_deepspeech']