Integ test fix

Author: dk19y

test/integration/__init__.py

@@ -25,30 +25,30 @@
 eia_sub_dir = 'model_eia'
 
 model_cpu_dir = os.path.join(mnist_path, cpu_sub_dir)
-mnist_cpu_script = os.path.join(model_cpu_dir, 'mnist.py')
+mnist_cpu_script = os.path.join(model_cpu_dir, 'code/mnist.py')
 model_cpu_tar = file_utils.make_tarfile(mnist_cpu_script,
                                         os.path.join(model_cpu_dir, "model.pth"),
                                         model_cpu_dir)
 
 model_cpu_1d_dir = os.path.join(model_cpu_dir, '1d')
-mnist_1d_script = os.path.join(model_cpu_1d_dir, 'mnist_1d.py')
+mnist_1d_script = os.path.join(model_cpu_1d_dir, 'code/mnist_1d.py')
 model_cpu_1d_tar = file_utils.make_tarfile(mnist_1d_script,
                                            os.path.join(model_cpu_1d_dir, "model.pth"),
                                            model_cpu_1d_dir)
 
 model_gpu_dir = os.path.join(mnist_path, gpu_sub_dir)
-mnist_gpu_script = os.path.join(model_gpu_dir, 'mnist.py')
+mnist_gpu_script = os.path.join(model_gpu_dir, 'code/mnist.py')
 model_gpu_tar = file_utils.make_tarfile(mnist_gpu_script,
                                         os.path.join(model_gpu_dir, "model.pth"),
                                         model_gpu_dir)
 
 model_eia_dir = os.path.join(mnist_path, eia_sub_dir)
-mnist_eia_script = os.path.join(model_eia_dir, 'mnist.py')
+mnist_eia_script = os.path.join(model_eia_dir, 'code/mnist.py')
 model_eia_tar = file_utils.make_tarfile(mnist_eia_script,
                                         os.path.join(model_eia_dir, "model.pth"),
                                         model_eia_dir)
 
-call_model_fn_once_script = os.path.join(model_cpu_dir, 'call_model_fn_once.py')
+call_model_fn_once_script = os.path.join(model_cpu_dir, 'code/call_model_fn_once.py')
 call_model_fn_once_tar = file_utils.make_tarfile(call_model_fn_once_script,
                                                  os.path.join(model_cpu_dir, "model.pth"),
                                                  model_cpu_dir,