"""AstraFlow dataset registry.

Provides dataset loading functions for rollout data acquisition and eval.
No dependency on ``train_worker`` — this is a self-contained copy.
"""

from .aime24x4 import get_aime_2024x4_test_dataset
from .aime25x4 import get_aime_2025x4_test_dataset
from .amc24 import get_amc_2024x4_test_dataset
from .asearcher import get_asearcher_rl_dataset
from .dapo_filter import get_dapo_filter_rl_dataset
from .deepscaler import get_deepscaler_rl_dataset
from .deepcoder_preview import (
    get_deepcoder_preview_codeforces_test_dataset,
    get_deepcoder_preview_primeintellect_rl_dataset,
)
from .human_eval import get_human_eval_test_dataset
from .livecodebench import (
    get_livecodebench_single_turn_rl_dataset,
    get_livecodebench_single_turn_test_dataset,
)
from .math500 import get_math500_test_dataset
from .minervamath import get_minerva_math_test_dataset
from .olympiadbench import get_olympiad_bench_test_dataset

__all__ = [
    "get_aime_2024x4_test_dataset",
    "get_aime_2025x4_test_dataset",
    "get_amc_2024x4_test_dataset",
    "get_asearcher_rl_dataset",
    "get_dapo_filter_rl_dataset",
    "get_deepcoder_preview_codeforces_test_dataset",
    "get_deepcoder_preview_primeintellect_rl_dataset",
    "get_human_eval_test_dataset",
    "get_deepscaler_rl_dataset",
    "get_livecodebench_single_turn_rl_dataset",
    "get_livecodebench_single_turn_test_dataset",
    "get_math500_test_dataset",
    "get_minerva_math_test_dataset",
    "get_olympiad_bench_test_dataset",
]
