fix: notebook code

aws · Oct 26, 2022 · 6eae73b · 6eae73b
1 parent f4ae210
commit 6eae73b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sagemaker-pipelines/tabular/tuning-step/sagemaker-pipelines-tuning-step.ipynb b/sagemaker-pipelines/tabular/tuning-step/sagemaker-pipelines-tuning-step.ipynb
@@ -577,7 +577,11 @@
     "\n",
     "best_model = Model(\n",
     "    image_uri=image_uri,\n",
-    "    model_data=step_tuning.get_top_model_s3_uri(top_k=0, s3_bucket=model_bucket_key),\n",
+    "    model_data=step_tuning.get_top_model_s3_uri(\n",
+    "        top_k=0,\n",
+    "        s3_bucket=default_bucket,\n",
+    "        prefix=model_prefix\n",
+    "    ),\n",
     "    predictor_cls=XGBoostPredictor,\n",
     "    sagemaker_session=pipeline_session,\n",
     "    role=role,\n",