Skip to content

Commit 973095b

Browse files
committed
fix bug for extending tags
1 parent a140214 commit 973095b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sagemaker-train/src/sagemaker/train/evaluate/execution.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def _create_evaluation_pipeline(
6868
resolved_pipeline_definition = template.render(pipeline_name=pipeline_name)
6969

7070
# Create tags for the pipeline
71-
tags = tags.extend([
71+
tags.extend([
7272
{"key": _TAG_SAGEMAKER_MODEL_EVALUATION, "value": "true"}
7373
])
7474

@@ -207,7 +207,7 @@ def _get_or_create_pipeline(
207207

208208
# Get tags using Tag.get_all
209209
tags_list = Tag.get_all(resource_arn=pipeline_arn, session=session, region=region)
210-
tags = tags.extend({tag.key: tag.value for tag in tags_list})
210+
tags.extend({tag.key: tag.value for tag in tags_list})
211211

212212
# Validate tag
213213
if tags.get(_TAG_SAGEMAKER_MODEL_EVALUATION) == "true":

0 commit comments

Comments
 (0)