From 9f05293efbcb0d7686da75c74f9395ffb9e2d704 Mon Sep 17 00:00:00 2001 From: Mikhail Khludnev Date: Tue, 13 Feb 2024 23:43:56 +0300 Subject: [PATCH] ruff --- libs/partners/nvidia-trt/tests/unit_tests/test_llms.py | 5 +++-- 1 file changed, 3 insertions(+), 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 index c92c0c7d8363a..8ef80cdea4688 100644 --- 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