Support image_name in docker_settings

mlcommons · Jan 28, 2025 · fc0d4f7 · fc0d4f7
1 parent 09e07b4
commit fc0d4f7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/automation/script/docker_utils.py b/automation/script/docker_utils.py
@@ -102,7 +102,7 @@ def prepare_docker_inputs(input_params, docker_settings,
 
     keys = [
         "mlc_repo", "mlc_repo_branch", "base_image", "os", "os_version",
-        "mlc_repos", "skip_mlc_sys_upgrade", "extra_sys_deps",
+        "mlc_repos", "skip_mlc_sys_upgrade", "extra_sys_deps", "image_name",
         "gh_token", "fake_run_deps", "run_final_cmds", "real_run", "copy_files", "path", "user"
     ]