sagemaker 2.254.2.dev0__tar.gz → 2.256.0__tar.gz

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sagemaker
-Version: 2.254.2.dev0
+Version: 2.256.0
 Summary: Open source library for training and deploying models on Amazon SageMaker.
 Project-URL: Homepage, https://github.com/aws/sagemaker-python-sdk
 Author: Amazon Web Services

sagemaker-2.256.0/VERSION

@@ -0,0 +1 @@
+2.256.0

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/requirements/extras/test_requirements.txt

@@ -62,4 +62,4 @@ mypy-boto3-s3==1.35.76
 mypy-extensions==1.0.0
 mypy==1.9.0
 # apache-airflow transitive dependancy
-google-re2<1.1.20250805; python_version < "3.10"
+google-re2<1.1.20250805; python_version < "3.10"

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/estimator.py

@@ -1350,7 +1350,7 @@ class EstimatorBase(with_metaclass(ABCMeta, object)):  # pylint: disable=too-man
             )
         return None
 
-    @_telemetry_emitter(feature=Feature.ESTIMATOR, func_name="estimator.fit")
+    @_telemetry_emitter(feature=Feature.ESTIMATOR_V2, func_name="estimator.fit")
     @runnable_by_pipeline
     def fit(
         self,

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/experiments/experiment.py

@@ -95,7 +95,7 @@ class Experiment(_base_types.Record):
         )
 
     @classmethod
-    @_telemetry_emitter(feature=Feature.MLOPS, func_name="experiment.create")
+    @_telemetry_emitter(feature=Feature.MLOPS_V2, func_name="experiment.create")
     def create(
         cls,
         experiment_name,

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/feature_store/feature_processor/_config_uploader.py

@@ -120,9 +120,6 @@ class ConfigUploader:
         stored_function = StoredFunction(
             sagemaker_session=sagemaker_session,
             s3_base_uri=s3_base_uri,
-            hmac_key=self.remote_decorator_config.environment_variables[
-                "REMOTE_FUNCTION_SECRET_KEY"
-            ],
             s3_kms_key=self.remote_decorator_config.s3_kms_key,
         )
         stored_function.save(func)

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/image_uri_config/xgboost.json

@@ -395,6 +395,48 @@
                     "us-west-2": "246618743249"
                 },
                 "repository": "sagemaker-xgboost"
+            },
+            "3.0-5": {
+                "registries": {
+                    "af-south-1": "510948584623",
+                    "ap-east-1": "651117190479",
+                    "ap-northeast-1": "354813040037",
+                    "ap-northeast-2": "366743142698",
+                    "ap-northeast-3": "867004704886",
+                    "ap-south-1": "720646828776",
+                    "ap-south-2": "628508329040",
+                    "ap-southeast-1": "121021644041",
+                    "ap-southeast-2": "783357654285",
+                    "ap-southeast-3": "951798379941",
+                    "ap-southeast-4": "106583098589",
+                    "ca-central-1": "341280168497",
+                    "ca-west-1": "190319476487",
+                    "cn-north-1": "450853457545",
+                    "cn-northwest-1": "451049120500",
+                    "eu-central-1": "492215442770",
+                    "eu-central-2": "680994064768",
+                    "eu-north-1": "662702820516",
+                    "eu-south-1": "978288397137",
+                    "eu-south-2": "104374241257",
+                    "eu-west-1": "141502667606",
+                    "eu-west-2": "764974769150",
+                    "eu-west-3": "659782779980",
+                    "il-central-1": "898809789911",
+                    "me-central-1": "272398656194",
+                    "me-south-1": "801668240914",
+                    "sa-east-1": "737474898029",
+                    "us-east-1": "683313688378",
+                    "us-east-2": "257758044811",
+                    "us-gov-east-1": "237065988967",
+                    "us-gov-west-1": "414596584902",
+                    "us-iso-east-1": "833128469047",
+                    "us-isob-east-1": "281123927165",
+                    "us-isof-east-1": "108575199400",
+                    "us-isof-south-1": "124985052026",
+                    "us-west-1": "746614075791",
+                    "us-west-2": "246618743249"
+                },
+                "repository": "sagemaker-xgboost"
             }
         }
     },
@@ -794,6 +836,48 @@
                     "us-west-2": "246618743249"
                 },
                 "repository": "sagemaker-xgboost"
+            },
+            "3.0-5": {
+                "registries": {
+                    "af-south-1": "510948584623",
+                    "ap-east-1": "651117190479",
+                    "ap-northeast-1": "354813040037",
+                    "ap-northeast-2": "366743142698",
+                    "ap-northeast-3": "867004704886",
+                    "ap-south-1": "720646828776",
+                    "ap-south-2": "628508329040",
+                    "ap-southeast-1": "121021644041",
+                    "ap-southeast-2": "783357654285",
+                    "ap-southeast-3": "951798379941",
+                    "ap-southeast-4": "106583098589",
+                    "ca-central-1": "341280168497",
+                    "ca-west-1": "190319476487",
+                    "cn-north-1": "450853457545",
+                    "cn-northwest-1": "451049120500",
+                    "eu-central-1": "492215442770",
+                    "eu-central-2": "680994064768",
+                    "eu-north-1": "662702820516",
+                    "eu-south-1": "978288397137",
+                    "eu-south-2": "104374241257",
+                    "eu-west-1": "141502667606",
+                    "eu-west-2": "764974769150",
+                    "eu-west-3": "659782779980",
+                    "il-central-1": "898809789911",
+                    "me-central-1": "272398656194",
+                    "me-south-1": "801668240914",
+                    "sa-east-1": "737474898029",
+                    "us-east-1": "683313688378",
+                    "us-east-2": "257758044811",
+                    "us-gov-east-1": "237065988967",
+                    "us-gov-west-1": "414596584902",
+                    "us-iso-east-1": "833128469047",
+                    "us-isob-east-1": "281123927165",
+                    "us-isof-east-1": "108575199400",
+                    "us-isof-south-1": "124985052026",
+                    "us-west-1": "746614075791",
+                    "us-west-2": "246618743249"
+                },
+                "repository": "sagemaker-xgboost"
             }
         }
     },

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/iterators.py

@@ -17,6 +17,7 @@ from abc import ABC, abstractmethod
 import io
 
 from sagemaker.exceptions import ModelStreamError, InternalStreamFailure
+from sagemaker.utils import _MAX_BUFFER_SIZE
 
 
 def handle_stream_errors(chunk):
@@ -182,5 +183,15 @@ class LineIterator(BaseIterator):
                 # print and move on to next response byte
                 print("Unknown event type:" + chunk)
                 continue
+
+            # Check buffer size before writing to prevent unbounded memory consumption
+            chunk_size = len(chunk["PayloadPart"]["Bytes"])
+            current_size = self.buffer.getbuffer().nbytes
+            if current_size + chunk_size > _MAX_BUFFER_SIZE:
+                raise RuntimeError(
+                    f"Line buffer exceeded maximum size of {_MAX_BUFFER_SIZE} bytes. "
+                    f"No newline found in stream."
+                )
+
             self.buffer.seek(0, io.SEEK_END)
             self.buffer.write(chunk["PayloadPart"]["Bytes"])

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/jumpstart/estimator.py

@@ -62,7 +62,7 @@ class JumpStartEstimator(Estimator):
     This class sets defaults based on the model ID and version.
     """
 
-    @_telemetry_emitter(feature=Feature.JUMPSTART, func_name="jumpstart_estimator.create")
+    @_telemetry_emitter(feature=Feature.JUMPSTART_V2, func_name="jumpstart_estimator.create")
     def __init__(
         self,
         model_id: Optional[str] = None,
@@ -649,7 +649,7 @@ class JumpStartEstimator(Estimator):
 
         super(JumpStartEstimator, self).__init__(**estimator_init_kwargs.to_kwargs_dict())
 
-    @_telemetry_emitter(feature=Feature.JUMPSTART, func_name="jumpstart_estimator.fit")
+    @_telemetry_emitter(feature=Feature.JUMPSTART_V2, func_name="jumpstart_estimator.fit")
     def fit(
         self,
         inputs: Optional[Union[str, Dict, TrainingInput, FileSystemInput]] = None,
@@ -837,7 +837,7 @@ class JumpStartEstimator(Estimator):
             additional_kwargs=additional_kwargs,
         )
 
-    @_telemetry_emitter(feature=Feature.JUMPSTART, func_name="jumpstart_estimator.deploy")
+    @_telemetry_emitter(feature=Feature.JUMPSTART_V2, func_name="jumpstart_estimator.deploy")
     def deploy(
         self,
         initial_instance_count: Optional[int] = None,

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/jumpstart/model.py

@@ -86,7 +86,7 @@ class JumpStartModel(Model):
     This class sets defaults based on the model ID and version.
     """
 
-    @_telemetry_emitter(feature=Feature.JUMPSTART, func_name="jumpstart_model.create")
+    @_telemetry_emitter(feature=Feature.JUMPSTART_V2, func_name="jumpstart_model.create")
     def __init__(
         self,
         model_id: Optional[str] = None,
@@ -643,7 +643,7 @@ class JumpStartModel(Model):
                 **kwargs,
             )
 
-    @_telemetry_emitter(feature=Feature.JUMPSTART, func_name="jumpstart_model.deploy")
+    @_telemetry_emitter(feature=Feature.JUMPSTART_V2, func_name="jumpstart_model.deploy")
     def deploy(
         self,
         initial_instance_count: Optional[int] = None,

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/local/data.py

@@ -26,6 +26,7 @@ from six.moves.urllib.parse import urlparse
 import sagemaker.amazon.common
 import sagemaker.local.utils
 import sagemaker.utils
+from sagemaker.utils import _SENSITIVE_SYSTEM_PATHS
 
 
 def get_data_source_instance(data_source, sagemaker_session):
@@ -122,6 +123,15 @@ class LocalFileDataSource(DataSource):
         super(LocalFileDataSource, self).__init__()
 
         self.root_path = os.path.abspath(root_path)
+
+        # Validate that the path is not in restricted locations
+        for restricted_path in _SENSITIVE_SYSTEM_PATHS:
+            if self.root_path != "/" and self.root_path.startswith(restricted_path):
+                raise ValueError(
+                    f"Local Mode does not support mounting from restricted system paths. "
+                    f"Got: {root_path}"
+                )
+
         if not os.path.exists(self.root_path):
             raise RuntimeError("Invalid data source: %s does not exist." % self.root_path)
 

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/local/local_session.py

@@ -85,7 +85,7 @@ class LocalSagemakerClient(object):  # pylint: disable=too-many-public-methods
         """
         self.sagemaker_session = sagemaker_session or LocalSession()
 
-    @_telemetry_emitter(Feature.LOCAL_MODE, "local_session.create_processing_job")
+    @_telemetry_emitter(Feature.LOCAL_MODE_V2, "local_session.create_processing_job")
     def create_processing_job(
         self,
         ProcessingJobName,
@@ -168,7 +168,7 @@ class LocalSagemakerClient(object):  # pylint: disable=too-many-public-methods
             raise ClientError(error_response, "describe_processing_job")
         return LocalSagemakerClient._processing_jobs[ProcessingJobName].describe()
 
-    @_telemetry_emitter(Feature.LOCAL_MODE, "local_session.create_training_job")
+    @_telemetry_emitter(Feature.LOCAL_MODE_V2, "local_session.create_training_job")
     def create_training_job(
         self,
         TrainingJobName,
@@ -239,7 +239,7 @@ class LocalSagemakerClient(object):  # pylint: disable=too-many-public-methods
             raise ClientError(error_response, "describe_training_job")
         return LocalSagemakerClient._training_jobs[TrainingJobName].describe()
 
-    @_telemetry_emitter(Feature.LOCAL_MODE, "local_session.create_transform_job")
+    @_telemetry_emitter(Feature.LOCAL_MODE_V2, "local_session.create_transform_job")
     def create_transform_job(
         self,
         TransformJobName,
@@ -285,7 +285,7 @@ class LocalSagemakerClient(object):  # pylint: disable=too-many-public-methods
             raise ClientError(error_response, "describe_transform_job")
         return LocalSagemakerClient._transform_jobs[TransformJobName].describe()
 
-    @_telemetry_emitter(Feature.LOCAL_MODE, "local_session.create_model")
+    @_telemetry_emitter(Feature.LOCAL_MODE_V2, "local_session.create_model")
     def create_model(
         self, ModelName, PrimaryContainer, *args, **kwargs
     ):  # pylint: disable=unused-argument
@@ -335,7 +335,7 @@ class LocalSagemakerClient(object):  # pylint: disable=too-many-public-methods
             raise ClientError(error_response, "describe_endpoint_config")
         return LocalSagemakerClient._endpoint_configs[EndpointConfigName].describe()
 
-    @_telemetry_emitter(Feature.LOCAL_MODE, "local_session.create_endpoint_config")
+    @_telemetry_emitter(Feature.LOCAL_MODE_V2, "local_session.create_endpoint_config")
     def create_endpoint_config(self, EndpointConfigName, ProductionVariants, Tags=None):
         """Create the endpoint configuration.
 
@@ -367,7 +367,7 @@ class LocalSagemakerClient(object):  # pylint: disable=too-many-public-methods
             raise ClientError(error_response, "describe_endpoint")
         return LocalSagemakerClient._endpoints[EndpointName].describe()
 
-    @_telemetry_emitter(Feature.LOCAL_MODE, "local_session.create_endpoint")
+    @_telemetry_emitter(Feature.LOCAL_MODE_V2, "local_session.create_endpoint")
     def create_endpoint(self, EndpointName, EndpointConfigName, Tags=None):
         """Create the endpoint.
 
@@ -436,7 +436,7 @@ class LocalSagemakerClient(object):  # pylint: disable=too-many-public-methods
         if ModelName in LocalSagemakerClient._models:
             del LocalSagemakerClient._models[ModelName]
 
-    @_telemetry_emitter(Feature.LOCAL_MODE, "local_session.create_pipeline")
+    @_telemetry_emitter(Feature.LOCAL_MODE_V2, "local_session.create_pipeline")
     def create_pipeline(
         self, pipeline, pipeline_description, **kwargs  # pylint: disable=unused-argument
     ):

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/local/utils.py

@@ -48,10 +48,7 @@ def copy_directory_structure(destination_directory, relative_path):
             destination_directory
     """
     full_path = os.path.join(destination_directory, relative_path)
-    if os.path.exists(full_path):
-        return
-
-    os.makedirs(destination_directory, relative_path)
+    os.makedirs(full_path, exist_ok=True)
 
 
 def move_to_destination(source, destination, job_name, sagemaker_session, prefix=""):

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/modules/configs.py

@@ -248,7 +248,8 @@ class OutputDataConfig(shapes.OutputDataConfig):
     """OutputDataConfig.
 
     The OutputDataConfig class is a subclass of ``sagemaker_core.shapes.OutputDataConfig``
-    and allows the user to specify the output data configuration for the training job.
+    and allows the user to specify the output data configuration for the training job
+    (will not be carried over to any model repository or deployment).
 
     Parameters:
         s3_output_path (Optional[str]):

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/modules/train/model_trainer.py

@@ -807,7 +807,7 @@ class ModelTrainer(BaseModel):
             args["session_chaining_config"] = self._session_chaining_config
             return args
 
-    @_telemetry_emitter(feature=Feature.MODEL_TRAINER, func_name="model_trainer.train")
+    @_telemetry_emitter(feature=Feature.MODEL_TRAINER_V2, func_name="model_trainer.train")
     @validate_call
     def train(
         self,

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/processing.py

@@ -1927,7 +1927,7 @@ class FrameworkProcessor(ScriptProcessor):
                 # Some py3 containers has typing, which may breaks pip install
                 pip uninstall --yes typing
 
-                pip install -r requirements.txt
+                python -m pip install -r requirements.txt
             fi
 
             {entry_point_command} {entry_point} "$@"

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/remote_function/client.py

@@ -59,7 +59,7 @@ _FINISHED = "FINISHED"
 logger = logging_config.get_logger()
 
 
-@_telemetry_emitter(feature=Feature.REMOTE_FUNCTION, func_name="remote_function.remote")
+@_telemetry_emitter(feature=Feature.REMOTE_FUNCTION_V2, func_name="remote_function.remote")
 def remote(
     _func=None,
     *,
@@ -362,7 +362,6 @@ def remote(
                             s3_uri=s3_path_join(
                                 job_settings.s3_root_uri, job.job_name, EXCEPTION_FOLDER
                             ),
-                            hmac_key=job.hmac_key,
                         )
                     except ServiceError as serr:
                         chained_e = serr.__cause__
@@ -399,7 +398,6 @@ def remote(
                 return serialization.deserialize_obj_from_s3(
                     sagemaker_session=job_settings.sagemaker_session,
                     s3_uri=s3_path_join(job_settings.s3_root_uri, job.job_name, RESULTS_FOLDER),
-                    hmac_key=job.hmac_key,
                 )
 
             if job.describe()["TrainingJobStatus"] == "Stopped":
@@ -979,7 +977,6 @@ class Future(object):
                 job_return = serialization.deserialize_obj_from_s3(
                     sagemaker_session=sagemaker_session,
                     s3_uri=s3_path_join(job.s3_uri, RESULTS_FOLDER),
-                    hmac_key=job.hmac_key,
                 )
             except DeserializationError as e:
                 client_exception = e
@@ -991,7 +988,6 @@ class Future(object):
                 job_exception = serialization.deserialize_exception_from_s3(
                     sagemaker_session=sagemaker_session,
                     s3_uri=s3_path_join(job.s3_uri, EXCEPTION_FOLDER),
-                    hmac_key=job.hmac_key,
                 )
             except ServiceError as serr:
                 chained_e = serr.__cause__
@@ -1081,7 +1077,6 @@ class Future(object):
                     self._return = serialization.deserialize_obj_from_s3(
                         sagemaker_session=self._job.sagemaker_session,
                         s3_uri=s3_path_join(self._job.s3_uri, RESULTS_FOLDER),
-                        hmac_key=self._job.hmac_key,
                     )
                     self._state = _FINISHED
                     return self._return
@@ -1090,7 +1085,6 @@ class Future(object):
                         self._exception = serialization.deserialize_exception_from_s3(
                             sagemaker_session=self._job.sagemaker_session,
                             s3_uri=s3_path_join(self._job.s3_uri, EXCEPTION_FOLDER),
-                            hmac_key=self._job.hmac_key,
                         )
                     except ServiceError as serr:
                         chained_e = serr.__cause__

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/remote_function/core/pipeline_variables.py

@@ -164,7 +164,6 @@ class _DelayedReturnResolver:
     def __init__(
         self,
         delayed_returns: List[_DelayedReturn],
-        hmac_key: str,
         properties_resolver: _PropertiesResolver,
         parameter_resolver: _ParameterResolver,
         execution_variable_resolver: _ExecutionVariableResolver,
@@ -175,7 +174,6 @@ class _DelayedReturnResolver:
 
         Args:
             delayed_returns: list of delayed returns to resolve.
-            hmac_key: key used to encrypt serialized and deserialized function and arguments.
             properties_resolver: resolver used to resolve step properties.
             parameter_resolver: resolver used to pipeline parameters.
             execution_variable_resolver: resolver used to resolve execution variables.
@@ -197,7 +195,6 @@ class _DelayedReturnResolver:
             return uri, deserialize_obj_from_s3(
                 sagemaker_session=settings["sagemaker_session"],
                 s3_uri=uri,
-                hmac_key=hmac_key,
             )
 
         with ThreadPoolExecutor() as executor:
@@ -247,7 +244,6 @@ def resolve_pipeline_variables(
     context: Context,
     func_args: Tuple,
     func_kwargs: Dict,
-    hmac_key: str,
     s3_base_uri: str,
     **settings,
 ):
@@ -257,7 +253,6 @@ def resolve_pipeline_variables(
         context: context for the execution.
         func_args: function args.
         func_kwargs: function kwargs.
-        hmac_key: key used to encrypt serialized and deserialized function and arguments.
         s3_base_uri: the s3 base uri of the function step that the serialized artifacts
             will be uploaded to. The s3_base_uri = s3_root_uri + pipeline_name.
         **settings: settings to pass to the deserialization function.
@@ -280,7 +275,6 @@ def resolve_pipeline_variables(
     properties_resolver = _PropertiesResolver(context)
     delayed_return_resolver = _DelayedReturnResolver(
         delayed_returns=delayed_returns,
-        hmac_key=hmac_key,
         properties_resolver=properties_resolver,
         parameter_resolver=parameter_resolver,
         execution_variable_resolver=execution_variable_resolver,

{sagemaker-2.254.2.dev0 → sagemaker-2.256.0}/src/sagemaker/remote_function/core/serialization.py

@@ -152,7 +152,7 @@ class CloudpickleSerializer:
 
 # TODO: use dask serializer in case dask distributed is installed in users' environment.
 def serialize_func_to_s3(
-    func: Callable, sagemaker_session: Session, s3_uri: str, hmac_key: str, s3_kms_key: str = None
+    func: Callable, sagemaker_session: Session, s3_uri: str, s3_kms_key: str = None
 ):
     """Serializes function and uploads it to S3.
 
@@ -160,7 +160,6 @@ def serialize_func_to_s3(
         sagemaker_session (sagemaker.session.Session):
             The underlying Boto3 session which AWS service calls are delegated to.
         s3_uri (str): S3 root uri to which resulting serialized artifacts will be uploaded.
-        hmac_key (str): Key used to calculate hmac-sha256 hash of the serialized func.
         s3_kms_key (str): KMS key used to encrypt artifacts uploaded to S3.
         func: function to be serialized and persisted
     Raises:
@@ -169,14 +168,13 @@ def serialize_func_to_s3(
 
     _upload_payload_and_metadata_to_s3(
         bytes_to_upload=CloudpickleSerializer.serialize(func),
-        hmac_key=hmac_key,
         s3_uri=s3_uri,
         sagemaker_session=sagemaker_session,
         s3_kms_key=s3_kms_key,
     )
 
 
-def deserialize_func_from_s3(sagemaker_session: Session, s3_uri: str, hmac_key: str) -> Callable:
+def deserialize_func_from_s3(sagemaker_session: Session, s3_uri: str) -> Callable:
     """Downloads from S3 and then deserializes data objects.
 
     This method downloads the serialized training job outputs to a temporary directory and
@@ -186,7 +184,6 @@ def deserialize_func_from_s3(sagemaker_session: Session, s3_uri: str, hmac_key:
         sagemaker_session (sagemaker.session.Session):
             The underlying sagemaker session which AWS service calls are delegated to.
         s3_uri (str): S3 root uri to which resulting serialized artifacts will be uploaded.
-        hmac_key (str): Key used to calculate hmac-sha256 hash of the serialized func.
     Returns :
         The deserialized function.
     Raises:
@@ -198,16 +195,12 @@ def deserialize_func_from_s3(sagemaker_session: Session, s3_uri: str, hmac_key:
 
     bytes_to_deserialize = _read_bytes_from_s3(f"{s3_uri}/payload.pkl", sagemaker_session)
 
-    _perform_integrity_check(
-        expected_hash_value=metadata.sha256_hash, secret_key=hmac_key, buffer=bytes_to_deserialize
-    )
+    _perform_integrity_check(expected_hash_value=metadata.sha256_hash, buffer=bytes_to_deserialize)
 
     return CloudpickleSerializer.deserialize(f"{s3_uri}/payload.pkl", bytes_to_deserialize)
 
 
-def serialize_obj_to_s3(
-    obj: Any, sagemaker_session: Session, s3_uri: str, hmac_key: str, s3_kms_key: str = None
-):
+def serialize_obj_to_s3(obj: Any, sagemaker_session: Session, s3_uri: str, s3_kms_key: str = None):
     """Serializes data object and uploads it to S3.
 
     Args:
@@ -215,7 +208,6 @@ def serialize_obj_to_s3(
             The underlying Boto3 session which AWS service calls are delegated to.
         s3_uri (str): S3 root uri to which resulting serialized artifacts will be uploaded.
         s3_kms_key (str): KMS key used to encrypt artifacts uploaded to S3.
-        hmac_key (str): Key used to calculate hmac-sha256 hash of the serialized obj.
         obj: object to be serialized and persisted
     Raises:
         SerializationError: when fail to serialize object to bytes.
@@ -223,7 +215,6 @@ def serialize_obj_to_s3(
 
     _upload_payload_and_metadata_to_s3(
         bytes_to_upload=CloudpickleSerializer.serialize(obj),
-        hmac_key=hmac_key,
         s3_uri=s3_uri,
         sagemaker_session=sagemaker_session,
         s3_kms_key=s3_kms_key,
@@ -270,14 +261,13 @@ def json_serialize_obj_to_s3(
     )
 
 
-def deserialize_obj_from_s3(sagemaker_session: Session, s3_uri: str, hmac_key: str) -> Any:
+def deserialize_obj_from_s3(sagemaker_session: Session, s3_uri: str) -> Any:
     """Downloads from S3 and then deserializes data objects.
 
     Args:
         sagemaker_session (sagemaker.session.Session):
             The underlying sagemaker session which AWS service calls are delegated to.
         s3_uri (str): S3 root uri to which resulting serialized artifacts will be uploaded.
-        hmac_key (str): Key used to calculate hmac-sha256 hash of the serialized obj.
     Returns :
         Deserialized python objects.
     Raises:
@@ -290,15 +280,13 @@ def deserialize_obj_from_s3(sagemaker_session: Session, s3_uri: str, hmac_key: s
 
     bytes_to_deserialize = _read_bytes_from_s3(f"{s3_uri}/payload.pkl", sagemaker_session)
 
-    _perform_integrity_check(
-        expected_hash_value=metadata.sha256_hash, secret_key=hmac_key, buffer=bytes_to_deserialize
-    )
+    _perform_integrity_check(expected_hash_value=metadata.sha256_hash, buffer=bytes_to_deserialize)
 
     return CloudpickleSerializer.deserialize(f"{s3_uri}/payload.pkl", bytes_to_deserialize)
 
 
 def serialize_exception_to_s3(
-    exc: Exception, sagemaker_session: Session, s3_uri: str, hmac_key: str, s3_kms_key: str = None
+    exc: Exception, sagemaker_session: Session, s3_uri: str, s3_kms_key: str = None
 ):
     """Serializes exception with traceback and uploads it to S3.
 
@@ -307,7 +295,6 @@ def serialize_exception_to_s3(
             The underlying Boto3 session which AWS service calls are delegated to.
         s3_uri (str): S3 root uri to which resulting serialized artifacts will be uploaded.
         s3_kms_key (str): KMS key used to encrypt artifacts uploaded to S3.
-        hmac_key (str): Key used to calculate hmac-sha256 hash of the serialized exception.
         exc: Exception to be serialized and persisted
     Raises:
         SerializationError: when fail to serialize object to bytes.
@@ -316,7 +303,6 @@ def serialize_exception_to_s3(
 
     _upload_payload_and_metadata_to_s3(
         bytes_to_upload=CloudpickleSerializer.serialize(exc),
-        hmac_key=hmac_key,
         s3_uri=s3_uri,
         sagemaker_session=sagemaker_session,
         s3_kms_key=s3_kms_key,
@@ -325,7 +311,6 @@ def serialize_exception_to_s3(
 
 def _upload_payload_and_metadata_to_s3(
     bytes_to_upload: Union[bytes, io.BytesIO],
-    hmac_key: str,
     s3_uri: str,
     sagemaker_session: Session,
     s3_kms_key,
@@ -334,7 +319,6 @@ def _upload_payload_and_metadata_to_s3(
 
     Args:
         bytes_to_upload (bytes): Serialized bytes to upload.
-        hmac_key (str): Key used to calculate hmac-sha256 hash of the serialized obj.
         s3_uri (str): S3 root uri to which resulting serialized artifacts will be uploaded.
         sagemaker_session (sagemaker.session.Session):
             The underlying Boto3 session which AWS service calls are delegated to.
@@ -342,7 +326,7 @@ def _upload_payload_and_metadata_to_s3(
     """
     _upload_bytes_to_s3(bytes_to_upload, f"{s3_uri}/payload.pkl", s3_kms_key, sagemaker_session)
 
-    sha256_hash = _compute_hash(bytes_to_upload, secret_key=hmac_key)
+    sha256_hash = _compute_hash(bytes_to_upload)
 
     _upload_bytes_to_s3(
         _MetaData(sha256_hash).to_json(),
@@ -352,14 +336,13 @@ def _upload_payload_and_metadata_to_s3(
     )
 
 
-def deserialize_exception_from_s3(sagemaker_session: Session, s3_uri: str, hmac_key: str) -> Any:
+def deserialize_exception_from_s3(sagemaker_session: Session, s3_uri: str) -> Any:
     """Downloads from S3 and then deserializes exception.
 
     Args:
         sagemaker_session (sagemaker.session.Session):
             The underlying sagemaker session which AWS service calls are delegated to.
         s3_uri (str): S3 root uri to which resulting serialized artifacts will be uploaded.
-        hmac_key (str): Key used to calculate hmac-sha256 hash of the serialized exception.
     Returns :
         Deserialized exception with traceback.
     Raises:
@@ -372,9 +355,7 @@ def deserialize_exception_from_s3(sagemaker_session: Session, s3_uri: str, hmac_
 
     bytes_to_deserialize = _read_bytes_from_s3(f"{s3_uri}/payload.pkl", sagemaker_session)
 
-    _perform_integrity_check(
-        expected_hash_value=metadata.sha256_hash, secret_key=hmac_key, buffer=bytes_to_deserialize
-    )
+    _perform_integrity_check(expected_hash_value=metadata.sha256_hash, buffer=bytes_to_deserialize)
 
     return CloudpickleSerializer.deserialize(f"{s3_uri}/payload.pkl", bytes_to_deserialize)
 
@@ -399,18 +380,18 @@ def _read_bytes_from_s3(s3_uri, sagemaker_session):
         ) from e
 
 
-def _compute_hash(buffer: bytes, secret_key: str) -> str:
-    """Compute the hmac-sha256 hash"""
-    return hmac.new(secret_key.encode(), msg=buffer, digestmod=hashlib.sha256).hexdigest()
+def _compute_hash(buffer: bytes) -> str:
+    """Compute the sha256 hash"""
+    return hashlib.sha256(buffer).hexdigest()
 
 
-def _perform_integrity_check(expected_hash_value: str, secret_key: str, buffer: bytes):
+def _perform_integrity_check(expected_hash_value: str, buffer: bytes):
     """Performs integrity checks for serialized code/arguments uploaded to s3.
 
     Verifies whether the hash read from s3 matches the hash calculated
     during remote function execution.
     """
-    actual_hash_value = _compute_hash(buffer=buffer, secret_key=secret_key)
+    actual_hash_value = _compute_hash(buffer=buffer)
     if not hmac.compare_digest(expected_hash_value, actual_hash_value):
         raise DeserializationError(
             "Integrity check for the serialized function or data failed. "