Skip to content

Commit 31785e2

Browse files
author
Deng
committed
fix test typo
1 parent fc1f72a commit 31785e2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integ/test_neo_mxnet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
from tests.integ.timeout import timeout, timeout_and_delete_endpoint_by_name
2525

2626
NEO_MXNET_VERSION = "1.4.1" # Neo doesn't support MXNet 1.6 yet.
27-
INF_MXNET_VERSION = "1.5.0"
27+
INF_MXNET_VERSION = "1.5.1"
2828

2929

3030
@pytest.fixture(scope="module")
@@ -130,7 +130,7 @@ def test_inferentia_deploy_model(
130130
model_data,
131131
role,
132132
entry_point=script_path,
133-
framework_version=NEO_MXNET_VERSION,
133+
framework_version=INF_MXNET_VERSION,
134134
sagemaker_session=sagemaker_session,
135135
)
136136

0 commit comments

Comments
 (0)