reflecting David,s comments on the PR

Author: Venkatesan

sagemaker-python-sdk/mxnet_mnist_byom/mxnet_mnist.ipynb

@@ -61,10 +61,13 @@
    "execution_count": 2,
    "metadata": {
     "collapsed": true
+    "isConfigCell": true
    },
    "outputs": [],
    "source": [
-    "role = '<your SageMaker execution role here>'"
+    "import boto3\n",
+    "assumed_role = boto3.client('sts').get_caller_identity()['Arn']\n",
+    "role = re.sub(r'^(.+)sts::(\\d+):assumed-role/(.+?)/.*$', r'\\1iam::\\2:role/\\3', assumed_role)"
    ]
   },
   {
@@ -271,6 +274,13 @@
     "print(predictor.endpoint)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "If you do not want continied use of the endpoint, you can remove it. Remember, open endpoints are charged. If this is a simple test or practice, it is recommended to delete them."
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -279,7 +289,7 @@
    },
    "outputs": [],
    "source": [
-    "sagemaker.Session().delete_endpoint(predictor.endpoint)"
+    "# sagemaker.Session().delete_endpoint(predictor.endpoint)"
    ]
   },
   {
@@ -305,9 +315,9 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
-   "language": "python",
-   "name": "python3"
+   "display_name": "conda_python3",
+   "language": "conda_python3",
+   "name": "conda_python3"
   },
   "language_info": {
    "codemirror_mode": {

sagemaker-python-sdk/tensorflow_iris_byom/tensorflow_BYOM_iris.ipynb

@@ -43,10 +43,13 @@
    "execution_count": null,
    "metadata": {
     "collapsed": true
+    "isConfigCell": true
    },
    "outputs": [],
    "source": [
-    "role = '<your SageMaker execution role here>'"
+    "import boto3\n",
+    "assumed_role = boto3.client('sts').get_caller_identity()['Arn']\n",
+    "role = re.sub(r'^(.+)sts::(\\d+):assumed-role/(.+?)/.*$', r'\\1iam::\\2:role/\\3', assumed_role)"
    ]
   },
   {
@@ -61,9 +64,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "import os\n",
@@ -119,9 +120,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "classifier = estimator_fn(run_config = None, params = None)"
@@ -195,9 +194,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "classifier.train(input_fn = train_func, steps = 1000)"
@@ -209,7 +206,7 @@
    "source": [
     "## Set up hosting for the model\n",
     "\n",
-    "### Export the model from mxnet\n",
+    "### Export the model from tensorflow\n",
     "\n",
     "In order to set up hosting, we have to import the model from training to hosting. We will begin by exporting the model from TensorFlow and saving it down. Analogous to the [MXNet example](../mxnet_mnist_byom/mxnet_mnist.ipynb), some structure needs to be followed. The exported model has to be converted into a form that is readable by ``sagemaker.mxnet.model.MXNetModel``. The following code describes exporting the model in a form that does the same:\n",
     "\n",
@@ -219,9 +216,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "exported_model = classifier.export_savedmodel(export_dir_base = 'export/Servo/', \n",
@@ -288,9 +283,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "metadata": {
-    "collapsed": true
-   },
+   "metadata": {},
    "outputs": [],
    "source": [
     "%%time\n",
@@ -313,9 +306,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "predict_samples = {}\n",
-    "predict_samples ={INPUT_TENSOR_NAME: np.array([[6.4,3.2,4.5,1.5]])}\n",
-    "predictor.predict(np.array([[6.4,3.2,4.5,1.5]]))"
+    "sample = [6.4,3.2,4.5,1.5]\n",
+    "predictor.predict(sample)"
    ]
   },
   {
@@ -335,9 +327,25 @@
    "source": [
     "os.remove('model.tar.gz')\n",
     "import shutil\n",
-    "shutil.rmtree('export')\n",
-    "\n",
-    "sagemaker.Session().delete_endpoint(predictor.endpoint)"
+    "shutil.rmtree('export')"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "If you do not want continied use of the endpoint, you can remove it. Remember, open endpoints are charged. If this is a simple test or practice, it is recommended to delete them."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "collapsed": true
+   },
+   "outputs": [],
+   "source": [
+    "# sagemaker.Session().delete_endpoint(predictor.endpoint)"
    ]
   }
  ],