Fix oncall test tf_dlrm_criteo-v5p-8 (#599)

* Update solutionsteam_tf_nightly_supported_config.py * code style
GoogleCloudPlatform · Feb 12, 2025 · c85649c · c85649c
1 parent efcb1cb
commit c85649c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dags/solutions_team/configs/tensorflow/solutionsteam_tf_nightly_supported_config.py b/dags/solutions_team/configs/tensorflow/solutionsteam_tf_nightly_supported_config.py
@@ -229,7 +229,10 @@ def get_tf_dlrm_v1_config(
       set_up_cmds += common.set_up_se()
     else:
       set_up_cmds += common.set_up_pjrt()
-  set_up_cmds += ("sudo apt-get install numactl",)
+  set_up_cmds += (
+      "sudo apt-get -y update",
+      "sudo apt-get install numactl",
+  )
   params_override = {
       "runtime": {"distribution_strategy": "tpu"},
       "task": {