Skip to content

Commit 2509469

Browse files
author
Nikhil Kulkarni
committed
Add sagemaker installation command and remove duplicate import
1 parent ddd1507 commit 2509469

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

sagemaker_neo_compilation_jobs/xgboost_customer_churn/xgboost_customer_churn_neo.ipynb

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,16 @@
4949
"- The IAM role arn used to give training and hosting access to your data. See the documentation for how to create these. Note, if more than one role is required for notebook instances, training, and/or hosting, please replace the boto regexp with a the appropriate full IAM role arn string(s)."
5050
]
5151
},
52+
{
53+
"cell_type": "code",
54+
"execution_count": null,
55+
"metadata": {},
56+
"outputs": [],
57+
"source": [
58+
"import sys\n",
59+
"!{sys.executable} -m pip install -qU \"sagemaker>=1.71.0,<2.0.0\""
60+
]
61+
},
5262
{
5363
"cell_type": "code",
5464
"execution_count": null,
@@ -91,7 +101,6 @@
91101
"import json\n",
92102
"from IPython.display import display\n",
93103
"from time import strftime, gmtime\n",
94-
"import sagemaker\n",
95104
"from sagemaker.serializers import CSVSerializer"
96105
]
97106
},

0 commit comments

Comments
 (0)