ruff and mypy fixes

Author: XkunW

tests/test_imports.py

@@ -23,11 +23,11 @@ def test_imports(self):
             import vec_inf.client._helper
             import vec_inf.client._slurm_script_generator
             import vec_inf.client._slurm_templates
-            import vec_inf.client._slurm_vars  # noqa: F401
+            import vec_inf.client._slurm_vars
             import vec_inf.client._utils
             import vec_inf.client.api
             import vec_inf.client.config
-            import vec_inf.client.models
+            import vec_inf.client.models  # noqa: F401
 
         except ImportError as e:
             pytest.fail(f"Import failed: {e}")

vec_inf/client/_slurm_vars.py

@@ -66,10 +66,8 @@ def load_yaml_config(path: Path) -> dict[str, Any]:
 MAX_CPUS_PER_TASK = _config["limits"]["max_cpus_per_task"]
 
 # Create dynamic Literal types
-QOS: TypeAlias = str  # Runtime validation will handle the actual values
-PARTITION: TypeAlias = str  # Runtime validation will handle the actual values
-QOS = Literal[tuple(_config["allowed_values"]["qos"])]
-PARTITION = Literal[tuple(_config["allowed_values"]["partition"])]
+QOS = Literal[tuple(_config["allowed_values"]["qos"])]  # type: ignore[valid-type]
+PARTITION = Literal[tuple(_config["allowed_values"]["partition"])]  # type: ignore[valid-type]
 
 # Extract default arguments
 DEFAULT_ARGS = _config["default_args"]

vec_inf/client/_utils.py

@@ -223,7 +223,6 @@ def process_config(config: dict[str, Any]) -> list[ModelConfig]:
             ModelConfig(model_name=name, **model_data)
             for name, model_data in config.get("models", {}).items()
         ]
-        
 
     def update_config(
         config: dict[str, Any], user_config: dict[str, Any]