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

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

Author: pre-commit-ci[bot]

tests/vec_inf/cli/test_cli.py

@@ -330,9 +330,7 @@ def test_launch_command_no_model_weights_parent_dir(runner, debug_helper, base_p
             stack.enter_context(patch_obj)
 
         # Mock load_config to return empty list
-        stack.enter_context(
-            patch("vec_inf.client._utils.load_config", return_value=[])
-        )
+        stack.enter_context(patch("vec_inf.client._utils.load_config", return_value=[]))
 
         result = runner.invoke(cli, ["launch", "test-model"])
         debug_helper.print_debug_info(result)
@@ -376,9 +374,7 @@ def test_launch_command_model_not_in_config_with_weights(
         )
 
 
-def test_launch_command_model_not_found(
-    runner, debug_helper, test_paths, base_patches
-):
+def test_launch_command_model_not_found(runner, debug_helper, test_paths, base_patches):
     """Test handling of a model that's neither in config nor has weights."""
 
     def custom_path_exists(p):
@@ -410,8 +406,7 @@ def custom_path_exists(p):
         assert result.exit_code == 1
         assert (
             "'unknown-model' not found in configuration and model weights "
-            "not found at expected path '/model-weights/unknown-model'"
-            in result.output
+            "not found at expected path '/model-weights/unknown-model'" in result.output
         )
 
 
@@ -450,10 +445,7 @@ def test_metrics_command_pending_server(
 
         assert result.exit_code == 0
         assert "ERROR" in result.output
-        assert (
-            "Pending resources for server initialization"
-            in result.output
-        )
+        assert "Pending resources for server initialization" in result.output
 
 
 def test_metrics_command_server_not_ready(
@@ -473,10 +465,7 @@ def test_metrics_command_server_not_ready(
 
         assert result.exit_code == 0
         assert "ERROR" in result.output
-        assert (
-            "Server not ready"
-            in result.output
-        )
+        assert "Server not ready" in result.output
 
 
 @patch("requests.get")

vec_inf/cli/_cli.py

@@ -134,7 +134,9 @@ def launch(
     """Launch a model on the cluster."""
     try:
         # Convert cli_kwargs to LaunchOptions
-        launch_options = LaunchOptions(**{k: v for k, v in cli_kwargs.items() if k != "json_mode"})
+        launch_options = LaunchOptions(
+            **{k: v for k, v in cli_kwargs.items() if k != "json_mode"}
+        )
 
         # Start the client and launch model inference server
         client = VecInfClient()

vec_inf/cli/_helper.py

@@ -15,7 +15,7 @@
 from vec_inf.client._models import ModelInfo, StatusResponse
 
 
-class LaunchResponseFormatter():
+class LaunchResponseFormatter:
     """CLI Helper class for formatting LaunchResponse."""
 
     def __init__(self, model_name: str, params: dict[str, Any]):
@@ -68,7 +68,7 @@ def format_table_output(self) -> Table:
         return table
 
 
-class StatusResponseFormatter():
+class StatusResponseFormatter:
     """CLI Helper class for formatting StatusResponse."""
 
     def __init__(self, status_info: StatusResponse):
@@ -102,7 +102,7 @@ def output_table(self) -> Table:
         return table
 
 
-class MetricsResponseFormatter():
+class MetricsResponseFormatter:
     """CLI Helper class for formatting MetricsResponse."""
 
     def __init__(self, metrics: dict[str, float]):
@@ -189,7 +189,7 @@ def format_metrics(self) -> None:
         )
 
 
-class ListCmdDisplay():
+class ListCmdDisplay:
     """CLI Helper class for displaying model listing functionality."""
 
     def __init__(self, console: Console, json_mode: bool = False):
@@ -218,7 +218,9 @@ def _format_single_model_output(
                 table.add_row(field, str(value))
         return table
 
-    def _format_all_models_output(self, model_infos: list[ModelInfo]) -> Union[list[str], list[Panel]]:
+    def _format_all_models_output(
+        self, model_infos: list[ModelInfo]
+    ) -> Union[list[str], list[Panel]]:
         """Format output table for all models."""
         # Sort by model type priority
         sorted_model_infos = sorted(
@@ -254,4 +256,3 @@ def display_all_models_output(self, model_infos: list[ModelInfo]) -> None:
         else:
             panels = self._format_all_models_output(model_infos)
             self.console.print(Columns(panels, equal=True))
-

vec_inf/client/_helper.py

@@ -74,7 +74,9 @@ def _get_model_configuration(self) -> ModelConfig:
         )
 
         if not model_weights_parent_dir:
-            raise ModelNotFoundError("Could not determine model weights parent directory")
+            raise ModelNotFoundError(
+                "Could not determine model weights parent directory"
+            )
 
         model_weights_path = Path(model_weights_parent_dir, self.model_name)
 
@@ -284,7 +286,9 @@ def _process_running_state(self) -> None:
         )
 
         if isinstance(server_status, tuple):
-            self.status_info.server_status, self.status_info.failed_reason = server_status
+            self.status_info.server_status, self.status_info.failed_reason = (
+                server_status
+            )
             return
 
         if server_status == "RUNNING":

vec_inf/client/_utils.py

@@ -153,8 +153,9 @@ def load_config() -> list[ModelConfig]:
                         config.setdefault("models", {})[name] = data
         else:
             warnings.warn(
-                f"WARNING: Could not find user config: {user_path}, revert to default config located at {default_path}", UserWarning,
-                stacklevel=2
+                f"WARNING: Could not find user config: {user_path}, revert to default config located at {default_path}",
+                UserWarning,
+                stacklevel=2,
             )
 
     return [

vec_inf/client/api.py

@@ -182,7 +182,6 @@ def shutdown_model(self, slurm_job_id: int) -> bool:
             raise SlurmJobError(f"Failed to shutdown model: {stderr}")
         return True
 
-
     def wait_until_ready(
         self,
         slurm_job_id: int,