Skip to content

Commit e5b47ea

Browse files
authored
fix(experiments/run.py): Stop duplication of RUN_TC_TAG on Consecutive Experiment Runs (#3608)
1 parent 9e2a586 commit e5b47ea

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

src/sagemaker/experiments/run.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -647,7 +647,8 @@ def _append_run_tc_label_to_tags(tags: Optional[List[Dict[str, str]]] = None) ->
647647
"""
648648
if not tags:
649649
tags = []
650-
tags.append(RUN_TC_TAG)
650+
if RUN_TC_TAG not in tags:
651+
tags.append(RUN_TC_TAG)
651652
return tags
652653

653654
def __enter__(self):

tests/unit/sagemaker/experiments/test_run.py

+5
Original file line numberDiff line numberDiff line change
@@ -936,6 +936,11 @@ def test_append_run_tc_label_to_tags():
936936
assert len(ret) == 2
937937
assert expected_tc_tag in ret
938938

939+
tags = [expected_tc_tag]
940+
ret = Run._append_run_tc_label_to_tags(tags)
941+
assert len(ret) == 1
942+
assert expected_tc_tag in ret
943+
939944

940945
def _verify_tc_status_before_enter_init(trial_component):
941946
assert not trial_component.start_time

0 commit comments

Comments
 (0)