changed some syntax

Basil Beirouti · Basil Beirouti · commit 560c3d852f28 · 2022-06-01T17:12:45.000-07:00
diff --git a/tests/unit/test_mxnet.py b/tests/unit/test_mxnet.py
@@ -747,7 +747,7 @@ def test_model_prepare_container_def_no_instance_type_or_image(
 
 
 def test_attach(sagemaker_session, mxnet_training_version, mxnet_training_py_version):
-    if mxnet_training_py_version == "py37" or mxnet_training_py_version == "py38":
+    if mxnet_training_py_version in ["py37", "py38"]:
         training_image = "1.dkr.ecr.us-west-2.amazonaws.com/mxnet-training:{1}-cpu-{0}".format(
             mxnet_training_py_version, mxnet_training_version
         )

Original file line number	Diff line number	Diff line change
`@@ -747,7 +747,7 @@ def test_model_prepare_container_def_no_instance_type_or_image(`
`747`	`747`
`748`	`748`
`749`	`749`	`def test_attach(sagemaker_session, mxnet_training_version, mxnet_training_py_version):`
`750`		`- if mxnet_training_py_version == "py37" or mxnet_training_py_version == "py38":`
	`750`	`+ if mxnet_training_py_version in ["py37", "py38"]:`
`751`	`751`	`training_image = "1.dkr.ecr.us-west-2.amazonaws.com/mxnet-training:{1}-cpu-{0}".format(`
`752`	`752`	`mxnet_training_py_version, mxnet_training_version`
`753`	`753`	`)`