Rename ArithmeticDataset to BasicArithmeticDataset

This commit is contained in:
Andreas Koepf 2025-01-24 10:31:26 +01:00
parent 4817c33b65
commit 0e9250bce0
7 changed files with 27 additions and 27 deletions

View file

@ -37,9 +37,9 @@ The goal is to generate virtually infinite data with adjustable complexity.
#### Basic Arithmetic
Generates arithmetic problems with configurable complexity:
```python
from reasoning_gym.arithmetic import ArithmeticDataset, ArithmeticDatasetConfig
from reasoning_gym.arithmetic import BasicArithmeticDataset, BasicArithmeticDatasetConfig
config = ArithmeticDatasetConfig(
config = BasicArithmeticDatasetConfig(
min_terms=2, # Minimum number of terms in expression
max_terms=4, # Maximum number of terms
min_digits=1, # Minimum digits per number
@ -49,7 +49,7 @@ config = ArithmeticDatasetConfig(
seed=42 # For reproducibility
)
dataset = ArithmeticDataset(config)
dataset = BasicArithmeticDataset(config)
for item in dataset:
print(item)
```