How about passing parameters in the form of a jinja template, like {{inputs.parameters.timestamp}}?
custom_job_task = CustomTrainingJobOp(
project="xxx",
display_name=f"xxx-{timestamp}",
worker_pool_specs=[
{
"containerSpec": {
"env": [
{"name": "BUCKET_NAME", "value": "bucket_name/{{inputs.parameters.timestamp}}"},
],
"imageUri": "xxx",
},
"replicaCount": "1",
"machineSpec": {
"machineType": "xxx",
"acceleratorType": xxx,
"acceleratorCount": 1,
},
}
],
)
User | Count |
---|---|
2 | |
1 | |
1 | |
1 | |
1 |