refactor: Rename ArithmeticDataset to BasicArithmeticDataset

This commit is contained in:
Andreas Koepf (aider) 2025-01-24 10:02:02 +01:00
parent 1b556b70f7
commit 95d9974911
3 changed files with 10 additions and 10 deletions

View file

@ -6,7 +6,7 @@ Arithmetic tasks for training reasoning capabilities:
- Leg counting
"""
from .basic_arithmetic import ArithmeticDataset, ArithmeticDatasetConfig, arithmetic_dataset
from .basic_arithmetic import BasicArithmeticDataset, ArithmeticDatasetConfig, basic_arithmetic_dataset
from .chain_sum import ChainSum, ChainSumConfig, chain_sum_dataset
from .fraction_simplification import FractionSimplificationConfig, FractionSimplificationDataset, fraction_simplification_dataset
from .gcd import GCDConfig, GCDDataset, gcd_dataset
@ -15,9 +15,9 @@ from .leg_counting import LegCountingConfig, LegCountingDataset, leg_counting_da
from .prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset, prime_factorization_dataset
__all__ = [
"ArithmeticDataset",
"BasicArithmeticDataset",
"ArithmeticDatasetConfig",
"arithmetic_dataset",
"basic_arithmetic_dataset",
"ChainSum",
"ChainSumConfig",
"chain_sum_dataset",