@@ -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