Fix batch launch script generation for bind path, mypy and ruff fixes

Author: XkunW

vec_inf/client/_helper.py

@@ -554,7 +554,7 @@ def _get_launch_params(
                     raise ValueError(
                         f"Mismatch found for {arg}: {params[arg]} != {params['models'][model_name][arg]}, check your configuration"
                     )
-            # Check for required fields without default vals, will raise an error if missing
+            # Check for required fields, will raise an error if missing any
             utils.check_required_fields(
                 {**params["models"][model_name], **common_params}
             )

vec_inf/client/_slurm_script_generator.py

@@ -219,7 +219,9 @@ def __init__(self, params: dict[str, Any]):
         self.use_container = self.params["venv"] == CONTAINER_MODULE_NAME
         for model_name in self.params["models"]:
             self.params["models"][model_name]["additional_binds"] = (
-                f",{self.params['bind']}" if self.params.get("bind") else ""
+                f",{self.params['models'][model_name]['bind']}"
+                if self.params["models"][model_name].get("bind")
+                else ""
             )
             self.params["models"][model_name]["model_weights_path"] = str(
                 Path(

vec_inf/client/_slurm_templates.py

@@ -215,7 +215,7 @@ class BatchModelLaunchScriptTemplate(TypedDict):
         Shebang line for the script
     container_setup : list[str]
         Commands for container setup
-    bind_path : list[str]
+    bind_path : str
         Bind path environment variable for the container
     server_address_setup : list[str]
         Commands to setup the server address
@@ -227,7 +227,7 @@ class BatchModelLaunchScriptTemplate(TypedDict):
 
     shebang: str
     container_setup: str
-    bind_path: list[str]
+    bind_path: str
     server_address_setup: list[str]
     write_to_json: list[str]
     launch_cmd: list[str]