Merge branch 'main' into codex/fix-429-rate-limit-salvage

Author: eric-tramel

docs/colab_notebooks/1-the-basics.ipynb

@@ -2,15 +2,17 @@
  "cells": [
   {
    "cell_type": "markdown",
-   "id": "e9bc2aab",
-   "metadata": {},
+   "id": "f5bc03e0",
+   "metadata": {
+    "nemo_colab_inject": true
+   },
    "source": [
     "<a href=\"https://colab.research.google.com/github/NVIDIA-NeMo/DataDesigner/blob/main/docs/colab_notebooks/1-the-basics.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
    ]
   },
   {
    "cell_type": "markdown",
-   "id": "33dcb5be",
+   "id": "3454d676",
    "metadata": {},
    "source": [
     "# 🎨 Data Designer Tutorial: The Basics\n",
@@ -22,7 +24,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "adb77b8d",
+   "id": "4737bc0d",
    "metadata": {},
    "source": [
     "### 📦 Import Data Designer\n",
@@ -34,8 +36,10 @@
   },
   {
    "cell_type": "markdown",
-   "id": "170ce1ea",
-   "metadata": {},
+   "id": "cf21b784",
+   "metadata": {
+    "nemo_colab_inject": true
+   },
    "source": [
     "### ⚡ Colab Setup\n",
     "\n",
@@ -45,8 +49,10 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "67e478f9",
-   "metadata": {},
+   "id": "a87ee38a",
+   "metadata": {
+    "nemo_colab_inject": true
+   },
    "outputs": [],
    "source": [
     "%%capture\n",
@@ -56,8 +62,10 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "533fc40d",
-   "metadata": {},
+   "id": "d6c4cc8a",
+   "metadata": {
+    "nemo_colab_inject": true
+   },
    "outputs": [],
    "source": [
     "import getpass\n",
@@ -74,7 +82,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "9ad92889",
+   "id": "848dea00",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -84,7 +92,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "0232c4c6",
+   "id": "b97786a8",
    "metadata": {},
    "source": [
     "### ⚙️ Initialize the Data Designer interface\n",
@@ -97,7 +105,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "fbbd0cab",
+   "id": "b31c1fc9",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -106,7 +114,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "305f635e",
+   "id": "2fef5ae8",
    "metadata": {},
    "source": [
     "### 🎛️ Define model configurations\n",
@@ -123,7 +131,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "d0865b58",
+   "id": "7a9f6398",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -153,7 +161,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "6e1624f7",
+   "id": "1d0a178f",
    "metadata": {},
    "source": [
     "### 🏗️ Initialize the Data Designer Config Builder\n",
@@ -168,7 +176,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "33562cda",
+   "id": "aacc0ec5",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -177,7 +185,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "d8ec3063",
+   "id": "4be3497f",
    "metadata": {},
    "source": [
     "## 🎲 Getting started with sampler columns\n",
@@ -194,7 +202,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "70de1b0a",
+   "id": "e212d83e",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -203,7 +211,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "991a8f34",
+   "id": "c28350d3",
    "metadata": {},
    "source": [
     "Let's start designing our product review dataset by adding product category and subcategory columns.\n"
@@ -212,7 +220,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "222cbbcc",
+   "id": "070f14e7",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -293,7 +301,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "29ca2aa3",
+   "id": "e0d8497d",
    "metadata": {},
    "source": [
     "Next, let's add samplers to generate data related to the customer and their review.\n"
@@ -302,7 +310,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "4ca9ba1c",
+   "id": "62e84282",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -339,7 +347,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "f4d54299",
+   "id": "8cb147fa",
    "metadata": {},
    "source": [
     "## 🦜 LLM-generated columns\n",
@@ -354,7 +362,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "361b63b1",
+   "id": "37a4a6d0",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -390,7 +398,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "49ca028a",
+   "id": "49559576",
    "metadata": {},
    "source": [
     "### 🔁 Iteration is key – preview the dataset!\n",
@@ -407,7 +415,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "068ea8c3",
+   "id": "0d52b447",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -417,7 +425,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "bf196a77",
+   "id": "088a5004",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -428,7 +436,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "36ebb017",
+   "id": "9780021a",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -438,7 +446,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "1dcba545",
+   "id": "c9122bc6",
    "metadata": {},
    "source": [
     "### 📊 Analyze the generated data\n",
@@ -451,7 +459,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "e5164902",
+   "id": "4d6bb3c5",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -461,7 +469,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "cc433fae",
+   "id": "6003ae71",
    "metadata": {},
    "source": [
     "### 🆙 Scale up!\n",
@@ -474,7 +482,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "17132fe2",
+   "id": "e343639d",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -484,7 +492,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "6669442a",
+   "id": "cd328abd",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -497,7 +505,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "ee689b41",
+   "id": "6a09793a",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -509,7 +517,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "6965e6ac",
+   "id": "769dd181",
    "metadata": {},
    "source": [
     "## ⏭️ Next Steps\n",