Format code with ruff

This commit is contained in:
Andy Lee
2025-12-24 00:23:25 +00:00
parent b754474c44
commit a0e53ef8f1
4 changed files with 68 additions and 8 deletions

View File

@@ -314,7 +314,9 @@ class WeChatHistoryReader(BaseReader):
return concatenated_groups
def _create_concatenated_content(self, message_group: dict, contact_name: str) -> tuple[str, str]:
def _create_concatenated_content(
self, message_group: dict, contact_name: str
) -> tuple[str, str]:
"""
Create concatenated content from a group of messages.

View File

@@ -113,7 +113,8 @@ def load_vidore_v2_data(
# Try to get a sample to see actual language values
try:
sample_ds = cast(
Dataset, load_dataset(dataset_path, "queries", split=split, revision=revision)
Dataset,
load_dataset(dataset_path, "queries", split=split, revision=revision),
)
if len(sample_ds) > 0 and "language" in sample_ds.column_names:
sample_langs = set(sample_ds["language"])