open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-24 17:05:03 +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 e4b032c814
commit 3a8a613573
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):