Add new probability problems dataset and extend combinatorics with additional task types

This commit is contained in:
Ritvik19 2026-04-18 19:26:10 +05:30
parent 9847d71dce
commit dc0d81c096
5 changed files with 978 additions and 4 deletions

View file

@ -8,6 +8,11 @@ from .conditional_probability import (
ConditionalProbabilityCurriculum,
ConditionalProbabilityDataset,
)
from .probability_problems import (
ProbabilityProblemsConfig,
ProbabilityProblemsCurriculum,
ProbabilityProblemsDataset,
)
__all__ = [
"CoinFlipDataset",
@ -16,4 +21,7 @@ __all__ = [
"ConditionalProbabilityDataset",
"ConditionalProbabilityConfig",
"ConditionalProbabilityCurriculum",
"ProbabilityProblemsDataset",
"ProbabilityProblemsConfig",
"ProbabilityProblemsCurriculum",
]