File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -29,13 +29,12 @@ def instance_type(request):
29
29
30
30
@pytest .fixture (scope = 'module' )
31
31
def tfs_predictor (instance_type , sagemaker_session , tf_full_version ):
32
- image = '237082650222.dkr.ecr.us-west-2.amazonaws.com/sagemaker-tensorflow-serving:1.11-cpu'
33
32
endpoint_name = sagemaker .utils .name_from_base ('sagemaker-tensorflow-serving' )
34
33
model_data = sagemaker_session .upload_data (
35
34
path = 'tests/data/tensorflow-serving-test-model.tar.gz' ,
36
35
key_prefix = 'tensorflow-serving/models' )
37
36
with timeout_and_delete_endpoint_by_name (endpoint_name , sagemaker_session ):
38
- model = Model (model_data = model_data , role = 'SageMakerRole' , image = image ,
37
+ model = Model (model_data = model_data , role = 'SageMakerRole' ,
39
38
framework_version = tf_full_version ,
40
39
sagemaker_session = sagemaker_session )
41
40
predictor = model .deploy (1 , instance_type , endpoint_name = endpoint_name )
You can’t perform that action at this time.
0 commit comments