add-dataset

Scaffold and register AReaL dataset loaders for SFT and RL workflows.

5.7k|573|Updated Feb 24, 2025
One-click install
npx skills add https://github.com/inclusionAI/AReaL --skill add-dataset
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: add-dataset
Source: https://github.com/inclusionAI/AReaL/tree/main/.claude/skills/add-dataset
Command: npx skills add https://github.com/inclusionAI/AReaL --skill add-dataset

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

This Skill helps data engineers and ML engineers add new dataset loaders to AReaL, standardizing the process of integrating diverse data sources for SFT and RL workflows.

Core Features & Use Cases

  • Template-based scaffolding for creating areal/dataset/<name>.py loaders and updates to areal/dataset/init.py.
  • Dual-mode support for SFT and RL datasets, including required fields and processing steps.
  • Validation guidance and testing scaffolds to ensure the loaders produce HuggingFace Datasets with expected schema.

Quick Start

Use the add-dataset skill to scaffold a new dataset loader and register it in the areal project.

Frequently Asked Questions about add-dataset

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I add a new dataset loader to AReaL for SFT and RL workflows?

To add a new dataset loader to AReaL, you create a loader file using provided code templates, define the SFT or RL dataset structures, and register it in areal/dataset/__init__.py. This standardizes integrating diverse data sources.

What dataset structure is required for SFT and RL data in HuggingFace datasets?

SFT and RL dataset structures require specific fields and processing steps to ensure the loader outputs HuggingFace Datasets with the expected schema. The skill provides map and filter column expectations to validate these required fields.

Does AReaL support integrating HuggingFace datasets with custom column expectations?

Yes, AReaL supports integrating HuggingFace datasets by guiding the creation of custom loaders. These loaders define map and filter operations to ensure data conforms to the expected column schema for SFT and RL.

How do I test a custom dataset loader to ensure it produces the correct schema?

You test a custom dataset loader using provided testing scaffolds. These scaffolds validate that the loader correctly processes data and produces HuggingFace Datasets matching the expected SFT or RL schema.

What is the best way to register a Python dataset loader in an ML project?

The best way to register a Python dataset loader is to add it to the project's dataset module initialization file. This skill provides registration snippets to update areal/dataset/__init__.py, ensuring the loader is recognized.

Why do I need to update areal/dataset/__init__.py when adding a dataset?

Updating areal/dataset/__init__.py is required to formally register the new dataset loader within the AReaL framework, making it accessible for SFT and RL workflows and ensuring proper module integration.