Quickstart: augment a LeRobot dataset with one call

This guide takes you from zero to a finished dataset augmentation. You provide four things: a source dataset, a prompt, an episode count, and a destination. Kite generates the episodes on its GPUs and delivers a standard LeRobot dataset you can train on directly.

Prerequisites

Before you begin, you need:

A Kite account with API access enabled
An API key — create one from Developers → API keys in the dashboard. Keys start with kite_ and are shown only once.
curl (or any HTTP client)

Set your API key

Export the key so the examples below can use it.

export KITE_API_KEY=kite_aBcDeFgH...

Verify it works

Confirm your key is valid against the API base URL, https://api.kiteml.com/v1.

curl https://api.kiteml.com/v1/keys/me \
  -H "Authorization: Bearer $KITE_API_KEY"

Start an augmentation

One call starts a run. Point it at a source dataset, describe the change, choose how many episodes, and where the results go.

curl -X POST https://api.kiteml.com/v1/augmentations \
  -H "Authorization: Bearer $KITE_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "source": { "repo_id": "lerobot/pusht" },
    "instructions": "change the table surface to white marble, vary the lighting",
    "config": { "episode_count": 20 },
    "output": { "type": "download" }
  }'

You get back the augmentation resource — including its id — immediately, with status: "queued".

Track it to completion

Poll the run until status is succeeded. It moves through queued → processing → succeeded with a live progress value.

curl https://api.kiteml.com/v1/augmentations/aug_01J8X4... \
  -H "Authorization: Bearer $KITE_API_KEY"

Get your dataset

A completed download run exposes its files under output.files. Fetch each one, preserving its path, to reconstruct a standard LeRobot Parquet dataset on disk — ready to train on with no conversion.

That’s the full loop. For the details — idempotency, Hugging Face delivery, and the complete endpoint list — continue to Augmentations.

Need help? Reach out at support@kiteml.com.

Get started

Platform API

Quickstart: augment a LeRobot dataset with one call

Prerequisites

​Prerequisites

Prerequisites