@@ -39,6 +39,7 @@ def tfs_predictor(sagemaker_session, tf_full_version):
39
39
role = "SageMakerRole" ,
40
40
framework_version = tf_full_version ,
41
41
sagemaker_session = sagemaker_session ,
42
+ name = endpoint_name ,
42
43
)
43
44
predictor = model .deploy (1 , "ml.c5.xlarge" , endpoint_name = endpoint_name )
44
45
yield predictor
@@ -67,6 +68,7 @@ def tfs_predictor_with_model_and_entry_point_same_tar(
67
68
role = "SageMakerRole" ,
68
69
framework_version = tf_full_version ,
69
70
sagemaker_session = sagemaker_local_session ,
71
+ name = endpoint_name ,
70
72
)
71
73
predictor = model .deploy (1 , "local" , endpoint_name = endpoint_name )
72
74
@@ -100,6 +102,7 @@ def tfs_predictor_with_model_and_entry_point_and_dependencies(
100
102
dependencies = dependencies ,
101
103
framework_version = tf_full_version ,
102
104
sagemaker_session = sagemaker_local_session ,
105
+ name = endpoint_name ,
103
106
)
104
107
105
108
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
123
126
role = "SageMakerRole" ,
124
127
framework_version = ei_tf_full_version ,
125
128
sagemaker_session = sagemaker_session ,
129
+ name = endpoint_name ,
126
130
)
127
131
predictor = model .deploy (
128
132
1 , cpu_instance_type , endpoint_name = endpoint_name , accelerator_type = "ml.eia1.medium"
0 commit comments