Implement previously missed typechecks

Signed-off-by: Samuel Monson <[email protected]>

Author: sjmonson

src/guidellm/utils/cli.py

@@ -35,7 +35,7 @@ def __init__(self, *types: click.ParamType):
         self.types = types
         self.name = "".join(t.name for t in types)
 
-    def convert(self, value, param, ctx):
+    def convert(self, value, param, ctx):  # noqa: RET503
         fails = []
         for t in self.types:
             try:

tests/unit/preprocess/test_dataset.py

@@ -133,8 +133,6 @@ def test_process_dataset_non_empty(
     mock_save_to_file,
     tokenizer_mock,
 ):
-    from guidellm.preprocess.dataset import process_dataset
-
     mock_dataset = [{"prompt": "Hello"}, {"prompt": "How are you?"}]
     mock_load_dataset.return_value = (mock_dataset, {"prompt_column": "prompt"})
     mock_check_processor.return_value = tokenizer_mock

tests/unit/test_cli.py

@@ -2,6 +2,9 @@
 Unit tests for CLI functionality, specifically the version flag.
 """
 
+import importlib.metadata
+import re
+
 import pytest
 from click.testing import CliRunner
 
@@ -26,7 +29,6 @@ def test_version_flag_displays_actual_version():
     result = runner.invoke(cli, ["--version"])
 
     assert result.exit_code == 0
-    import re
 
     version_pattern = r"guidellm version: \d+\.\d+"
     assert re.search(version_pattern, result.output)
@@ -84,8 +86,6 @@ def test_version_flag_case_sensitivity():
 @pytest.mark.integration
 def test_version_integration_with_actual_version():
     """Integration test to verify version matches importlib.metadata."""
-    import importlib.metadata
-
     try:
         actual_version = importlib.metadata.version("guidellm")