Function load_math_dataset

examples/unsloth/math_agent.py:69–82

Load the GSM-hard math dataset from the local JSONL file. Args: limit: Optional maximum number of problems to load. If None, loads all problems. Returns: A list of GsmProblem instances.

(limit: Optional[int] = None)

def load_math_dataset(limit: Optional[int] = None) -> Dataset[GsmProblem]:
    """Load the GSM-hard math dataset from the local JSONL file.

    Args:
        limit: Optional maximum number of problems to load. If None, loads all problems.

    Returns:
        A list of GsmProblem instances.
    """
    with open("data_gsmhard.jsonl", "r") as f:
        problems = [GsmProblem(**json.loads(line)) for line in f]
    if limit is not None:
        problems = problems[:limit]
    return problems


@rollout