fix integ test

laurenyu · laurenyu · commit 2b58b04025ff · 2020-06-10T10:36:07.000-07:00
diff --git a/tests/integ/test_mxnet_train.py b/tests/integ/test_mxnet_train.py
@@ -71,7 +71,7 @@ def test_attach_deploy(mxnet_training_job, sagemaker_session, cpu_instance_type)
             1,
             cpu_instance_type,
             entry_point="mnist.py",
-            source_dir=os.path.join(DATA_DIR, "mxnet_mnist", "mnist.py"),
+            source_dir=os.path.join(DATA_DIR, "mxnet_mnist"),
             endpoint_name=endpoint_name,
         )
         data = numpy.zeros(shape=(1, 1, 28, 28))

Original file line number	Diff line number	Diff line change
`@@ -71,7 +71,7 @@ def test_attach_deploy(mxnet_training_job, sagemaker_session, cpu_instance_type)`
`71`	`71`	`1,`
`72`	`72`	`cpu_instance_type,`
`73`	`73`	`entry_point="mnist.py",`
`74`		`- source_dir=os.path.join(DATA_DIR, "mxnet_mnist", "mnist.py"),`
	`74`	`+ source_dir=os.path.join(DATA_DIR, "mxnet_mnist"),`
`75`	`75`	`endpoint_name=endpoint_name,`
`76`	`76`	`)`
`77`	`77`	`data = numpy.zeros(shape=(1, 1, 28, 28))`