fix: make marketplace jobnames unique

Author: mufaddal-rohawala

tests/integ/marketplace_utils.py

@@ -11,7 +11,6 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
-import uuid
 
 REGION_ACCOUNT_MAP = {
     "us-east-1": "865070037744",
@@ -35,7 +34,3 @@
     "cn-north-1": "295401494951",
     "cn-northwest-1": "304690803264",
 }
-
-
-def random_8_digit_alpha_numeric_gen():
-    return str(uuid.uuid4().hex.upper()[0:8])

tests/integ/test_marketplace.py

@@ -24,11 +24,10 @@
 from sagemaker import AlgorithmEstimator, ModelPackage
 from sagemaker.serializers import CSVSerializer
 from sagemaker.tuner import IntegerParameter, HyperparameterTuner
-from sagemaker.utils import sagemaker_timestamp
-from sagemaker.utils import _aws_partition
+from sagemaker.utils import sagemaker_timestamp, _aws_partition, unique_name_from_base
 from tests.integ import DATA_DIR
 from tests.integ.timeout import timeout, timeout_and_delete_endpoint_by_name
-from tests.integ.marketplace_utils import REGION_ACCOUNT_MAP, random_8_digit_alpha_numeric_gen
+from tests.integ.marketplace_utils import REGION_ACCOUNT_MAP
 
 
 # All these tests require a manual 1 time subscription to the following Marketplace items:
@@ -117,7 +116,7 @@ def test_marketplace_attach(sagemaker_session, cpu_instance_type):
             instance_count=1,
             instance_type=cpu_instance_type,
             sagemaker_session=sagemaker_session,
-            base_job_name="test-marketplace" + random_8_digit_alpha_numeric_gen(),
+            base_job_name=unique_name_from_base("test-marketplace"),
         )
 
         train_input = mktplace.sagemaker_session.upload_data(
@@ -205,7 +204,7 @@ def test_marketplace_tuning_job(sagemaker_session, cpu_instance_type):
         instance_count=1,
         instance_type=cpu_instance_type,
         sagemaker_session=sagemaker_session,
-        base_job_name="test-marketplace" + random_8_digit_alpha_numeric_gen(),
+        base_job_name=unique_name_from_base("test-marketplace"),
     )
 
     train_input = mktplace.sagemaker_session.upload_data(
@@ -218,7 +217,7 @@ def test_marketplace_tuning_job(sagemaker_session, cpu_instance_type):
 
     tuner = HyperparameterTuner(
         estimator=mktplace,
-        base_tuning_job_name="byo" + random_8_digit_alpha_numeric_gen(),
+        base_tuning_job_name=unique_name_from_base("byo"),
         objective_metric_name="validation:accuracy",
         hyperparameter_ranges=hyperparameter_ranges,
         max_jobs=2,
@@ -248,7 +247,7 @@ def test_marketplace_transform_job(sagemaker_session, cpu_instance_type):
         instance_count=1,
         instance_type=cpu_instance_type,
         sagemaker_session=sagemaker_session,
-        base_job_name="test-marketplace" + random_8_digit_alpha_numeric_gen(),
+        base_job_name=unique_name_from_base("test-marketplace"),
     )
 
     train_input = algo.sagemaker_session.upload_data(