Skip to content

Commit 98f4178

Browse files
committed
update test for new MMS usage (aka not supported)
1 parent a635631 commit 98f4178

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

tests/integ/test_mxnet_train.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,13 @@ def test_attach_deploy(mxnet_training_job, sagemaker_session, cpu_instance_type)
6767

6868
with timeout_and_delete_endpoint_by_name(endpoint_name, sagemaker_session):
6969
estimator = MXNet.attach(mxnet_training_job, sagemaker_session=sagemaker_session)
70-
predictor = estimator.deploy(1, cpu_instance_type, endpoint_name=endpoint_name)
70+
predictor = estimator.deploy(
71+
1,
72+
cpu_instance_type,
73+
entry_point="mnist.py",
74+
source_dir=os.path.join(DATA_DIR, "mxnet_mnist", "mnist.py"),
75+
endpoint_name=endpoint_name,
76+
)
7177
data = numpy.zeros(shape=(1, 1, 28, 28))
7278
result = predictor.predict(data)
7379
assert result is not None

0 commit comments

Comments
 (0)