Updates to cleanup Model Customization notebooks

Author: nargokul

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

@@ -103,7 +103,7 @@
    "metadata": {},
    "source": [
     "import random\n",
-    "#! ada credentials update --provider=isengard --account=052150106756 --role=Admin --profile=default --once\n",
+    "#! ada credentials update --provider=isengard --account=<> --role=Admin --profile=default --once\n",
     "#! aws configure set region  us-west-2\n",
     "\n",
     "from sagemaker.train.dpo_trainer import DPOTrainer\n",
@@ -171,7 +171,23 @@
    "source": [
     "from pprint import pprint\n",
     "\n",
-    "pprint(training_job)"
+    "def pretty_print(obj):\n",
+    "    def parse_unassigned(item):\n",
+    "        if isinstance(item, Unassigned):\n",
+    "            return None\n",
+    "        if isinstance(item, dict):\n",
+    "            return {k: parse_unassigned(v) for k, v in item.items() if parse_unassigned(v) is not None}\n",
+    "        if isinstance(item, list):\n",
+    "            return [parse_unassigned(x) for x in item if parse_unassigned(x) is not None]\n",
+    "        if isinstance(item, str) and \"Unassigned object\" in item:\n",
+    "            pairs = re.findall(r\"(\\w+)=([^<][^=]*?)(?=\\s+\\w+=|$)\", item)\n",
+    "            result = {k: v.strip(\"'\\\"\") for k, v in pairs}\n",
+    "            return result if result else None\n",
+    "        return item\n",
+    "\n",
+    "    cleaned = parse_unassigned(obj.__dict__ if hasattr(obj, '__dict__') else obj)\n",
+    "    print(json.dumps(cleaned, indent=2, default=str))\n",
+    "pretty_print(training_job)"
    ],
    "outputs": [],
    "execution_count": null
@@ -182,13 +198,20 @@
    },
    "cell_type": "code",
    "source": [
+    "# Print the training job object\n",
+    "\n",
     "import json\n",
     "from sagemaker.core.utils.utils import Unassigned\n",
     "from sagemaker.core.resources import TrainingJob\n",
     "import pprint\n",
     "response = TrainingJob.get(training_job_name=\"generate-sql-queries-bas-base-judge-y6cfcrah49j7-090dlKtAnQ\")\n",
-    "print(json.dumps({k: v for k, v in response.__dict__.items() if not isinstance(v, Unassigned) and \"Unassigned object\" not in str(v)}, indent=2, default=str))\n",
     "\n",
+    "import json\n",
+    "import re\n",
+    "from sagemaker.core.utils.utils import Unassigned\n",
+    "\n",
+    "# Usage\n",
+    "pretty_print(response)\n",
     "\n"
    ],
    "id": "eb2b3188-582d-4a3b-9f32-e7f17f962aa0",