fix mxnet estimator unit tests

Rui Wang Napieralski · Rui Wang Napieralski · commit aae58db04e07 · 2020-09-28T22:11:21.000-07:00
diff --git a/tests/unit/test_mxnet.py b/tests/unit/test_mxnet.py
@@ -175,7 +175,7 @@ def _create_compilation_job(input_shape, output_location):
 
 
 def _neo_inference_image(mxnet_version):
-    return "301217895009.dkr.ecr.us-west-2.amazonaws.com/sagemaker-neo-{}:{}-cpu-py3".format(
+    return "301217895009.dkr.ecr.us-west-2.amazonaws.com/sagemaker-inference-{}:{}-cpu-py3".format(
         FRAMEWORK.lower(), mxnet_version
     )
 

Original file line number	Diff line number	Diff line change
`@@ -175,7 +175,7 @@ def _create_compilation_job(input_shape, output_location):`
`175`	`175`
`176`	`176`
`177`	`177`	`def _neo_inference_image(mxnet_version):`
`178`		`- return "301217895009.dkr.ecr.us-west-2.amazonaws.com/sagemaker-neo-{}:{}-cpu-py3".format(`
	`178`	`+ return "301217895009.dkr.ecr.us-west-2.amazonaws.com/sagemaker-inference-{}:{}-cpu-py3".format(`
`179`	`179`	`FRAMEWORK.lower(), mxnet_version`
`180`	`180`	`)`
`181`	`181`