"""Lightweight dataset helpers (dummy datasets, episode attention masks)."""

from datasets import Dataset


def create_dummy_dataset(num_samples: int = 100) -> Dataset:
    """Create a dummy dataset with the specified number of samples.

    Args:
        num_samples (int): Number of samples to create in the dataset.

    Returns:
        Dataset: A HuggingFace Dataset containing dummy prompts.
    """
    dummy_data = {"prompt": [f"Sample prompt {i}" for i in range(num_samples)]}
    return Dataset.from_dict(dummy_data)