File size: 528 Bytes
87c8483 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | """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)
|