open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-26 17:13:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use native types List->list, Dict->dict, Set->set, Tuple->tuple

Browse source
This commit is contained in:
Andreas Koepf 2025-02-21 15:13:19 +01:00
parent 5d02064b5a
commit 3e7ff3b084
95 changed files with 754 additions and 760 deletions
Download patch file Download diff file Expand all files Collapse all files

7
reasoning_gym/arc/board_format.py
View file

@ -1,5 +1,4 @@
from dataclasses import dataclass, field
from typing import List, Tuple
ARC_PROMPT_TEMPLATE = """Find the common rule that maps an input grid to an output grid, given the examples below.
@ -21,7 +20,7 @@ class BoardFormattingOptions:
def format_board(
board: List[List[int]], formatting_options: BoardFormattingOptions, with_board_shape: bool = False
board: list[list[int]], formatting_options: BoardFormattingOptions, with_board_shape: bool = False
) -> str:
"""
Format a board as a string
@ -65,7 +64,7 @@ def format_board(
def format_board_pair(
index: int,
pair: dict[str, List[List[int]]],
pair: dict[str, list[list[int]]],
formatting_options: BoardFormattingOptions,
) -> str:
"""
@ -82,7 +81,7 @@ def format_board_pair(
return f"Example {index}:\n\nInput:\n{input_element}\nOutput:\n{output_element}\n\n"
def parse_board(formatted_str: str, formatting_options: BoardFormattingOptions) -> Tuple[Tuple[int, ...], ...]:
def parse_board(formatted_str: str, formatting_options: BoardFormattingOptions) -> tuple[tuple[int, ...], ...]:
"""
Convert a formatted board string back to a tuple grid using formatting options
"""