fix: Fix Experiment Run integ test w.r.t unexpected boto3 version

Author: qidewenwhen

tests/data/experiment/inference.py

@@ -10,21 +10,21 @@
 #  on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 #  express or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
-import json
-import logging
 import os
-import pickle as pkl
-
-import boto3
-import numpy as np
-import sagemaker_xgboost_container.encoder as xgb_encoders
 
 sdk_name = "sagemaker-dev-1.0.tar.gz"
 code_dir = "/opt/ml/code"
 
 sdk_file = f"{code_dir}/{sdk_name}"
 os.system(f"pip install {sdk_file}")
 
+import json
+import logging
+import pickle as pkl
+import boto3
+import numpy as np
+import sagemaker_xgboost_container.encoder as xgb_encoders
+
 
 def _get_client_config_in_dict(cfg_in_str) -> dict:
     return json.loads(cfg_in_str) if cfg_in_str else None

tests/data/experiment/process_job_script_for_run_clz.py

@@ -13,15 +13,14 @@
 """This script file runs on SageMaker processing job"""
 from __future__ import absolute_import
 
-import json
-import logging
 import os
-import boto3
 
 sdk_file = "sagemaker-dev-1.0.tar.gz"
 os.system(f"pip install {sdk_file}")
 
-
+import json
+import logging
+import boto3
 from sagemaker import Session
 from sagemaker.experiments import load_run
 

tests/data/experiment/train_job_script_for_run_clz.py

@@ -13,15 +13,15 @@
 """This script file runs on SageMaker training job"""
 from __future__ import absolute_import
 
-import json
-import logging
-import time
 import os
-import boto3
 
 sdk_file = "sagemaker-dev-1.0.tar.gz"
 os.system(f"pip install {sdk_file}")
 
+import json
+import logging
+import time
+import boto3
 from sagemaker import Session
 from sagemaker.experiments import load_run, Run
 

tests/integ/sagemaker/experiments/test_run.py

@@ -171,7 +171,6 @@ def verify_is_run():
 _RUN_LOAD = "load"
 
 
-@pytest.mark.skip(reason="re:Invent keynote3 blocker. Revisit after release")
 def test_run_from_local_and_train_job_and_all_exp_cfg_match(
     sagemaker_session,
     dev_sdk_tar,
@@ -272,7 +271,6 @@ def test_run_from_local_and_train_job_and_all_exp_cfg_match(
         )
 
 
-@pytest.mark.skip(reason="re:Invent keynote3 blocker. Revisit after release")
 def test_run_from_local_and_train_job_and_exp_cfg_not_match(
     sagemaker_session,
     dev_sdk_tar,
@@ -359,7 +357,6 @@ def test_run_from_local_and_train_job_and_exp_cfg_not_match(
         )
 
 
-@pytest.mark.skip(reason="re:Invent keynote3 blocker. Revisit after release")
 def test_run_from_train_job_only(
     sagemaker_session,
     dev_sdk_tar,
@@ -409,7 +406,6 @@ def test_run_from_train_job_only(
 
 
 # dev_sdk_tar is required to trigger generating the dev SDK tar
-@pytest.mark.skip(reason="re:Invent keynote3 blocker. Revisit after release")
 def test_run_from_processing_job_and_override_default_exp_config(
     sagemaker_session,
     dev_sdk_tar,
@@ -571,7 +567,6 @@ def test_run_from_transform_job(
 
 
 # dev_sdk_tar is required to trigger generating the dev SDK tar
-@pytest.mark.skip(reason="re:Invent keynote3 blocker. Revisit after release")
 def test_load_run_auto_pass_in_exp_config_to_job(
     sagemaker_session,
     dev_sdk_tar,