Skip to content

Commit 7cd8cbb

Browse files
committed
Fix preserve_script_path arg
1 parent 67f9a23 commit 7cd8cbb

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/integration/__init__.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -30,21 +30,21 @@
3030
model_cpu_tar = file_utils.make_tarfile(mnist_cpu_script,
3131
os.path.join(model_cpu_dir, "model.pth"),
3232
model_cpu_dir,
33-
True)
33+
preserve_script_path=True)
3434

3535
model_cpu_1d_dir = os.path.join(model_cpu_dir, '1d')
3636
mnist_1d_script = os.path.join(model_cpu_1d_dir, code_sub_dir, 'mnist_1d.py')
3737
model_cpu_1d_tar = file_utils.make_tarfile(mnist_1d_script,
3838
os.path.join(model_cpu_1d_dir, "model.pth"),
3939
model_cpu_1d_dir,
40-
True)
40+
preserve_script_path=True)
4141

4242
model_gpu_dir = os.path.join(mnist_path, gpu_sub_dir)
4343
mnist_gpu_script = os.path.join(model_gpu_dir, code_sub_dir, 'mnist.py')
4444
model_gpu_tar = file_utils.make_tarfile(mnist_gpu_script,
4545
os.path.join(model_gpu_dir, "model.pth"),
4646
model_gpu_dir,
47-
True)
47+
preserve_script_path=True)
4848

4949
model_eia_dir = os.path.join(mnist_path, eia_sub_dir)
5050
mnist_eia_script = os.path.join(model_eia_dir, 'mnist.py')
@@ -57,7 +57,7 @@
5757
os.path.join(model_cpu_dir, "model.pth"),
5858
model_cpu_dir,
5959
"model_call_model_fn_once.tar.gz",
60-
True)
60+
preserve_script_path=True)
6161

6262
ROLE = 'dummy/unused-role'
6363
DEFAULT_TIMEOUT = 20

0 commit comments

Comments
 (0)