Fix linter errors

Author: jaredoconnell

src/guidellm/__main__.py

@@ -7,7 +7,11 @@
 import click
 
 from guidellm.backend import BackendType
-from guidellm.benchmark import ProfileType, benchmark_generative_text, display_benchmarks_report
+from guidellm.benchmark import (
+    ProfileType,
+    benchmark_generative_text,
+    display_benchmarks_report,
+)
 from guidellm.config import print_config
 from guidellm.preprocess.dataset import ShortPromptStrategy, process_dataset
 from guidellm.scheduler import StrategyType

src/guidellm/benchmark/__init__.py

@@ -62,6 +62,6 @@
     "SynchronousProfile",
     "ThroughputProfile",
     "benchmark_generative_text",
-    "display_benchmarks_report",
     "create_profile",
+    "display_benchmarks_report",
 ]

src/guidellm/benchmark/entrypoints.py

@@ -1,4 +1,3 @@
-import os
 from collections.abc import Iterable
 from pathlib import Path
 from typing import Any, Literal, Optional, Union
@@ -136,11 +135,11 @@ async def benchmark_generative_text(
 
     return report, saved_path
 
-def display_benchmarks_report(file: str):
+def display_benchmarks_report(file: Path):
     console = GenerativeBenchmarksConsole(enabled=True)
-    if not os.path.exists(file):
+    if not file.exists():
         console.print_line(f"File {file} not found.")
         return
     report = GenerativeBenchmarksReport.load_file(file)
     console.benchmarks = report.benchmarks
-    console.print_full_report()
+    console.print_full_report()

src/guidellm/benchmark/output.py

@@ -960,4 +960,4 @@ def print_full_report(self):
         self.print_benchmarks_metadata()
         self.print_benchmarks_info()
         self.print_benchmarks_stats()
-        self.enabled = orig_enabled
+        self.enabled = orig_enabled

tests/unit/entrypoints/test_display_entrypoint.py

@@ -1,34 +1,44 @@
-import os
 import unittest
+from pathlib import Path
+
 import pytest
 
 from guidellm.benchmark import display_benchmarks_report
 
-@pytest.fixture()
+
+@pytest.fixture
 def get_test_asset_dir():
-    def _() -> str:
-        return os.path.dirname(os.path.abspath(__file__)) + "/assets"
+    def _() -> Path:
+        return Path(__file__).parent / "assets"
 
     return _
 
 
 def test_display_entrypoint_json(capfd, get_test_asset_dir):
-    generic_test_display_entrypoint("benchmarks_stripped.json", capfd, get_test_asset_dir)
+    generic_test_display_entrypoint(
+        "benchmarks_stripped.json",
+        capfd,
+        get_test_asset_dir,
+    )
 
 
 def test_display_entrypoint_yaml(capfd, get_test_asset_dir):
-    generic_test_display_entrypoint("benchmarks_stripped.yaml", capfd, get_test_asset_dir)
+    generic_test_display_entrypoint(
+        "benchmarks_stripped.yaml",
+        capfd,
+        get_test_asset_dir,
+    )
 
 
 def generic_test_display_entrypoint(filename, capfd, get_test_asset_dir):
     asset_dir = get_test_asset_dir()
-    display_benchmarks_report(asset_dir + "/" + filename)
+    display_benchmarks_report(asset_dir / filename)
     out, err = capfd.readouterr()
-    expected_output_path = asset_dir + "/benchmarks_stripped_output.txt"
-    with open(expected_output_path, 'r', encoding='utf_8') as file:
+    expected_output_path = asset_dir / "benchmarks_stripped_output.txt"
+    with expected_output_path.open(encoding="utf_8") as file:
         expected_output = file.read()
     assert out == expected_output
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()