如何设置VertexAI中PipelineJob的run name运行名称

你可以通过在初始化PipelineJob时传入job_id参数来自定义指定run name的取值，该参数对应的就是Vertex AI控制台显示的运行唯一名称。

注意事项 #

• job_id需要满足全局唯一性要求：同项目同区域下不能与历史已运行/正在运行的流水线作业ID重复，传入重复ID会触发创建失败报错
• 仅支持小写字母、数字、连字符（-），必须以字母开头，长度不超过128字符

修改后的代码示例 #

import google.cloud.aiplatform as vertexai

vertexai.init(project=PROJECT_ID,staging_bucket=PIPELINE_ROOT)

job = vertexai.PipelineJob(
    display_name='pipeline-test-1',
    template_path='xgb_pipe.json',
    # 自定义run name，可替换为你需要的取值
    job_id="xgb-pipeline-test-run-20240520-001"
)

job.run()

如果需要自动生成唯一ID避免重复，可以拼接时间戳实现：

import google.cloud.aiplatform as vertexai
from datetime import datetime

vertexai.init(project=PROJECT_ID,staging_bucket=PIPELINE_ROOT)

# 自动生成带时间戳的唯一job id
custom_job_id = f"xgb-pipeline-run-{datetime.now().strftime('%Y%m%d-%H%M%S')}"

job = vertexai.PipelineJob(
    display_name='pipeline-test-1',
    template_path='xgb_pipe.json',
    job_id=custom_job_id
)

job.run()

补充说明：job_id和display_name作用不同，display_name支持重复、支持中文、空格等特殊字符，仅用作控制台展示的可读名称，不会影响run name的最终取值。

内容的提问来源于stack exchange，提问作者Pentium10