fixed code quality issues. fixed tests

Author: Dmytro Parfeniuk

src/guidellm/scheduler/base.py

@@ -36,7 +36,9 @@ class SchedulerResult:
     :param current_result: The result of the current request, if any.
     :type current_result: Optional[Union[TextGenerationResult, Exception]]
     :param batch_results: The result of the current batch of requests, if any
-    :type batch_results: Optional[List[Union[TextGenerationResult, TextGenerationError]]]
+    :type batch_results: Optional[List[
+                            Union[TextGenerationResult, TextGenerationError]]
+                         ]
     """
 
     completed: bool

tests/unit/test_main.py

@@ -252,6 +252,7 @@ def test_generate_benchmark_report_invoke_smoke(
         max_requests=10,
         output_path="benchmark_report.json",
         cont_refresh_table=False,
+        batch_size=None,
     )
     assert report is not None
 
@@ -308,6 +309,7 @@ def test_generate_benchmark_report_emulated_with_dataset_requests(
             rate=None,
             max_seconds=10,
             max_requests="dataset",
+            batch_size=None,
             output_path="benchmark_report.json",
             cont_refresh_table=False,
         )
@@ -397,6 +399,7 @@ def test_generate_benchmark_report_openai_limited_by_file_dataset(
             rate=rate,
             max_seconds=None,
             max_requests="dataset",
+            batch_size=None,
             output_path="benchmark_report.json",
             cont_refresh_table=False,
         )