[pre-commit.ci] Add auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

vec_inf/client/_client_vars.py

@@ -67,13 +67,14 @@
     "error": "err_file",
 }
 
-# vLLM engine args mapping between short and long names 
+# vLLM engine args mapping between short and long names
 VLLM_SHORT_TO_LONG_MAP = {
     "-tp": "--tensor-parallel-size",
     "-pp": "--pipeline-parallel-size",
     "-O": "--compilation-config",
 }
 
+
 # Slurm script templates
 class ShebangConfig(TypedDict):
     """TypedDict for SLURM script shebang configuration.

vec_inf/client/_helper.py

@@ -160,12 +160,11 @@ def _process_vllm_args(self, arg_string: str) -> dict[str, Any]:
                 if key.strip() in VLLM_SHORT_TO_LONG_MAP:
                     key = VLLM_SHORT_TO_LONG_MAP[key.strip()]
                 vllm_args[key.strip()] = value.strip()
+            elif "-O" in arg.strip():
+                key = VLLM_SHORT_TO_LONG_MAP["-O"]
+                vllm_args[key] = arg.strip()[2:].strip()
             else:
-                if "-O" in arg.strip():
-                    key = VLLM_SHORT_TO_LONG_MAP["-O"]
-                    vllm_args[key] = arg.strip()[2:].strip()
-                else:
-                    vllm_args[arg.strip()] = True
+                vllm_args[arg.strip()] = True
         return vllm_args
 
     def _get_launch_params(self) -> dict[str, Any]: