fix unintentional syntax changes

GoogleCloudPlatform · Jan 12, 2024 · 4b0b8b3 · 4b0b8b3
1 parent 2b26134
commit 4b0b8b3
Showing 1 changed file with 1 addition and 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
@@ -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"