Skip to content

Commit db3426a

Browse files
authored
infra: use unique model name in TFS integ tests (#1754)
1 parent e6a1e8e commit db3426a

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/integ/test_tfs.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ def tfs_predictor(sagemaker_session, tf_full_version):
3939
role="SageMakerRole",
4040
framework_version=tf_full_version,
4141
sagemaker_session=sagemaker_session,
42+
name=endpoint_name,
4243
)
4344
predictor = model.deploy(1, "ml.c5.xlarge", endpoint_name=endpoint_name)
4445
yield predictor
@@ -67,6 +68,7 @@ def tfs_predictor_with_model_and_entry_point_same_tar(
6768
role="SageMakerRole",
6869
framework_version=tf_full_version,
6970
sagemaker_session=sagemaker_local_session,
71+
name=endpoint_name,
7072
)
7173
predictor = model.deploy(1, "local", endpoint_name=endpoint_name)
7274

@@ -100,6 +102,7 @@ def tfs_predictor_with_model_and_entry_point_and_dependencies(
100102
dependencies=dependencies,
101103
framework_version=tf_full_version,
102104
sagemaker_session=sagemaker_local_session,
105+
name=endpoint_name,
103106
)
104107

105108
predictor = model.deploy(1, "local", endpoint_name=endpoint_name)
@@ -123,6 +126,7 @@ def tfs_predictor_with_accelerator(sagemaker_session, ei_tf_full_version, cpu_in
123126
role="SageMakerRole",
124127
framework_version=ei_tf_full_version,
125128
sagemaker_session=sagemaker_session,
129+
name=endpoint_name,
126130
)
127131
predictor = model.deploy(
128132
1, cpu_instance_type, endpoint_name=endpoint_name, accelerator_type="ml.eia1.medium"

0 commit comments

Comments
 (0)