From 6885e6963402e1740c8f7fb55aa00738502e818a Mon Sep 17 00:00:00 2001
From: Priya Kasimbeg <kasimbeg@google.com>
Date: Tue, 28 Nov 2023 00:49:16 +0000
Subject: [PATCH] fix arg in submission_runner

---
 submission_runner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/submission_runner.py b/submission_runner.py
index 2e642c8ce..152da65fd 100644
--- a/submission_runner.py
+++ b/submission_runner.py
@@ -228,7 +228,7 @@ def train_once(
       ]
       eager_backend_workloads = ['librispeech_deepspeech']
       aot_eager_backend_workloads = []
-      base_workload = workloads.get_base_workload_name(FLAGS.workload)
+      base_workload = workloads.get_base_workload_name(workload)
       if base_workload in compile_error_workloads:
         logging.warning(
             'These workloads cannot be fully compiled under current '