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

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

Author: pre-commit-ci[bot]

tests/e2e/default_single_streaming.py

@@ -1,10 +1,9 @@
 from litserve import LitServer, LitAPI
-import time
-import numpy as np
+
 
 class SimpleStreamingAPI(LitAPI):
     def setup(self, device) -> None:
-        self.model = lambda x,y: x*y
+        self.model = lambda x, y: x * y
 
     def decode_request(self, request):
         return request["input"]
@@ -21,4 +20,4 @@ def encode_response(self, output_stream):
 if __name__ == "__main__":
     api = SimpleStreamingAPI()
     server = LitServer(api, stream=True)
-    server.run(port=8000)
+    server.run(port=8000)

tests/e2e/test_e2e.py

@@ -297,7 +297,6 @@ def test_openai_parity_with_response_format():
         )
 
 
-
 @e2e_from_file("tests/e2e/default_single_streaming.py")
 def test_e2e_single_streaming():
     resp = requests.post("http://127.0.0.1:8000/predict", json={"input": 4.0}, headers=None, stream=True)
@@ -306,7 +305,7 @@ def test_e2e_single_streaming():
     outputs = []
     for line in resp.iter_lines():
         if line:
-            outputs.append(json.loads(line.decode('utf-8')))
+            outputs.append(json.loads(line.decode("utf-8")))
 
     assert len(outputs) > 1, "Expected multiple streamed outputs"
     assert len(outputs) == 3, "Expected 3 streamed outputs"
@@ -316,4 +315,4 @@ def test_e2e_single_streaming():
     for i, output in enumerate(outputs[:-1]):
         assert 0 < output["output"] <= 12.0, f"Intermediate output {i} is not within expected range"
         if i > 0:
-            assert output["output"] >= outputs[i-1]["output"], "Outputs are not monotonically increasing"
+            assert output["output"] >= outputs[i - 1]["output"], "Outputs are not monotonically increasing"