Change: Minimize use of lower case tc name (aws#769)

Author: danabens

src/sagemaker/experiments/_metrics.py

@@ -283,7 +283,7 @@ def _send_metrics(self, metrics):
     def _construct_batch_put_metrics_request(self, batch):
         """Creates dictionary object used as request to metrics service."""
         return {
-            "TrialComponentName": self._trial_component_name,
+            "TrialComponentName": self._trial_component_name.lower(),
             "MetricData": list(map(lambda x: x.to_raw_metric_data(), batch)),
         }
 

src/sagemaker/experiments/run.py

@@ -600,10 +600,7 @@ def _generate_trial_component_name(run_name: str, experiment_name: str) -> str:
             raise ValueError(
                 err_msg_template.format("experiment_name", len(experiment_name), max_len)
             )
-        experiment_scoped_tc_name = "{}{}{}".format(experiment_name, DELIMITER, run_name)
-        # https://t.corp.amazon.com/P77144351
-        lower_cased_tc_name = experiment_scoped_tc_name.lower()
-        return lower_cased_tc_name
+        return "{}{}{}".format(experiment_name, DELIMITER, run_name)
 
     @staticmethod
     def _extract_run_name_from_tc_name(trial_component_name: str, experiment_name: str) -> str:

tests/integ/sagemaker/experiments/test_run.py

@@ -14,6 +14,7 @@
 
 import datetime
 import os
+import time
 
 import pytest
 
@@ -569,12 +570,13 @@ def _check_run_from_local_end_result(sagemaker_session, tc, is_complete_log=True
     assert s3_prefix in tc.output_artifacts[file_artifact_name].value
     assert "text/plain" == tc.output_artifacts[file_artifact_name].media_type
 
-    # https://t.corp.amazon.com/P77144351
-    # assert len(tc.metrics) == 1
-    # metric_summary = tc.metrics[0]
-    # assert metric_summary.metric_name == metric_name
-    # assert metric_summary.max == 9.0
-    # assert metric_summary.min == 0.0
+    # wait for metrics -> eureka propagation for test consistency
+    time.sleep(3)
+    assert len(tc.metrics) == 1
+    metric_summary = tc.metrics[0]
+    assert metric_summary.metric_name == metric_name
+    assert metric_summary.max == 9.0
+    assert metric_summary.min == 0.0
 
 
 def _check_run_from_job_result(sagemaker_session, tc_name=None, is_init=True, has_extra_load=False):

tests/unit/sagemaker/experiments/test_run.py

@@ -710,10 +710,10 @@ def test_list(mock_tc_search, mock_tc_list, mock_tc_load, run_obj, sagemaker_ses
         [
             TrialComponentSearchResult(
                 trial_component_name=Run._generate_trial_component_name(
-                    "a" + str(i), TEST_EXP_NAME
+                    "A" + str(i), TEST_EXP_NAME
                 ),
-                trial_component_arn="b" + str(i),
-                display_name="c" + str(i),
+                trial_component_arn="B" + str(i),
+                display_name="C" + str(i),
                 creation_time=creation_time + datetime.timedelta(hours=i),
                 last_modified_time=last_modified_time + datetime.timedelta(hours=i),
                 last_modified_by={},
@@ -724,12 +724,12 @@ def test_list(mock_tc_search, mock_tc_list, mock_tc_load, run_obj, sagemaker_ses
     ]
     mock_tc_list.return_value = [
         TrialComponentSummary(
-            trial_component_name=Run._generate_trial_component_name("a" + str(i), TEST_EXP_NAME),
-            trial_component_arn="b" + str(i),
-            display_name="c" + str(i),
-            source_arn="d" + str(i),
+            trial_component_name=Run._generate_trial_component_name("A" + str(i), TEST_EXP_NAME),
+            trial_component_arn="B" + str(i),
+            display_name="C" + str(i),
+            source_arn="D" + str(i),
             status=TrialComponentStatus(
-                primary_status=_TrialComponentStatusType.InProgress.value, message="e" + str(i)
+                primary_status=_TrialComponentStatusType.InProgress.value, message="E" + str(i)
             ),
             start_time=start_time + datetime.timedelta(hours=i),
             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
         (
             _TrialComponent(
                 trial_component_name=Run._generate_trial_component_name(
-                    "a" + str(i), TEST_EXP_NAME
+                    "A" + str(i), TEST_EXP_NAME
                 ),
-                trial_component_arn="b" + str(i),
-                display_name="c" + str(i),
-                source_arn="d" + str(i),
+                trial_component_arn="B" + str(i),
+                display_name="C" + str(i),
+                source_arn="D" + str(i),
                 status=TrialComponentStatus(
-                    primary_status=_TrialComponentStatusType.InProgress.value, message="e" + str(i)
+                    primary_status=_TrialComponentStatusType.InProgress.value, message="E" + str(i)
                 ),
                 start_time=start_time + datetime.timedelta(hours=i),
                 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
     for i in range(tc_list_len_half):
         run = run_list[i]
         assert run.experiment_name == TEST_EXP_NAME
-        assert run.run_name == "a" + str(i)
+        assert run.run_name == "A" + str(i)
         assert run._experiment
         assert run._trial
         assert isinstance(run._trial_component, _TrialComponent)
         assert run._trial_component.trial_component_name == Run._generate_trial_component_name(
-            "a" + str(i), TEST_EXP_NAME
+            "A" + str(i), TEST_EXP_NAME
         )
         assert run._in_load is False
         assert run._inside_load_context is False