Merge branch 'main' into atqy-fix-byoc-rapids-sagemaker-hpo

aws · Aug 16, 2022 · da1c102 · da1c102
2 parents 358b1ba + aabd058
commit da1c102
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/advanced_functionality/causal-inference/causal-inference-container.ipynb b/advanced_functionality/causal-inference/causal-inference-container.ipynb
@@ -108,7 +108,7 @@
     "fi\n",
     "\n",
     "# Get the login command from ECR and execute it directly\n",
-    "aws ecr get-login-password --region ${region}|docker login --username AWS --password-stdin ${fullname}\n",
+    "$(aws ecr get-login --region $region --registry-ids $account --no-include-email)\n",
     "\n",
     "# Build the docker image locally with the image name and then push it to ECR\n",
     "# with the full name.\n",
@@ -176,6 +176,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "! mkdir data\n",
+    "\n",
     "# S3 bucket where the training data is located.\n",
     "data_bucket = f\"sagemaker-sample-files\"\n",
     "data_prefix = \"datasets/tabular/uci_heart_failure/\"\n",

diff --git a/sagemaker_batch_transform/working_with_tfrecords/working-with-tfrecords.ipynb b/sagemaker_batch_transform/working_with_tfrecords/working-with-tfrecords.ipynb
@@ -50,7 +50,7 @@
     "import sagemaker\n",
     "import tensorflow as tf\n",
     "\n",
-    "bucket = \"<your_bucket_name>\"\n",
+    "bucket = sagemaker.Session().default_bucket()\n",
     "training_prefix = \"training\"\n",
     "batch_input_prefix = \"batch_input\"\n",
     "batch_output_prefix = \"batch_output\"\n",
@@ -292,7 +292,7 @@
     "tf_serving_model = Model(\n",
     "    model_data=estimator.model_data,\n",
     "    role=sagemaker.get_execution_role(),\n",
-    "    image=estimator.image_name,\n",
+    "    image_uri=estimator.image_uri,\n",
     "    framework_version=estimator.framework_version,\n",
     "    sagemaker_session=estimator.sagemaker_session,\n",
     ")\n",