change: update sagemaker version for pathways to 2.132.1

Author: Rohan Gujarathi

src/sagemaker/remote_function/job.py

@@ -42,7 +42,7 @@
 SAGEMAKER_WHL_CHANNEL_NAME = "sagemaker_whl_file"
 
 SAGEMAKER_SDK_WHL_FILE = (
-    "s3://sagemaker-pathways/test/pysdk/sagemaker-2.120.1.dev0-py2.py3-none-any.whl"
+    "s3://sagemaker-pathways/test/pysdk/sagemaker-2.132.1.dev0-py2.py3-none-any.whl"
 )
 
 JOBS_CONTAINER_ENTRYPOINT = [

src/sagemaker/remote_function/runtime_environment/bootstrap_runtime_environment.py

@@ -31,7 +31,7 @@
 REMOTE_FUNCTION_WORKSPACE = "sm_rf_user_ws"
 BASE_CHANNEL_PATH = "/opt/ml/input/data"
 FAILURE_REASON_PATH = "/opt/ml/output/failure"
-SAGEMAKER_WHL_FILE_NAME = "sagemaker-2.120.1.dev0-py2.py3-none-any.whl"
+SAGEMAKER_WHL_FILE_NAME = "sagemaker-2.132.1.dev0-py2.py3-none-any.whl"
 SAGEMAKER_WHL_CHANNEL = "sagemaker_whl_file"
 
 

src/sagemaker/remote_function/runtime_environment/runtime_environment_manager.py

@@ -154,8 +154,8 @@ def _export_conda_env(self, env_name, local_path):
     def _replace_sagemaker_in_conda_env_yml(self, data):
         """Replace the sagemaker in conda yml file with local sagemaker installer"""
 
-        sagemaker_with_version = b"sagemaker==2.120.1.dev0"
-        sagemaker_whl_file = b"sagemaker-2.120.1.dev0-py2.py3-none-any.whl[remote_function]"
+        sagemaker_with_version = b"sagemaker==2.132.1.dev0"
+        sagemaker_whl_file = b"sagemaker-2.132.1.dev0-py2.py3-none-any.whl[remote_function]"
 
         return data.replace(sagemaker_with_version, sagemaker_whl_file)
 

src/sagemaker/session.py

@@ -187,7 +187,7 @@ def _initialize(
         else:
             self.sagemaker_metrics_client = self.boto_session.client("sagemaker-metrics")
         prepend_user_agent(self.sagemaker_metrics_client)
-        
+
         self.s3_client = self.boto_session.client("s3", region_name=self.boto_region_name)
         self.s3_resource = self.boto_session.resource("s3", region_name=self.boto_region_name)
 

tests/integ/sagemaker/remote_function/conftest.py

@@ -70,7 +70,7 @@
     "dependencies:\n"
     "  - scipy=1.10.0\n"
     "  - pip:\n"
-    "    - sagemaker-2.120.1.dev0-py2.py3-none-any.whl[remote_function]\n"
+    "    - sagemaker-2.132.1.dev0-py2.py3-none-any.whl[remote_function]\n"
     "prefix: /opt/conda/bin/conda\n"
 )
 

tests/unit/sagemaker/remote_function/test_job.py

@@ -39,7 +39,7 @@
 REMOTE_FUNCTION_WORKSPACE = "sm_rf_user_ws"
 SAGEMAKER_WHL_CHANNEL_NAME = "sagemaker_whl_file"
 SAGEMAKER_SDK_WHL_FILE = (
-    "s3://sagemaker-pathways/test/pysdk/sagemaker-2.120.1.dev0-py2.py3-none-any.whl"
+    "s3://sagemaker-pathways/test/pysdk/sagemaker-2.132.1.dev0-py2.py3-none-any.whl"
 )
 
 EXPECTED_FUNCTION_URI = S3_URI + "/function.pkl"