Skip to content

Commit 420be37

Browse files
danabensDewen Qi
authored and
Dewen Qi
committed
fix: lowercase run name (aws#767)
1 parent 6c076c7 commit 420be37

File tree

2 files changed

+19
-16
lines changed

2 files changed

+19
-16
lines changed

src/sagemaker/experiments/run.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -600,7 +600,10 @@ def _generate_trial_component_name(run_name: str, experiment_name: str) -> str:
600600
raise ValueError(
601601
err_msg_template.format("experiment_name", len(experiment_name), max_len)
602602
)
603-
return "{}{}{}".format(experiment_name, DELIMITER, run_name)
603+
experiment_scoped_tc_name = "{}{}{}".format(experiment_name, DELIMITER, run_name)
604+
# https://t.corp.amazon.com/P77144351
605+
lower_cased_tc_name = experiment_scoped_tc_name.lower()
606+
return lower_cased_tc_name
604607

605608
@staticmethod
606609
def _extract_run_name_from_tc_name(trial_component_name: str, experiment_name: str) -> str:

tests/unit/sagemaker/experiments/test_run.py

+15-15
Original file line numberDiff line numberDiff line change
@@ -710,10 +710,10 @@ def test_list(mock_tc_search, mock_tc_list, mock_tc_load, run_obj, sagemaker_ses
710710
[
711711
TrialComponentSearchResult(
712712
trial_component_name=Run._generate_trial_component_name(
713-
"A" + str(i), TEST_EXP_NAME
713+
"a" + str(i), TEST_EXP_NAME
714714
),
715-
trial_component_arn="B" + str(i),
716-
display_name="C" + str(i),
715+
trial_component_arn="b" + str(i),
716+
display_name="c" + str(i),
717717
creation_time=creation_time + datetime.timedelta(hours=i),
718718
last_modified_time=last_modified_time + datetime.timedelta(hours=i),
719719
last_modified_by={},
@@ -724,12 +724,12 @@ def test_list(mock_tc_search, mock_tc_list, mock_tc_load, run_obj, sagemaker_ses
724724
]
725725
mock_tc_list.return_value = [
726726
TrialComponentSummary(
727-
trial_component_name=Run._generate_trial_component_name("A" + str(i), TEST_EXP_NAME),
728-
trial_component_arn="B" + str(i),
729-
display_name="C" + str(i),
730-
source_arn="D" + str(i),
727+
trial_component_name=Run._generate_trial_component_name("a" + str(i), TEST_EXP_NAME),
728+
trial_component_arn="b" + str(i),
729+
display_name="c" + str(i),
730+
source_arn="d" + str(i),
731731
status=TrialComponentStatus(
732-
primary_status=_TrialComponentStatusType.InProgress.value, message="E" + str(i)
732+
primary_status=_TrialComponentStatusType.InProgress.value, message="e" + str(i)
733733
),
734734
start_time=start_time + datetime.timedelta(hours=i),
735735
end_time=end_time + datetime.timedelta(hours=i),
@@ -743,13 +743,13 @@ def test_list(mock_tc_search, mock_tc_list, mock_tc_load, run_obj, sagemaker_ses
743743
(
744744
_TrialComponent(
745745
trial_component_name=Run._generate_trial_component_name(
746-
"A" + str(i), TEST_EXP_NAME
746+
"a" + str(i), TEST_EXP_NAME
747747
),
748-
trial_component_arn="B" + str(i),
749-
display_name="C" + str(i),
750-
source_arn="D" + str(i),
748+
trial_component_arn="b" + str(i),
749+
display_name="c" + str(i),
750+
source_arn="d" + str(i),
751751
status=TrialComponentStatus(
752-
primary_status=_TrialComponentStatusType.InProgress.value, message="E" + str(i)
752+
primary_status=_TrialComponentStatusType.InProgress.value, message="e" + str(i)
753753
),
754754
start_time=start_time + datetime.timedelta(hours=i),
755755
end_time=end_time + datetime.timedelta(hours=i),
@@ -783,12 +783,12 @@ def test_list(mock_tc_search, mock_tc_list, mock_tc_load, run_obj, sagemaker_ses
783783
for i in range(tc_list_len_half):
784784
run = run_list[i]
785785
assert run.experiment_name == TEST_EXP_NAME
786-
assert run.run_name == "A" + str(i)
786+
assert run.run_name == "a" + str(i)
787787
assert run._experiment
788788
assert run._trial
789789
assert isinstance(run._trial_component, _TrialComponent)
790790
assert run._trial_component.trial_component_name == Run._generate_trial_component_name(
791-
"A" + str(i), TEST_EXP_NAME
791+
"a" + str(i), TEST_EXP_NAME
792792
)
793793
assert run._in_load is False
794794
assert run._inside_load_context is False

0 commit comments

Comments
 (0)