@@ -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