format: ruff format examples directory (#559)

* fix format in examples

* Apply suggestion from @gemini-code-assist[bot]

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

* Apply suggestion from @gemini-code-assist[bot]

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

---------

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>

Author: fishcrap

examples/config_converter.py

@@ -2,8 +2,9 @@
 import argparse
 import os
 import shlex
+from collections.abc import Callable
 from dataclasses import dataclass, field
-from typing import Any, Callable, Optional, Type
+from typing import Any
 
 import yaml
 from deepmerge import always_merger
@@ -17,11 +18,10 @@
 
 @dataclass
 class ArgSpec:
-
     arg: str = ""
-    arg_type: Type[Any] = field(default=str)
+    arg_type: type[Any] = field(default=str)
     description: str = ""
-    map_fn: Optional[Callable] = field(default=None)
+    map_fn: Callable | None = field(default=None)
 
 
 class Converter(abc.ABC):
@@ -43,7 +43,7 @@ def get_lite_template(self, template_path: str):
         """
         Load the areal template from the specified file.
         """
-        with open(template_path, "r", encoding="utf-8") as f:
+        with open(template_path, encoding="utf-8") as f:
             return yaml.safe_load(f)
 
     def flatten_dict(self, d, parent_key="", sep="."):
@@ -74,7 +74,7 @@ def convert_to_nested_args(self, args, ARG_MAP: dict) -> dict:
                 argspec = ARG_MAP[k]
                 if not argspec.arg:
                     print(
-                        colored(f"## Warning: For ", "yellow")
+                        colored("## Warning: For ", "yellow")
                         + colored(f"{k:>40}", "yellow", attrs=["bold"])
                         + colored(f",    # {argspec.description}!", "yellow")
                     )
@@ -87,23 +87,23 @@ def convert_to_nested_args(self, args, ARG_MAP: dict) -> dict:
                 if v is not None:
                     try:
                         # type conversion
-                        if arg_type == bool:
+                        if arg_type is bool:
                             v = (
                                 bool(v)
                                 if isinstance(v, bool)
                                 else v.lower() in ("1", "true", "yes", "on")
                             )
-                        elif arg_type == int:
+                        elif arg_type is int:
                             v = int(v)
-                        elif arg_type == float:
+                        elif arg_type is float:
                             v = float(v)
-                        elif arg_type == str:
+                        elif arg_type is str:
                             v = str(v)
                         else:
                             raise ValueError(f"Unsupported type: {arg_type}")
                     except Exception as e:
                         print(
-                            colored(f"## Error: For ", "red")
+                            colored("## Error: For ", "red")
                             + colored(f"{k:>40} {v}", "red", attrs=["bold"])
                             + colored(f",    # {e}!", "red")
                         )
@@ -117,7 +117,7 @@ def convert_to_nested_args(self, args, ARG_MAP: dict) -> dict:
             else:
                 unmapped[k] = v
                 print(
-                    colored(f"## Warning: For ", "yellow")
+                    colored("## Warning: For ", "yellow")
                     + colored(f"{k:>50}", "yellow", attrs=["bold"])
                     + colored(f",    # {CVRT_WARNING}!", "yellow")
                 )
@@ -135,7 +135,6 @@ def set_nested(self, d: dict, keys, value):
 
 
 class OpenRLHFConverter(Converter):
-
     ARG_MAP = {
         # Ray and vLLM
         "ref_num_nodes": ArgSpec("", int, CVRT_WARNING),
@@ -361,7 +360,7 @@ def _parse_args_from_script(
         in_command_block = False
 
         try:
-            with open(script_path, "r", encoding="utf-8") as f:
+            with open(script_path, encoding="utf-8") as f:
                 for line in f:
                     stripped_line = line.strip()
 
@@ -425,7 +424,6 @@ def _parse_args_from_script(
 
 
 def post_process_args(args: dict):
-
     if "allocation_mode" in args:
         # convert allocation_mode to sglang.dX.tY.pZ
         dp = args["cluster"]["n_nodes"] * args["cluster"]["n_gpus_per_node"]
@@ -442,7 +440,7 @@ def post_process_args(args: dict):
                 allocation_mode += "t1"
             else:
                 allocation_mode += f"t{args['allocation_mode']['sglang']['t']}"
-            allocation_mode += f"p1"
+            allocation_mode += "p1"
         allocation_mode += "+"
         if "engine" not in args["allocation_mode"]:
             allocation_mode += f"d{dp}t1p1"
@@ -452,7 +450,7 @@ def post_process_args(args: dict):
                 allocation_mode += "t1"
             else:
                 allocation_mode += f".t{args['allocation_mode']['engine']['t']}"
-            allocation_mode += f"p1"
+            allocation_mode += "p1"
 
         args["allocation_mode"] = allocation_mode
         args["cluster"]["n_nodes"] = args["cluster"]["n_nodes"] * 2
@@ -830,7 +828,7 @@ def __init__(self, src_config_path: str, template_path: str):
         self.template_path = template_path
 
     def parse(self) -> dict:
-        with open(self.src_config_path, "r", encoding="utf-8") as f:
+        with open(self.src_config_path, encoding="utf-8") as f:
             cfg = yaml.safe_load(f)
         return cfg
 
@@ -895,7 +893,7 @@ def main():
         **converter_args[args.convert_src]
     )
     lite_args = converter.convert()
-    yaml_str = yaml.dump(lite_args, sort_keys=False, allow_unicode=True)
+    # yaml_str = yaml.dump(lite_args, sort_keys=False, allow_unicode=True)
     with open(args.output_path, "w", encoding="utf-8") as f:
         yaml.dump(lite_args, f, sort_keys=False, allow_unicode=True)
     print(f"Converted areal config saved to {args.output_path}")

examples/countdown/countdown.py

@@ -4,7 +4,6 @@
 import random
 import re
 import sys
-from typing import Dict
 
 from tqdm import tqdm
 from transformers import AutoTokenizer
@@ -279,19 +278,19 @@ def combine_nums(a, b):
     # Implicitly makes assumptions about the order of operations and valid operations
     a = int(a)
     b = int(b)
-    possible = [[a + b, f"{a}+{b}={a+b}"], [a * b, f"{a}*{b}={a*b}"]]
+    possible = [[a + b, f"{a}+{b}={a + b}"], [a * b, f"{a}*{b}={a * b}"]]
     if a <= b:
-        possible.append([b - a, f"{b}-{a}={b-a}"])
+        possible.append([b - a, f"{b}-{a}={b - a}"])
         if a != 0 and b % a == 0:
-            possible.append([b // a, f"{b}/{a}={round(b//a,0)}"])
+            possible.append([b // a, f"{b}/{a}={round(b // a, 0)}"])
     else:
-        possible.append([a - b, f"{a}-{b}={a-b}"])
+        possible.append([a - b, f"{a}-{b}={a - b}"])
         if b != 0 and a % b == 0:
-            possible.append([a // b, f"{a}/{b}={round(a//b,0)}"])
+            possible.append([a // b, f"{a}/{b}={round(a // b, 0)}"])
     return possible
 
 
-class CountDown(object):
+class CountDown:
     def __init__(
         self,
         max_target=25,
@@ -330,7 +329,7 @@ def generate(self, target):
                 found = True
         return nums, solution
 
-    def get_task(self, apply_chat_template=False, return_raw=False) -> Dict[str, str]:
+    def get_task(self, apply_chat_template=False, return_raw=False) -> dict[str, str]:
         target = random.randint(self.min_target, self.max_target)
         nums, solution = self.generate(target)
 

examples/countdown/reward_score.py

@@ -34,7 +34,7 @@ def validate_equation(equation_str, available_numbers):
 
         # Each number should be used exactly once
         return numbers_in_eq == available_numbers
-    except:
+    except Exception:
         return False
 
 
@@ -72,28 +72,28 @@ def compute_score(
     do_print = random.randint(1, 64) == 1
 
     if do_print:
-        print(f"--------------------------------")
+        print("--------------------------------")
         print(f"Target: {target} | Numbers: {numbers}")
         print(f"Extracted equation: {equation}")
         print(f"Solution string: {solution_str}")
 
     if equation is None:
         if do_print:
-            print(f"No equation found")
+            print("No equation found")
         return 0
 
     # Validate equation uses correct numbers
     if not validate_equation(equation, numbers):
         if do_print:
-            print(f"Invalid equation")
+            print("Invalid equation")
         return format_score
 
     # Evaluate equation
     try:
         result = evaluate_equation(equation)
         if result is None:
             if do_print:
-                print(f"Could not evaluate equation")
+                print("Could not evaluate equation")
             return format_score
 
         if abs(result - target) < 1e-5:  # Account for floating point precision
@@ -104,7 +104,7 @@ def compute_score(
             if do_print:
                 print(f"Wrong result: equation = {result}, target = {target}")
             return format_score
-    except:
+    except Exception:
         if do_print:
-            print(f"Error evaluating equation")
+            print("Error evaluating equation")
         return format_score

examples/docs/debug/cmp_rollout.py

@@ -1,6 +1,5 @@
 import base64
 from io import BytesIO
-from typing import List
 
 import datasets
 import requests
@@ -9,7 +8,7 @@
 from transformers import AutoProcessor, Gemma3ForConditionalGeneration
 
 
-def image2base64(images: List[ImageObject] | ImageObject) -> List[str] | str:
+def image2base64(images: list[ImageObject] | ImageObject) -> list[str] | str:
     if isinstance(images, ImageObject):
         images = [images]
 

examples/search-agent/tongyi_deepresearch/react_agent.py

@@ -1,31 +1,29 @@
+import datetime
 import json
 import os
 import sys
 import time
-from datetime import datetime
 from pathlib import Path
-from typing import Dict, List
 
 import json5
 from qwen_agent.agents.fncall_agent import FnCallAgent
 from qwen_agent.llm.schema import Message
-from qwen_agent.settings import MAX_LLM_CALL_PER_RUN
 from transformers import PreTrainedTokenizer
 
 from areal.experimental.openai import ArealOpenAI
 from areal.utils import logging
 
 try:
-    from .prompt import *
-    from .tool_search import *
-    from .tool_visit import *
+    from .prompt import SYSTEM_PROMPT
+    from .tool_search import Search
+    from .tool_visit import Visit
 except ImportError:  # Fallback when executed directly (no package parent known)
     module_dir = Path(__file__).parent
     if str(module_dir) not in sys.path:
         sys.path.insert(0, str(module_dir))
-    from prompt import *
-    from tool_search import *
-    from tool_visit import *
+    from prompt import SYSTEM_PROMPT
+    from tool_search import Search
+    from tool_visit import Visit
 
 
 logger = logging.getLogger("Tongyi-DeepResearch react agent")
@@ -35,8 +33,6 @@
 
 MAX_LLM_CALL_PER_RUN = int(os.getenv("MAX_LLM_CALL_PER_RUN", 100))
 
-import datetime
-
 
 def today_date():
     return datetime.date.today().strftime("%Y-%m-%d")
@@ -52,7 +48,7 @@ def parse_judge_result(raw_response):
         try:
             mbe = parse_fn(raw_response.split("```json")[-1].split("```")[0].strip())
             break
-        except:
+        except Exception:
             logger.warning(f"Error parsing judge result with {parse_fn}.")
     if mbe is None and '"judgement": "incorrect"' in raw_response:
         mbe = dict(judgement="incorrect")
@@ -94,7 +90,7 @@ def count_tokens(self, messages):
         return len(prompt_token_ids)
 
     async def call_server(
-        self, client: ArealOpenAI, messages: List[Dict], max_attempts: int = 100
+        self, client: ArealOpenAI, messages: list[dict], max_attempts: int = 100
     ) -> str:
         attempts = 0
         while attempts < max_attempts:
@@ -119,7 +115,7 @@ async def call_server(
 
     async def run_agent(
         self, data, client: ArealOpenAI, save_path: str | None = None
-    ) -> List[List[Message]]:
+    ) -> list[list[Message]]:
         start_time = time.time()
         data["qid"]
         question = data["question"]
@@ -275,7 +271,7 @@ async def custom_call_tool(self, tool_name: str, tool_args: dict, **kwargs):
 
     async def calc_reward_with_llm_judge(
         self,
-        result: Dict[str, str],
+        result: dict[str, str],
     ):
         # Compute reward with LLM-as-Judge
         # judge_client = ArealOpenAI(engine=rollout_engine, tokenizer=tokenizer)
@@ -320,10 +316,10 @@ async def calc_reward_with_llm_judge(
 
     async def make_trajectory(
         self,
-        data: Dict[str, str],
+        data: dict[str, str],
         client: ArealOpenAI,
         save_path: str | None = None,
-    ) -> Dict:
+    ) -> dict:
         result = await self.run_agent(data, client, save_path=save_path)
         reward = await self.calc_reward_with_llm_judge(result)
         completions = result["completions"]

examples/search-agent/tongyi_deepresearch/tool_search.py

@@ -1,7 +1,6 @@
 import asyncio
 import json
 import os
-from typing import List, Optional, Union
 
 import aiohttp
 from qwen_agent.tools.base import BaseTool, register_tool
@@ -25,7 +24,7 @@ class Search(BaseTool):
         "required": ["query"],
     }
 
-    def __init__(self, cfg: Optional[dict] = None):
+    def __init__(self, cfg: dict | None = None):
         super().__init__(cfg)
 
     async def google_search_with_serp(self, query: str):
@@ -72,7 +71,7 @@ def contains_chinese_basic(text: str) -> bool:
                             snippet = (
                                 f"\n{page['snippet']}" if page.get("snippet") else ""
                             )
-                            redacted_version = f"{idx}. [{page.get('title','')}]({page.get('link','')}){date_published}{source}\n{snippet}"
+                            redacted_version = f"{idx}. [{page.get('title', '')}]({page.get('link', '')}){date_published}{source}\n{snippet}"
                             redacted_version = redacted_version.replace(
                                 "Your browser can't play this video.", ""
                             )
@@ -93,7 +92,7 @@ def contains_chinese_basic(text: str) -> bool:
     async def search_with_serp(self, query: str):
         return await self.google_search_with_serp(query)
 
-    async def call(self, params: Union[str, dict], **kwargs) -> str:  # type: ignore[override]
+    async def call(self, params: str | dict, **kwargs) -> str:  # type: ignore[override]
         try:
             query = params["query"]
         except Exception:
@@ -102,7 +101,7 @@ async def call(self, params: Union[str, dict], **kwargs) -> str:  # type: ignore
         if isinstance(query, str):
             return await self.search_with_serp(query)
 
-        assert isinstance(query, List)
+        assert isinstance(query, list)
         tasks = [self.search_with_serp(q) for q in query]
         responses = await asyncio.gather(*tasks)
         return "\n=======\n".join(responses)