Adapted to pytest framework

Author: Menglths

test/common/llmperf/run_inference.py

@@ -5,7 +5,6 @@
 from typing import Any, Dict, List
 
 import yaml
-
 from common.llmperf.utils.token_benchmark import run_token_benchmark
 from common.llmperf.utils.utils import reset_prefill_cache
 

test/common/llmperf/utils/openai_chat_completions_client.py

@@ -4,7 +4,6 @@
 from typing import Any, Dict, Tuple
 
 import requests
-
 from common.llmperf.utils import common_metrics
 from common.llmperf.utils.models import RequestConfig
 

test/common/llmperf/utils/token_benchmark.py

@@ -9,15 +9,17 @@
 from typing import Any, Dict, List, Optional, Tuple
 
 import pandas as pd
-from transformers import AutoTokenizer
-
 from common.llmperf.utils import common_metrics
 from common.llmperf.utils.models import RequestConfig
-from common.llmperf.utils.openai_chat_completions_client import \
-    OpenAIChatCompletionsClient
-from common.llmperf.utils.utils import (LLMPerfResults,
-                                        randomly_sample_sonnet_lines_prompt,
-                                        sample_random_positive_int)
+from common.llmperf.utils.openai_chat_completions_client import (
+    OpenAIChatCompletionsClient,
+)
+from common.llmperf.utils.utils import (
+    LLMPerfResults,
+    randomly_sample_sonnet_lines_prompt,
+    sample_random_positive_int,
+)
+from transformers import AutoTokenizer
 
 
 def get_token_throughput_latencies(

test/suites/E2E/test_demo_function.py

@@ -63,4 +63,4 @@ def test_capture_mix():
 def test_config():
     assert (
         config_instance.get_nested_config("database.host", "localhost") == "127.0.0.1"
-    )
+    )

test/suites/E2E/test_uc_performance.py

@@ -1,8 +1,6 @@
 import pytest
-
-from common.llmperf.run_inference import inference_results
-
 from common.capture_utils import export_vars
+from common.llmperf.run_inference import inference_results
 
 
 @pytest.mark.feature("uc_performance_test")