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

feat: Add Sized and Iterable base classes to ProceduralDataset

Browse source
This commit is contained in:
Andreas Koepf (aider) 2025-01-24 10:14:42 +01:00
parent 95d9974911
commit 2a6a9655d7
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
reasoning_gym/dataset.py
View file

@ -1,10 +1,11 @@
"""Base class for procedural dataset generators"""
from abc import ABC, abstractmethod
from collections.abc import Sized, Iterable
from random import Random
from typing import Optional
from typing import Optional, Iterator, Dict, Any
class ProceduralDataset(ABC):
class ProceduralDataset(ABC, Sized, Iterable[Dict[str, Any]]):
"""Abstract base class for procedural dataset generators"""
def __init__(self, seed: Optional[int] = None, size: int = 500):