Fix

Author: nargokul

v3-examples/model-customization-examples/dpo-trainer-e2e.ipynb

@@ -75,7 +75,7 @@
     "# Register dataset in SageMaker AI Registry\n",
     "# This creates a versioned dataset that can be referenced by ARN\n",
     "dataset = DataSet.create(\n",
-    "    name=\"demo-nargokul-6\", \n",
+    "    name=\"demo-6\",\n",
     "    data_location=\"s3://nova-mlflow-us-west-2/dataset/preference_dataset_train_256.jsonl\", \n",
     "    customization_technique=CustomizationTechnique.DPO, \n",
     "    wait=True\n",
@@ -145,7 +145,7 @@
     "    mlflow_resource_arn=\"arn:aws:sagemaker:us-west-2:<>:mlflow-tracking-server/ashwpat-test\",\n",
     "    \n",
     "    # Training data (from Step 1)\n",
-    "    training_dataset=\"arn:aws:sagemaker:us-west-2:<>:hub-content/F3LMYANDKWPZCROJVCKMJ7TOML6QMZBZRRQOVTUL45VUK7PJ4SXA/DataSet/demo-nargokul-6/0.0.4\",\n",
+    "    training_dataset=\"arn:aws:sagemaker:us-west-2:<>:hub-content/F3LMYANDKWPZCROJVCKMJ7TOML6QMZBZRRQOVTUL45VUK7PJ4SXA/DataSet/demo-6/0.0.4\",\n",
     "    \n",
     "    # Output configuration\n",
     "    s3_output_path=\"s3://nova-mlflow-us-west-2/output\",\n",

v3-examples/model-customization-examples/dpo_trainer_example_notebook_v3_prod.ipynb

@@ -63,7 +63,7 @@
     "# Register dataset in SageMaker AI Registry\n",
     "# This creates a versioned dataset that can be referenced by ARN\n",
     "'''dataset = DataSet.create(\n",
-    "    name=\"demo-nargokul-6\", \n",
+    "    name=\"demo-6\",\n",
     "    data_location=\"s3://nova-mlflow-us-west-2/dataset/preference_dataset_train_256.jsonl\", \n",
     "    customization_technique=CustomizationTechnique.DPO, \n",
     "    wait=True\n",
@@ -116,7 +116,12 @@
   {
    "cell_type": "code",
    "id": "0352bdaa-fa13-44c5-a70c-0d9bf7a10477",
-   "metadata": {},
+   "metadata": {
+    "ExecuteTime": {
+     "end_time": "2025-12-05T19:30:51.233369Z",
+     "start_time": "2025-12-05T19:30:51.101703Z"
+    }
+   },
    "source": [
     "# Create DPOTrainer instance with comprehensive configuration\n",
     "trainer = DPOTrainer(\n",
@@ -159,8 +164,20 @@
     "\n",
     "print(f\"Training completed! Job ARN: {training_job.training_job_arn}\")"
    ],
-   "outputs": [],
-   "execution_count": null
+   "outputs": [
+    {
+     "ename": "NameError",
+     "evalue": "name 'DataSet' is not defined",
+     "output_type": "error",
+     "traceback": [
+      "\u001B[0;31m---------------------------------------------------------------------------\u001B[0m",
+      "\u001B[0;31mNameError\u001B[0m                                 Traceback (most recent call last)",
+      "Cell \u001B[0;32mIn[1], line 2\u001B[0m\n\u001B[1;32m      1\u001B[0m \u001B[38;5;66;03m# Create DPOTrainer instance with comprehensive configuration\u001B[39;00m\n\u001B[0;32m----> 2\u001B[0m dataset \u001B[38;5;241m=\u001B[39m \u001B[43mDataSet\u001B[49m\u001B[38;5;241m.\u001B[39mget(name\u001B[38;5;241m=\u001B[39m\u001B[38;5;124m\"\u001B[39m\u001B[38;5;124marn:aws:sagemaker:us-east-1:729646638167:hub-content/sdktest/DataSet/dpo-nova-1-test-data/0.0.1\u001B[39m\u001B[38;5;124m\"\u001B[39m)\n\u001B[1;32m      4\u001B[0m trainer \u001B[38;5;241m=\u001B[39m DPOTrainer(\n\u001B[1;32m      5\u001B[0m     \u001B[38;5;66;03m# Base model from SageMaker Hub\u001B[39;00m\n\u001B[1;32m      6\u001B[0m     model\u001B[38;5;241m=\u001B[39m\u001B[38;5;124m\"\u001B[39m\u001B[38;5;124mmeta-textgeneration-llama-3-2-1b-instruct\u001B[39m\u001B[38;5;124m\"\u001B[39m,\n\u001B[0;32m   (...)\u001B[0m\n\u001B[1;32m     28\u001B[0m     accept_eula\u001B[38;5;241m=\u001B[39m\u001B[38;5;28;01mTrue\u001B[39;00m\n\u001B[1;32m     29\u001B[0m )\n\u001B[1;32m     31\u001B[0m \u001B[38;5;66;03m# Customize training hyperparameters\u001B[39;00m\n\u001B[1;32m     32\u001B[0m \u001B[38;5;66;03m# DPO-specific parameters are automatically loaded from the model's recipe\u001B[39;00m\n",
+      "\u001B[0;31mNameError\u001B[0m: name 'DataSet' is not defined"
+     ]
+    }
+   ],
+   "execution_count": 1
   },
   {
    "cell_type": "code",