Skip to content

Commit 2f7bb50

Browse files
committed
fixed typo.
1 parent 64805a9 commit 2f7bb50

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

tests/vec_inf/client/test_slurm_script_generator.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def basic_params(self):
1919
"""Generate basic SLURM configuration parameters."""
2020
return {
2121
"model_name": "test-model",
22-
"model_weights_parent_dir": "/path/to/model_wieghts",
22+
"model_weights_parent_dir": "/path/to/model_weights",
2323
"src_dir": "/path/to/src",
2424
"log_dir": "/path/to/logs",
2525
"num_nodes": "1",
@@ -72,7 +72,7 @@ def test_init_single_node(self, basic_params):
7272
assert not generator.is_multinode
7373
assert not generator.use_singularity
7474
assert generator.additional_binds == ""
75-
assert generator.model_weights_path == "/path/to/model_wieghts/test-model"
75+
assert generator.model_weights_path == "/path/to/model_weights/test-model"
7676

7777
def test_init_multinode(self, multinode_params):
7878
"""Test initialization with multi-node configuration."""
@@ -82,7 +82,7 @@ def test_init_multinode(self, multinode_params):
8282
assert generator.is_multinode
8383
assert not generator.use_singularity
8484
assert generator.additional_binds == ""
85-
assert generator.model_weights_path == "/path/to/model_wieghts/test-model"
85+
assert generator.model_weights_path == "/path/to/model_weights/test-model"
8686

8787
def test_init_singularity(self, singularity_params):
8888
"""Test initialization with Singularity configuration."""
@@ -92,7 +92,7 @@ def test_init_singularity(self, singularity_params):
9292
assert generator.use_singularity
9393
assert not generator.is_multinode
9494
assert generator.additional_binds == " --bind /scratch:/scratch,/data:/data"
95-
assert generator.model_weights_path == "/path/to/model_wieghts/test-model"
95+
assert generator.model_weights_path == "/path/to/model_weights/test-model"
9696

9797
def test_init_singularity_no_bind(self, basic_params):
9898
"""Test Singularity initialization without additional binds."""
@@ -104,7 +104,7 @@ def test_init_singularity_no_bind(self, basic_params):
104104
assert generator.use_singularity
105105
assert not generator.is_multinode
106106
assert generator.additional_binds == ""
107-
assert generator.model_weights_path == "/path/to/model_wieghts/test-model"
107+
assert generator.model_weights_path == "/path/to/model_weights/test-model"
108108

109109
def test_generate_shebang_single_node(self, basic_params):
110110
"""Test shebang generation for single-node setup."""
@@ -161,7 +161,7 @@ def test_generate_launch_cmd_venv(self, basic_params):
161161
launch_cmd = generator._generate_launch_cmd()
162162

163163
assert "source /path/to/venv/bin/activate" in launch_cmd
164-
assert "vllm serve /path/to/model_wieghts/test-model" in launch_cmd
164+
assert "vllm serve /path/to/model_weights/test-model" in launch_cmd
165165
assert "--served-model-name test-model" in launch_cmd
166166
assert "--tensor-parallel-size 4" in launch_cmd
167167
assert "--max-model-len 8192" in launch_cmd
@@ -173,7 +173,7 @@ def test_generate_launch_cmd_singularity(self, singularity_params):
173173
launch_cmd = generator._generate_launch_cmd()
174174

175175
assert "singularity exec --nv" in launch_cmd
176-
assert "--bind /path/to/model_wieghts/test-model" in launch_cmd
176+
assert "--bind /path/to/model_weights/test-model" in launch_cmd
177177
assert "--bind /scratch:/scratch,/data:/data" in launch_cmd
178178
assert "source" not in launch_cmd
179179

0 commit comments

Comments
 (0)