arize-datasets

Arize AX Datasets

Manage datasets in the Arize AI platform using the ax CLI.

Prerequisites

The user must have:

1. Arize AX CLI installed (pip install arize-ax-cli)
2. CLI configured with valid credentials (ax config init)

Core Dataset Commands

List All Datasets

ax datasets list

Options:

• --output <format> - Output format: table (default), json, csv, parquet
• --profile <name> - Use specific configuration profile
• --limit <n> - Limit number of results
• --offset <n> - Skip first n results (pagination)

Examples:

# List as table (default)
ax datasets list

# List as JSON
ax datasets list --output json

# List with pagination
ax datasets list --limit 10 --offset 0

# Use production profile
ax datasets list --profile production

Extracting Dataset IDs:

To find a specific dataset ID for use in other operations:

# Get all dataset IDs and names as JSON
ax datasets list --output json | jq '.[] | {id: .id, name: .name}'

# Find a dataset ID by name
ax datasets list --output json | jq -r '.[] | select(.name == "Training Data") | .id'

# Save dataset ID to a variable
DATASET_ID=$(ax datasets list --output json | jq -r '.[] | select(.name == "Training Data") | .id')
echo "Found dataset: $DATASET_ID"

# Use the ID in subsequent commands
ax datasets get "$DATASET_ID"
ax datasets delete "$DATASET_ID"

Without jq (using grep):

# List with grep to find dataset
ax datasets list --output json | grep -A 2 "Training Data" | grep "id"

# More reliable pattern
ax datasets list --output json | grep -B 1 '"name": "Training Data"' | grep "id" | cut -d'"' -f4

Get Dataset Details

Retrieve information about a specific dataset:

ax datasets get <dataset-id>

Options:

• --output <format> - Output format
• --profile <name> - Configuration profile to use

Examples:

# Get dataset details
ax datasets get ds_abc123xyz

# Get as JSON
ax datasets get ds_abc123xyz --output json

# Get from production environment
ax datasets get ds_abc123xyz --profile production