Updated scikit_learn_script_mode_local_training_and_serving.py with local mode config

Author: eitansela

.gitignore

@@ -213,3 +213,19 @@ xgboost_script_mode_local_training_and_serving/code/build/
 catboost_bring_your_own_container_local_training_toolkit/data/test/california_test.csv
 catboost_bring_your_own_container_local_training_toolkit/data/train/california_train.csv
 catboost_bring_your_own_container_local_training_toolkit/data/validation/california_validation.csv
+scikit_learn_script_mode_local_training_and_serving/code/build/lib/build/lib/scikit_learn_california.py
+scikit_learn_script_mode_local_training_and_serving/code/build/lib/MANIFEST.in
+scikit_learn_script_mode_local_training_and_serving/code/build/lib/scikit_learn_california.egg-info/dependency_links.txt
+scikit_learn_script_mode_local_training_and_serving/code/build/lib/scikit_learn_california.egg-info/PKG-INFO
+scikit_learn_script_mode_local_training_and_serving/code/build/lib/scikit_learn_california.egg-info/SOURCES.txt
+scikit_learn_script_mode_local_training_and_serving/code/build/lib/scikit_learn_california.egg-info/top_level.txt
+scikit_learn_script_mode_local_training_and_serving/code/build/lib/scikit_learn_california.py
+scikit_learn_script_mode_local_training_and_serving/code/build/lib/setup.cfg
+scikit_learn_script_mode_local_training_and_serving/code/build/lib/setup.py
+scikit_learn_script_mode_local_training_and_serving/code/MANIFEST.in
+scikit_learn_script_mode_local_training_and_serving/code/scikit_learn_california.egg-info/dependency_links.txt
+scikit_learn_script_mode_local_training_and_serving/code/scikit_learn_california.egg-info/PKG-INFO
+scikit_learn_script_mode_local_training_and_serving/code/scikit_learn_california.egg-info/SOURCES.txt
+scikit_learn_script_mode_local_training_and_serving/code/scikit_learn_california.egg-info/top_level.txt
+scikit_learn_script_mode_local_training_and_serving/code/setup.cfg
+scikit_learn_script_mode_local_training_and_serving/code/setup.py

scikit_learn_script_mode_local_training_and_serving/scikit_learn_script_mode_local_training_and_serving.py

@@ -16,6 +16,8 @@
 import os
 
 from sagemaker.sklearn import SKLearn
+from sagemaker.local import LocalSession
+
 import sagemaker
 import boto3
 from sklearn import datasets
@@ -27,12 +29,14 @@
 if local_mode:
     instance_type = "local"
     IAM_ROLE = 'arn:aws:iam::111111111111:role/service-role/AmazonSageMaker-ExecutionRole-20200101T000001'
+    sess = LocalSession()
+    sess.config = {'local': {'local_code': True}}  # Ensure full code locality, see: https://sagemaker.readthedocs.io/en/stable/overview.html#local-mode
 else:
     instance_type = "ml.m5.xlarge"
     IAM_ROLE = 'arn:aws:iam::<ACCOUNT>:role/service-role/AmazonSageMaker-ExecutionRole-XXX'
+    sess = sagemaker.Session()
+    bucket = sess.default_bucket()                    # Set a default S3 bucket
 
-sess = sagemaker.Session()
-bucket = sess.default_bucket()                    # Set a default S3 bucket
 prefix = 'DEMO-local-and-managed-infrastructure'
 
 def download_training_and_eval_data():
@@ -74,7 +78,8 @@ def main():
     sklearn = SKLearn(
         entry_point="scikit_learn_california.py",
         source_dir='code',
-        framework_version="1.0-1",
+        framework_version="1.2-1",
+        sagemaker_session=sess,
         instance_type=instance_type,
         role=IAM_ROLE,
         hyperparameters={"max_leaf_nodes": 30},
@@ -105,7 +110,7 @@ def main():
     do_inference_on_local_endpoint(predictor)
 
     print('About to delete the endpoint to stop paying (if in cloud mode).')
-    predictor.delete_endpoint(predictor.endpoint_name)
+    predictor.delete_endpoint()
 
 
 if __name__ == "__main__":