ruff

langchain-ai · Mar 3, 2024 · 9f05293 · 9f05293
1 parent c6a82a4
commit 9f05293
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/libs/partners/nvidia-trt/tests/unit_tests/test_llms.py b/libs/partners/nvidia-trt/tests/unit_tests/test_llms.py
@@ -11,7 +11,7 @@ def test_initialization() -> None:
     TritonTensorRTLLM(model_name="ensemble", server_url="http://localhost:8001")
 
 
-@patch('tritonclient.grpc.service_pb2_grpc.GRPCInferenceServiceStub')
+@patch("tritonclient.grpc.service_pb2_grpc.GRPCInferenceServiceStub")
 def test_default_verbose(ignore) -> None:
     llm = TritonTensorRTLLM(server_url="http://localhost:8001", model_name="ensemble")
     captured = StringIO()
@@ -20,7 +20,8 @@ def test_default_verbose(ignore) -> None:
     sys.stdout = sys.__stdout__
     assert "is_server_live" not in captured.getvalue()
 
-@patch('tritonclient.grpc.service_pb2_grpc.GRPCInferenceServiceStub')
+
+@patch("tritonclient.grpc.service_pb2_grpc.GRPCInferenceServiceStub")
 def test_verbose(ignore) -> None:
     llm = TritonTensorRTLLM(
         server_url="http://localhost:8001", model_name="ensemble", verbose_client=True