diff --git a/components/sample/keras/train_classifier/README.md b/components/sample/keras/train_classifier/README.md index 3cc6b8d2dae..196c50d12ab 100644 --- a/components/sample/keras/train_classifier/README.md +++ b/components/sample/keras/train_classifier/README.md @@ -53,7 +53,7 @@ training_set_labels_gcs_path = os.path.join(input_data_gcs_dir, 'training_set_la gfile.Copy(training_set_features_local_path, training_set_features_gcs_path) gfile.Copy(training_set_labels_local_path, training_set_labels_gcs_path) -output_model_uri_template = os.path.join(output_data_gcs_dir, '{{workflow.uid}}/{{pod.name}}/output_model_uri/data') +output_model_uri_template = os.path.join(output_data_gcs_dir, kfp.dsl.task_id_placeholder, 'output_model_uri', 'data') xor_model_config = requests.get(test_data_url_prefix + 'model_config.json').content diff --git a/samples/core/xgboost_training_cm/xgboost_training_cm.py b/samples/core/xgboost_training_cm/xgboost_training_cm.py index 1f17cd803f5..5b2e29746d6 100755 --- a/samples/core/xgboost_training_cm/xgboost_training_cm.py +++ b/samples/core/xgboost_training_cm/xgboost_training_cm.py @@ -212,7 +212,7 @@ def xgb_train_pipeline( workers=2, true_label='ACTION', ): - output_template = str(output) + '/{{workflow.uid}}/{{pod.name}}/data' + output_template = str(output) + '/' + dsl.task_id_placeholder + '/data' delete_cluster_op = dataproc_delete_cluster_op( project, diff --git a/sdk/python/kfp/dsl/__init__.py b/sdk/python/kfp/dsl/__init__.py index fb12902583a..08259339702 100644 --- a/sdk/python/kfp/dsl/__init__.py +++ b/sdk/python/kfp/dsl/__init__.py @@ -25,3 +25,6 @@ from ._ops_group import OpsGroup, ExitHandler, Condition, ParallelFor from ._component import python_component, graph_component, component from ._artifact_location import ArtifactLocation + +task_id_placeholder = '{{workflow.uid}}-{{pod.name}}' +run_id_placeholder = '{{workflow.uid}}'