From 4b0b8b377ff92b94f10771c73111f44eca4e609e Mon Sep 17 00:00:00 2001 From: gkroiz Date: Fri, 12 Jan 2024 23:40:43 +0000 Subject: [PATCH] fix unintentional syntax changes --- .../configs/pax/solutionsteam_pax_supported_config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dags/solutions_team/configs/pax/solutionsteam_pax_supported_config.py b/dags/solutions_team/configs/pax/solutionsteam_pax_supported_config.py index 0f66c65c..c2be0ee8 100644 --- a/dags/solutions_team/configs/pax/solutionsteam_pax_supported_config.py +++ b/dags/solutions_team/configs/pax/solutionsteam_pax_supported_config.py @@ -110,7 +110,7 @@ def get_pax_lm_config( job_log_dir = f"{log_dir}/{model_name}-{short_id}" set_up_cmds = get_setup_cmds(pax_version, ckp_path, job_log_dir) - runtime_version = (get_runtime_version(pax_version, tpu_version),) + runtime_version = get_runtime_version(pax_version, tpu_version) if runtime_version == RuntimeVersion.TPU_VM_V4_BASE.value: training_script_path = ".local/lib/python3.8/site-packages/paxml/main.py"