diff --git a/src/sagemaker/session.py b/src/sagemaker/session.py index fb65c9f2cf..7035d9547d 100644 --- a/src/sagemaker/session.py +++ b/src/sagemaker/session.py @@ -3193,6 +3193,9 @@ def tune( # noqa: C901 tune_request["Autotune"] = {"Mode": "Enabled"} tags = _append_project_tags(tags) + tags = self._append_sagemaker_config_tags( + tags, "{}.{}.{}".format(SAGEMAKER, TRAINING_JOB, TAGS) + ) if tags is not None: tune_request["Tags"] = tags @@ -3304,6 +3307,9 @@ def _get_tuning_request( tune_request["WarmStartConfig"] = warm_start_config tags = _append_project_tags(format_tags(tags)) + tags = self._append_sagemaker_config_tags( + tags, "{}.{}.{}".format(SAGEMAKER, TRAINING_JOB, TAGS) + ) if tags is not None: tune_request["Tags"] = tags