Updated XGBoost version to 1.7.1

Author: eitansela

xgboost_script_mode_local_training_and_serving/xgboost_script_mode_local_training_and_serving.py

@@ -16,7 +16,8 @@
 
 DUMMY_IAM_ROLE = 'arn:aws:iam::111111111111:role/service-role/AmazonSageMaker-ExecutionRole-20200101T000001'
 LOCAL_SESSION = LocalSession()
-LOCAL_SESSION.config={'local': {'local_code': True}} # Ensure full code locality, see: https://sagemaker.readthedocs.io/en/stable/overview.html#local-mode
+LOCAL_SESSION.config = {'local': {'local_code': True}}  # Ensure full code locality, see: https://sagemaker.readthedocs.io/en/stable/overview.html#local-mode
+FRAMEWORK_VERSION = "1.7-1"
 
 
 def do_inference_on_local_endpoint(predictor, libsvm_str):
@@ -27,7 +28,8 @@ def do_inference_on_local_endpoint(predictor, libsvm_str):
 
 def main():
     print('Starting model training.')
-    print('Note: if launching for the first time in local mode, container image download might take a few minutes to complete.')
+    print(
+        'Note: if launching for the first time in local mode, container image download might take a few minutes to complete.')
 
     hyperparameters = {
         "max_depth": "5",
@@ -46,9 +48,9 @@ def main():
         role=DUMMY_IAM_ROLE,
         instance_count=1,
         instance_type='local',
-        framework_version="1.2-1",
+        framework_version=FRAMEWORK_VERSION,
         sagemaker_session=LOCAL_SESSION,
-        output_path='file://model/'    # Save trained model and any additional artifacts locally
+        output_path='file://model/'  # Save trained model and any additional artifacts locally
     )
 
     train_input = TrainingInput("file://data/train/abalone", content_type="text/libsvm")
@@ -65,7 +67,7 @@ def main():
         role=DUMMY_IAM_ROLE,
         entry_point="inference.py",
         source_dir="./code",
-        framework_version="1.2-1",
+        framework_version=FRAMEWORK_VERSION,
         sagemaker_session=LOCAL_SESSION
     )