fixed mypy error

scarere · scarere · commit f660a0231332 · 2025-08-26T15:51:39.000-04:00
diff --git a/vec_inf/client/_slurm_script_generator.py b/vec_inf/client/_slurm_script_generator.py
@@ -39,7 +39,7 @@ def __init__(self, params: dict[str, Any]):
         self.model_weights_path = str(
             Path(self.params["model_weights_parent_dir"], self.params["model_name"])
         )
-        env_dict: dict = self.params.get("env", {})
+        env_dict: dict[str, str] = self.params.get("env", {})
         # Create string of environment variables
         self.env_str = ""
         for key, val in env_dict.items():

Original file line number	Diff line number	Diff line change
`@@ -39,7 +39,7 @@ def __init__(self, params: dict[str, Any]):`
`39`	`39`	`self.model_weights_path = str(`
`40`	`40`	`Path(self.params["model_weights_parent_dir"], self.params["model_name"])`
`41`	`41`	`)`
`42`		`- env_dict: dict = self.params.get("env", {})`
	`42`	`+ env_dict: dict[str, str] = self.params.get("env", {})`
`43`	`43`	`# Create string of environment variables`
`44`	`44`	`self.env_str = ""`
`45`	`45`	`for key, val in env_dict.items():`