Fix: pre-commit errors on main (#574)

* fix: tests/unit/test_batch.py:173:5: PT031 `pytest.warns()` block should contain a single simple statement
* fix: refactor test_server_terminate to improve readability of patch context
* rem comment
* update

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: bhimrazy

tests/unit/test_batch.py

@@ -170,12 +170,12 @@ def test_batch_predict_string_warning():
 
     mock_input = torch.tensor([[1.0], [2.0]])
 
+    # Simulate the behavior in run_batched_loop
+    y = api.predict(mock_input)
     with pytest.warns(
         UserWarning,
         match="When batching is enabled, 'predict' must return a list to handle multiple inputs correctly.",
     ):
-        # Simulate the behavior in run_batched_loop
-        y = api.predict(mock_input)
         api.unbatch(y)
 
 

tests/unit/test_lit_server.py

@@ -317,11 +317,11 @@ def test_server_terminate():
     server.verify_worker_status = MagicMock()
     server._transport = MagicMock()
 
-    with (
-        patch("litserve.server.LitServer._init_manager", return_value=MagicMock()) as mock_init_manager,
-        patch("litserve.server.LitServer._start_server", side_effect=Exception("mocked error")) as mock_start,
-        patch("litserve.server.LitServer.launch_inference_worker", return_value=([MagicMock()])) as mock_launch,
-    ):
+    with patch("litserve.server.LitServer._init_manager", return_value=MagicMock()) as mock_init_manager, patch(
+        "litserve.server.LitServer._start_server", side_effect=Exception("mocked error")
+    ) as mock_start, patch(
+        "litserve.server.LitServer.launch_inference_worker", return_value=([MagicMock()])
+    ) as mock_launch:
         with pytest.raises(Exception, match="mocked error"):
             server.run(port=8001)