Skip to content

Commit 9f4a899

Browse files
[pre-commit.ci] Add auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent afc29de commit 9f4a899

File tree

3 files changed

+7
-6
lines changed

3 files changed

+7
-6
lines changed

tests/vec_inf/client/test_helper.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -124,9 +124,7 @@ def test_process_env_vars(self, mock_path_exists, mock_load_config, mock_configs
124124
file_path = Path(__file__).parent / "test_vars.env"
125125

126126
launcher = ModelLauncher("test-model", {})
127-
env_vars = launcher._process_env_vars(
128-
f"CACHE_DIR=/cache,{file_path}"
129-
)
127+
env_vars = launcher._process_env_vars(f"CACHE_DIR=/cache,{file_path}")
130128
assert env_vars["CACHE_DIR"] == "/cache"
131129
assert env_vars["MY_VAR"] == "5"
132130
assert env_vars["VLLM_CACHE_ROOT"] == "/cache/vllm"

tests/vec_inf/client/test_slurm_script_generator.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def singularity_params(self, basic_params):
5959
"CACHE_DIR": "/cache",
6060
"MY_VAR": "5",
6161
"VLLM_CACHE_ROOT": "/cache/vllm",
62-
}
62+
},
6363
}
6464
)
6565
return singularity
@@ -101,7 +101,10 @@ def test_init_singularity(self, singularity_params):
101101
assert not generator.is_multinode
102102
assert generator.additional_binds == " --bind /scratch:/scratch,/data:/data"
103103
assert generator.model_weights_path == "/path/to/model_weights/test-model"
104-
assert generator.env_str == "--env CACHE_DIR=/cache,MY_VAR=5,VLLM_CACHE_ROOT=/cache/vllm"
104+
assert (
105+
generator.env_str
106+
== "--env CACHE_DIR=/cache,MY_VAR=5,VLLM_CACHE_ROOT=/cache/vllm"
107+
)
105108

106109
def test_init_singularity_no_bind(self, basic_params):
107110
"""Test Singularity initialization without additional binds."""

tests/vec_inf/client/test_vars.env

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
11
MY_VAR=5
2-
VLLM_CACHE_ROOT=/cache/vllm
2+
VLLM_CACHE_ROOT=/cache/vllm

0 commit comments

Comments
 (0)