refactor: Update TPU info retrieval to include command string in pod info fetching

Author: chengpinglin

dags/tpu_observability/tpu_info_format_validation_dags.py

@@ -417,9 +417,11 @@ def generate_second_node_pool_name(
           job_apply_time=apply_time,
       )
 
-      outputs_of_tpu_info = (
+      raw_metric_data = (
           tpu_info.get_tpu_info_from_pod.override(task_id="get_tpu_info")
-          .partial(info=cluster_info)
+          .partial(
+              info=cluster_info, cmd_str=tpu_info.TpuInfoCmd.TPU_INFO.value
+          )
           .expand(pod_name=running_pods)
       )
 
@@ -435,7 +437,7 @@ def generate_second_node_pool_name(
           tpu_info.get_tpu_info_from_pod.override(task_id="get_cli_output")
           .partial(info=cluster_info)
           .expand(
-              pod_name=pod_names,
+              pod_name=running_pods,
               cmd_str=[
                   tpu_info.TpuInfoCmd.HELP.value,
                   tpu_info.TpuInfoCmd.VERSION.value,