Skip to content

Commit 96ec1f3

Browse files
author
Dan Choi
committed
fix: use eia python version fixture in integration tests
1 parent a0f7931 commit 96ec1f3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integ/test_mxnet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -312,7 +312,7 @@ def test_deploy_model_with_accelerator(
312312
mxnet_training_job,
313313
sagemaker_session,
314314
mxnet_eia_latest_version,
315-
mxnet_inference_latest_py_version,
315+
mxnet_eia_py_version,
316316
cpu_instance_type,
317317
):
318318
endpoint_name = "test-mxnet-deploy-model-ei-{}".format(sagemaker_timestamp())
@@ -328,7 +328,7 @@ def test_deploy_model_with_accelerator(
328328
"SageMakerRole",
329329
entry_point=script_path,
330330
framework_version=mxnet_eia_latest_version,
331-
py_version=mxnet_inference_latest_py_version,
331+
py_version=mxnet_eia_py_version,
332332
sagemaker_session=sagemaker_session,
333333
)
334334
predictor = model.deploy(

0 commit comments

Comments
 (0)