[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/_slurm_script_generator.py

@@ -44,25 +44,24 @@ def __init__(self, params: dict[str, Any]):
 
     def _generate_env_str(self) -> str:
         """Generate the environment variables string for the Slurm script.
-        
+
         Returns
         -------
         str
             Formatted environment variables string for container or shell export commands.
         """
         env_dict: dict[str, str] = self.params.get("env", {})
-        
+
         if not env_dict:
             return ""
-        
+
         if self.use_container:
             # Format for container: --env KEY1=VAL1,KEY2=VAL2
             env_pairs = [f"{key}={val}" for key, val in env_dict.items()]
             return f"--env {','.join(env_pairs)}"
-        else:
-            # Format for shell: export KEY1=VAL1\nexport KEY2=VAL2
-            export_lines = [f"export {key}={val}" for key, val in env_dict.items()]
-            return "\n".join(export_lines)
+        # Format for shell: export KEY1=VAL1\nexport KEY2=VAL2
+        export_lines = [f"export {key}={val}" for key, val in env_dict.items()]
+        return "\n".join(export_lines)
 
     def _generate_script_content(self) -> str:
         """Generate the complete Slurm script content.

vec_inf/client/_slurm_templates.py

@@ -114,19 +114,19 @@ class SlurmScriptTemplate(TypedDict):
             'head_node_ip=$(srun --nodes=1 --ntasks=1 -w "$head_node" hostname --ip-address)',
             "\n# Check for RDMA devices and set environment variable accordingly",
             "if ! command -v ibv_devices >/dev/null 2>&1; then",
-            "   echo \"ibv_devices not found; forcing TCP. (No RDMA userland on host?)\"",
+            '   echo "ibv_devices not found; forcing TCP. (No RDMA userland on host?)"',
             "   export NCCL_IB_DISABLE=1",
-            "   export NCCL_ENV_ARG=\"--env NCCL_IB_DISABLE=1\"",
+            '   export NCCL_ENV_ARG="--env NCCL_IB_DISABLE=1"',
             "else",
             "   # Pick GID index based on link layer (IB vs RoCE)",
-            "   if ibv_devinfo 2>/dev/null | grep -q \"link_layer:.*Ethernet\"; then",
+            '   if ibv_devinfo 2>/dev/null | grep -q "link_layer:.*Ethernet"; then',
             "       # RoCEv2 typically needs a nonzero GID index; 3 is common, try 2 if your fabric uses it",
             "       export NCCL_IB_GID_INDEX={{NCCL_IB_GID_INDEX:-3}}",
-            "       export NCCL_ENV_ARG=\"--env NCCL_IB_GID_INDEX={{NCCL_IB_GID_INDEX:-3}}\"",
+            '       export NCCL_ENV_ARG="--env NCCL_IB_GID_INDEX={{NCCL_IB_GID_INDEX:-3}}"',
             "   else",
             "       # Native InfiniBand => GID 0",
             "       export NCCL_IB_GID_INDEX={{NCCL_IB_GID_INDEX:-0}}",
-            "       export NCCL_ENV_ARG=\"--env NCCL_IB_GID_INDEX={{NCCL_IB_GID_INDEX:-0}}\"",
+            '       export NCCL_ENV_ARG="--env NCCL_IB_GID_INDEX={{NCCL_IB_GID_INDEX:-0}}"',
             "   fi",
             "fi",
             "\n# Start Ray head node",