Skip to content

Commit 8d875e7

Browse files
committed
fix test
Signed-off-by: h-guo18 <67671475+h-guo18@users.noreply.github.com>
1 parent b960ced commit 8d875e7

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tools/launcher/tests/test_docker_execution.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ def test_scratch_dir_created(self, tmp_path):
5151
modelopt_src_path="/tmp/modelopt",
5252
experiment_title="cicd",
5353
)
54-
scratch_dir = os.path.join(job_dir, "cicd", "exp_123", "task_0")
54+
scratch_dir = os.path.join(job_dir, "cicd", "exp_123")
5555
assert os.path.isdir(scratch_dir)
5656

5757
def test_hf_local_mount(self, tmp_path):
@@ -100,7 +100,7 @@ def test_scratchspace_mount(self, tmp_path):
100100
experiment_title="cicd",
101101
)
102102
volumes = executor.volumes
103-
expected_scratch = os.path.join(job_dir, "cicd", "exp_456", "job_0")
103+
expected_scratch = os.path.join(job_dir, "cicd", "exp_456")
104104
assert any(f"{expected_scratch}:/scratchspace" in v for v in volumes)
105105

106106
def test_modelopt_mount(self, tmp_path):

0 commit comments

Comments
 (0)