Skip to content

Commit 163a730

Browse files
Update integration test to current sagemaker-python-sdk API.
1 parent 7bd66ed commit 163a730

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/integration/local/test_serving.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -89,10 +89,10 @@ def test_serving_calls_model_fn_once(image_uri, sagemaker_local_session, instanc
8989
@contextmanager
9090
def _predictor(model_tar, script, image, sagemaker_local_session, instance_type,
9191
model_server_workers=None):
92-
model = PyTorchModel('file://{}'.format(model_tar),
93-
ROLE,
94-
script,
95-
image=image,
92+
model = PyTorchModel(model_data='file://{}'.format(model_tar),
93+
role=ROLE,
94+
entry_point=script,
95+
image_uri=image,
9696
sagemaker_session=sagemaker_local_session,
9797
model_server_workers=model_server_workers)
9898

0 commit comments

Comments
 (0)