Skip to content

Commit 6f1df0d

Browse files
author
Jonathan Esterhazy
committed
remove temp image uri
1 parent 98d8e93 commit 6f1df0d

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/integ/test_tfs.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,13 +29,12 @@ def instance_type(request):
2929

3030
@pytest.fixture(scope='module')
3131
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'
3332
endpoint_name = sagemaker.utils.name_from_base('sagemaker-tensorflow-serving')
3433
model_data = sagemaker_session.upload_data(
3534
path='tests/data/tensorflow-serving-test-model.tar.gz',
3635
key_prefix='tensorflow-serving/models')
3736
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',
3938
framework_version=tf_full_version,
4039
sagemaker_session=sagemaker_session)
4140
predictor = model.deploy(1, instance_type, endpoint_name=endpoint_name)

0 commit comments

Comments
 (0)