Skip to content

Commit 6abf3af

Browse files
committed
fix: ensure hpt jobs inherit tags from config
1 parent ce10e01 commit 6abf3af

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

src/sagemaker/session.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3176,6 +3176,9 @@ def tune( # noqa: C901
31763176
tune_request["Autotune"] = {"Mode": "Enabled"}
31773177

31783178
tags = _append_project_tags(tags)
3179+
tags = self._append_sagemaker_config_tags(
3180+
tags, "{}.{}.{}".format(SAGEMAKER, TRAINING_JOB, TAGS)
3181+
)
31793182
if tags is not None:
31803183
tune_request["Tags"] = tags
31813184

@@ -3287,6 +3290,9 @@ def _get_tuning_request(
32873290
tune_request["WarmStartConfig"] = warm_start_config
32883291

32893292
tags = _append_project_tags(format_tags(tags))
3293+
tags = self._append_sagemaker_config_tags(
3294+
tags, "{}.{}.{}".format(SAGEMAKER, TRAINING_JOB, TAGS)
3295+
)
32903296
if tags is not None:
32913297
tune_request["Tags"] = tags
32923298

0 commit comments

Comments
 (0)