fix: tuning notebook formatting

aws · Oct 26, 2022 · 628c4f2 · 628c4f2
1 parent 8f9bac4
commit 628c4f2
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/sagemaker-pipelines/tabular/tuning-step/sagemaker-pipelines-tuning-step.ipynb b/sagemaker-pipelines/tabular/tuning-step/sagemaker-pipelines-tuning-step.ipynb
@@ -578,9 +578,7 @@
     "best_model = Model(\n",
     "    image_uri=image_uri,\n",
     "    model_data=step_tuning.get_top_model_s3_uri(\n",
-    "        top_k=0,\n",
-    "        s3_bucket=default_bucket,\n",
-    "        prefix=model_prefix\n",
+    "        top_k=0, s3_bucket=default_bucket, prefix=model_prefix\n",
     "    ),\n",
     "    predictor_cls=XGBoostPredictor,\n",
     "    sagemaker_session=pipeline_session,\n",
@@ -595,9 +593,7 @@
     "second_best_model = Model(\n",
     "    image_uri=image_uri,\n",
     "    model_data=step_tuning.get_top_model_s3_uri(\n",
-    "        top_k=1,\n",
-    "        s3_bucket=default_bucket,\n",
-    "        prefix=model_prefix\n",
+    "        top_k=1, s3_bucket=default_bucket, prefix=model_prefix\n",
     "    ),\n",
     "    predictor_cls=XGBoostPredictor,\n",
     "    sagemaker_session=pipeline_session,\n",
@@ -721,9 +717,7 @@
     "    inputs=[\n",
     "        ProcessingInput(\n",
     "            source=step_tuning.get_top_model_s3_uri(\n",
-    "                top_k=0,\n",
-    "                s3_bucket=default_bucket,\n",
-    "                prefix=model_prefix\n",
+    "                top_k=0, s3_bucket=default_bucket, prefix=model_prefix\n",
     "            ),\n",
     "            destination=\"/opt/ml/processing/model\",\n",
     "        ),\n",