Skip to content

Commit 8e66e26

Browse files
committed
ruff format
1 parent b4037e6 commit 8e66e26

1 file changed

Lines changed: 2 additions & 6 deletions

File tree

tests/vec_inf/client/test_slurm_script_generator.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -186,9 +186,7 @@ def test_generate_server_setup_singularity(self, singularity_params):
186186
"module load " in setup or "apptainer" in setup.lower()
187187
) # Remove module name since it's inconsistent between clusters
188188

189-
def test_generate_server_setup_singularity_no_weights(
190-
self, singularity_params
191-
):
189+
def test_generate_server_setup_singularity_no_weights(self, singularity_params):
192190
"""Test server setup when using hf_model (no local weights in bind path)."""
193191
params = singularity_params.copy()
194192
params["hf_model"] = "test-org/test-model"
@@ -345,9 +343,7 @@ def test_generate_launch_cmd_singularity(self, singularity_params):
345343
assert "apptainer exec --nv" in launch_cmd
346344
assert "source" not in launch_cmd
347345

348-
def test_generate_launch_cmd_singularity_no_local_weights(
349-
self, singularity_params
350-
):
346+
def test_generate_launch_cmd_singularity_no_local_weights(self, singularity_params):
351347
"""Test container launch when using hf_model instead of local weights."""
352348
params = singularity_params.copy()
353349
params["hf_model"] = "test-org/test-model"

0 commit comments

Comments
 (0)