open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +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

6
reasoning_gym/arc/rearc_utils/utils.py
View file

@ -1,5 +1,5 @@
import random
from typing import Any, List, Tuple
from typing import Any
from .dsl import *
@ -40,7 +40,7 @@ def get_pso_difficulty(example: dict) -> float:
return (pix_pct + col_pct + obj_dens) / 3
def unifint(rng: random.Random, diff_lb: float, diff_ub: float, bounds: Tuple[int, int]) -> int:
def unifint(rng: random.Random, diff_lb: float, diff_ub: float, bounds: tuple[int, int]) -> int:
"""
rng
diff_lb: lower bound for difficulty, must be in range [0, diff_ub]
@ -83,7 +83,7 @@ def strip_prefix(string: str, prefix: str) -> str:
return string[len(prefix) :]
def format_grid(grid: List[List[int]]) -> Grid:
def format_grid(grid: list[list[int]]) -> Grid:
"""
grid type casting
"""