Apply Ruff

yourbench_space/config.py

@@ -1,5 +1,5 @@
-from ruamel.yaml import YAML
 from loguru import logger
+from ruamel.yaml import YAML
 
 from yourbench_space import PATH
 from yourbench_space.utils import to_commentable_yaml
@@ -92,11 +92,15 @@ def save_yaml_file(config: dict, path: str):
 
     # Now we can add inline comments
     ingestion = config_cm["pipeline"]["ingestion"]
-    ingestion.yaml_set_comment_before_after_key("source_documents_dir", before="⚠️ Change this path to match your local directory")
+    ingestion.yaml_set_comment_before_after_key(
+        "source_documents_dir", before="⚠️ Change this path to match your local directory"
+    )
     ingestion.yaml_set_comment_before_after_key("output_dir", before="⚠️ This is where ingested data will be saved")
 
     upload = config_cm["pipeline"]["upload_ingest_to_hub"]
-    upload.yaml_set_comment_before_after_key("source_documents_dir", before="⚠️ Same as output_dir from ingestion — adjust as needed")
+    upload.yaml_set_comment_before_after_key(
+        "source_documents_dir", before="⚠️ Same as output_dir from ingestion — adjust as needed"
+    )
 
     with open(path, "w") as file:
         yaml.dump(config_cm, file)

yourbench_space/evaluation.py

@@ -18,7 +18,7 @@ def create_eval_file(eval_ds_name: str):
     if os.environ.get("SYSTEM") == "spaces":
         template_path = Path("/home/user/app/yourbench_space/lighteval_task/yourbench_task.py")
     else:
-         template_path = Path("yourbench_space/lighteval_task/yourbench_task.py")
+        template_path = Path("yourbench_space/lighteval_task/yourbench_task.py")
 
     subprocess.run(["lighteval", "tasks", "create", str(template_path), task_name, eval_ds_name])
 

yourbench_space/leaderboard_space/env.py

@@ -12,4 +12,4 @@ MODELS = [m[0] for m in INIT_MODELS]
 TASK = os.getenv("TASK")
 # With storage
 HF_TOKEN = os.getenv("HF_TOKEN")
-ORG_NAME = os.getenv("ORG_NAME")
+ORG_NAME = os.getenv("ORG_NAME")

yourbench_space/utils.py

@@ -5,10 +5,10 @@ import shutil
 import pathlib
 import subprocess
 from typing import List, Union, Optional
-from ruamel.yaml.comments import CommentedMap, CommentedSeq
 
 import pandas as pd
 from loguru import logger
+from ruamel.yaml.comments import CommentedMap, CommentedSeq
 
 import gradio as gr
 from datasets import load_dataset
@@ -35,6 +35,7 @@ STAGE_DISPLAY_MAP = {
     "lighteval": "Generate Lighteval Subset",
 }
 
+
 def to_commentable_yaml(obj):
     """
     Recursively converts standard Python dicts and lists into
@@ -44,14 +45,15 @@ def to_commentable_yaml(obj):
     # Convert dict to CommentedMap with recursively processed values
     if isinstance(obj, dict):
         return CommentedMap({k: to_commentable_yaml(v) for k, v in obj.items()})
-    
+
     # Convert list to CommentedSeq with recursively processed elements
     elif isinstance(obj, list):
         return CommentedSeq([to_commentable_yaml(i) for i in obj])
-    
+
     # Return non-container values as-is
     return obj
 
+
 def map_stage_names(stages: list[str]) -> list[str]:
     return [STAGE_DISPLAY_MAP.get(stage, stage) for stage in stages]