Skip to content

Commit fb924a9

Browse files
committed
Fix bugs in path definition
1 parent b04f6a4 commit fb924a9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/integration/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,13 +62,13 @@
6262
"model_call_model_fn_once.tar.gz",
6363
script_path="code")
6464

65-
default_model_dir = os.path.join(resnet18_path, default_sub_dir)
65+
default_model_dir = os.path.join(resnet18_path, code_sub_dir, default_sub_dir)
6666
default_model_script = os.path.join(default_model_dir, "resnet18.py")
6767
default_model_tar = file_utils.make_tarfile(
6868
default_model_script, os.path.join(default_model_dir, "model.pt"), default_model_dir, script_path="code"
6969
)
7070

71-
default_model_eia_dir = os.path.join(mnist_path, default_sub_dir)
71+
default_model_eia_dir = os.path.join(mnist_path, code_sub_dir, default_sub_eia_dir)
7272
default_model_eia_script = os.path.join(default_model_eia_dir, "mnist.py")
7373
default_model_eia_tar = file_utils.make_tarfile(
7474
default_model_eia_script, os.path.join(default_model_eia_dir, "model.pt"), default_model_eia_dir

0 commit comments

Comments
 (0)