Resolve ruff complaints

promptwright/dataset.py

@@ -1,6 +1,7 @@
 import json
 import re
 
+
 class Dataset:
     """
     A class to represent a dataset consisting of samples, where each sample contains messages with specific roles.

promptwright/hf_hub.py

@@ -2,6 +2,7 @@
 from huggingface_hub import login
 from huggingface_hub.utils import HfHubHTTPError, RepositoryNotFoundError
 
+
 class HFUploader:
     """
     HFUploader is a class for uploading datasets to the Hugging Face Hub.

promptwright/topic_tree.py

@@ -28,8 +28,7 @@ def validate_and_clean_response(response_text: str) -> str | list[str] | None:
         topics = extract_list(response_text)
         if topics:
             return [topic.strip() for topic in topics if topic.strip()]
-
-        return None
+        return None  # noqa: TRY300
     except (json.JSONDecodeError, ValueError) as e:
         print(f"Error parsing response: {str(e)}")
         return None

tests/test_engine.py

@@ -1,7 +1,9 @@
-import pytest
 from unittest.mock import MagicMock, patch
-import re
-from promptwright.engine import DataEngine, EngineArguments, Dataset, TopicTree
+
+import pytest
+
+from promptwright.engine import DataEngine, Dataset, EngineArguments
+
 
 @pytest.fixture
 def engine_args():