fix logging

mlcommons · Sep 28, 2023 · 27cb037 · 27cb037
1 parent b23aae4
commit 27cb037
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/submission_runner.py b/submission_runner.py
@@ -599,7 +599,7 @@ def main(_):
       workload_class_name=workload_metadata['workload_class_name'],
       workload_init_kwargs=workload_init_kwargs)
   workload_path = workload_metadata['workload_path']
-  workload_class = workload_metadata['workload_class']
+  workload_class = workload_metadata['workload_class_name']
   logging.info(f'workload : {workload_path}')
   logging.info(f'workload class: {workload_class}')