sagemaker-core 2.1.1__py3-none-any.whl → 2.3.1__py3-none-any.whl

sagemaker/core/remote_function/core/serialization.py

@@ -19,7 +19,6 @@ import json
 import io
 
 import sys
-import hmac
 import hashlib
 import pickle
 
@@ -156,7 +155,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.
 
@@ -164,7 +163,6 @@ def serialize_func_to_s3(
         sagemaker_session (sagemaker.core.helper.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:
@@ -173,14 +171,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
@@ -190,7 +187,6 @@ def deserialize_func_from_s3(sagemaker_session: Session, s3_uri: str, hmac_key:
         sagemaker_session (sagemaker.core.helper.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:
@@ -203,14 +199,14 @@ 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
+        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
+    obj: Any, sagemaker_session: Session, s3_uri: str, s3_kms_key: str = None
 ):
     """Serializes data object and uploads it to S3.
 
@@ -219,7 +215,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.
@@ -227,7 +222,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,
@@ -274,14 +268,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.core.helper.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:
@@ -295,14 +288,14 @@ 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
+        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.
 
@@ -311,7 +304,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.
@@ -320,7 +312,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,
@@ -329,7 +320,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,
@@ -338,7 +328,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.core.helper.session.Session):
             The underlying Boto3 session which AWS service calls are delegated to.
@@ -346,7 +335,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(),
@@ -356,14 +345,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.core.helper.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:
@@ -377,7 +365,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
+        expected_hash_value=metadata.sha256_hash, buffer=bytes_to_deserialize
     )
 
     return CloudpickleSerializer.deserialize(f"{s3_uri}/payload.pkl", bytes_to_deserialize)
@@ -403,19 +391,19 @@ 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)
-    if not hmac.compare_digest(expected_hash_value, actual_hash_value):
+    actual_hash_value = _compute_hash(buffer=buffer)
+    if expected_hash_value != actual_hash_value:
         raise DeserializationError(
             "Integrity check for the serialized function or data failed. "
             "Please restrict access to your S3 bucket"

sagemaker/core/remote_function/core/stored_function.py

@@ -55,7 +55,6 @@ class StoredFunction:
         self,
         sagemaker_session: Session,
         s3_base_uri: str,
-        hmac_key: str,
         s3_kms_key: str = None,
         context: Context = Context(),
     ):
@@ -66,13 +65,11 @@ class StoredFunction:
                 AWS service calls are delegated to.
             s3_base_uri: the base uri to which serialized artifacts will be uploaded.
             s3_kms_key: KMS key used to encrypt artifacts uploaded to S3.
-            hmac_key: Key used to encrypt serialized and deserialized function and arguments.
             context: Build or run context of a pipeline step.
         """
         self.sagemaker_session = sagemaker_session
         self.s3_base_uri = s3_base_uri
         self.s3_kms_key = s3_kms_key
-        self.hmac_key = hmac_key
         self.context = context
 
         # For pipeline steps, function code is at: base/step_name/build_timestamp/
@@ -114,7 +111,7 @@ class StoredFunction:
             sagemaker_session=self.sagemaker_session,
             s3_uri=s3_path_join(self.func_upload_path, FUNCTION_FOLDER),
             s3_kms_key=self.s3_kms_key,
-            hmac_key=self.hmac_key,
+            
         )
 
         logger.info(
@@ -126,7 +123,7 @@ class StoredFunction:
             obj=(args, kwargs),
             sagemaker_session=self.sagemaker_session,
             s3_uri=s3_path_join(self.func_upload_path, ARGUMENTS_FOLDER),
-            hmac_key=self.hmac_key,
+            
             s3_kms_key=self.s3_kms_key,
         )
 
@@ -144,7 +141,7 @@ class StoredFunction:
         )
         serialization._upload_payload_and_metadata_to_s3(
             bytes_to_upload=serialized_data.func,
-            hmac_key=self.hmac_key,
+            
             s3_uri=s3_path_join(self.func_upload_path, FUNCTION_FOLDER),
             sagemaker_session=self.sagemaker_session,
             s3_kms_key=self.s3_kms_key,
@@ -156,7 +153,7 @@ class StoredFunction:
         )
         serialization._upload_payload_and_metadata_to_s3(
             bytes_to_upload=serialized_data.args,
-            hmac_key=self.hmac_key,
+            
             s3_uri=s3_path_join(self.func_upload_path, ARGUMENTS_FOLDER),
             sagemaker_session=self.sagemaker_session,
             s3_kms_key=self.s3_kms_key,
@@ -172,7 +169,7 @@ class StoredFunction:
         func = serialization.deserialize_func_from_s3(
             sagemaker_session=self.sagemaker_session,
             s3_uri=s3_path_join(self.func_upload_path, FUNCTION_FOLDER),
-            hmac_key=self.hmac_key,
+            
         )
 
         logger.info(
@@ -182,7 +179,7 @@ class StoredFunction:
         args, kwargs = serialization.deserialize_obj_from_s3(
             sagemaker_session=self.sagemaker_session,
             s3_uri=s3_path_join(self.func_upload_path, ARGUMENTS_FOLDER),
-            hmac_key=self.hmac_key,
+            
         )
 
         logger.info("Resolving pipeline variables")
@@ -190,7 +187,7 @@ class StoredFunction:
             self.context,
             args,
             kwargs,
-            hmac_key=self.hmac_key,
+            
             s3_base_uri=self.s3_base_uri,
             sagemaker_session=self.sagemaker_session,
         )
@@ -206,7 +203,7 @@ class StoredFunction:
             obj=result,
             sagemaker_session=self.sagemaker_session,
             s3_uri=s3_path_join(self.results_upload_path, RESULTS_FOLDER),
-            hmac_key=self.hmac_key,
+            
             s3_kms_key=self.s3_kms_key,
         )
 

sagemaker/core/remote_function/errors.py

@@ -70,7 +70,7 @@ def _write_failure_reason_file(failure_msg):
             f.write(failure_msg)
 
 
-def handle_error(error, sagemaker_session, s3_base_uri, s3_kms_key, hmac_key) -> int:
+def handle_error(error, sagemaker_session, s3_base_uri, s3_kms_key) -> int:
     """Handle all exceptions raised during remote function execution.
 
     Args:
@@ -79,7 +79,6 @@ def handle_error(error, sagemaker_session, s3_base_uri, s3_kms_key, hmac_key) ->
              AWS service calls are delegated to.
         s3_base_uri (str): S3 root uri to which resulting serialized exception will be uploaded.
         s3_kms_key (str): KMS key used to encrypt artifacts uploaded to S3.
-        hmac_key (str): Key used to calculate hmac hash of the serialized exception.
     Returns :
         exit_code (int): Exit code to terminate current job.
     """
@@ -97,7 +96,6 @@ def handle_error(error, sagemaker_session, s3_base_uri, s3_kms_key, hmac_key) ->
         exc=error,
         sagemaker_session=sagemaker_session,
         s3_uri=s3_path_join(s3_base_uri, "exception"),
-        hmac_key=hmac_key,
         s3_kms_key=s3_kms_key,
     )
 

sagemaker/core/remote_function/invoke_function.py

@@ -98,7 +98,7 @@ def _load_pipeline_context(args) -> Context:
 
 
 def _execute_remote_function(
-    sagemaker_session, s3_base_uri, s3_kms_key, run_in_context, hmac_key, context
+    sagemaker_session, s3_base_uri, s3_kms_key, run_in_context, context
 ):
     """Execute stored remote function"""
     from sagemaker.core.remote_function.core.stored_function import StoredFunction
@@ -107,7 +107,6 @@ def _execute_remote_function(
         sagemaker_session=sagemaker_session,
         s3_base_uri=s3_base_uri,
         s3_kms_key=s3_kms_key,
-        hmac_key=hmac_key,
         context=context,
     )
 
@@ -138,15 +137,12 @@ def main(sys_args=None):
         run_in_context = args.run_in_context
         pipeline_context = _load_pipeline_context(args)
 
-        hmac_key = os.getenv("REMOTE_FUNCTION_SECRET_KEY")
-
         sagemaker_session = _get_sagemaker_session(region)
         _execute_remote_function(
             sagemaker_session=sagemaker_session,
             s3_base_uri=s3_base_uri,
             s3_kms_key=s3_kms_key,
             run_in_context=run_in_context,
-            hmac_key=hmac_key,
             context=pipeline_context,
         )
 
@@ -162,7 +158,6 @@ def main(sys_args=None):
             sagemaker_session=sagemaker_session,
             s3_base_uri=s3_uri,
             s3_kms_key=s3_kms_key,
-            hmac_key=hmac_key,
         )
     finally:
         sys.exit(exit_code)

sagemaker/core/remote_function/job.py

@@ -17,7 +17,6 @@ import dataclasses
 import json
 import os
 import re
-import secrets
 import shutil
 import sys
 import time
@@ -621,11 +620,6 @@ class _JobSettings:
             {"AWS_DEFAULT_REGION": self.sagemaker_session.boto_region_name}
         )
 
-        # The following will be overridden by the _Job.compile method.
-        # However, it needs to be kept here for feature store SDK.
-        # TODO: update the feature store SDK to set the HMAC key there.
-        self.environment_variables.update({"REMOTE_FUNCTION_SECRET_KEY": secrets.token_hex(32)})
-
         if spark_config and image_uri:
             raise ValueError("spark_config and image_uri cannot be specified at the same time!")
 
@@ -839,19 +833,17 @@ class _JobSettings:
 class _Job:
     """Helper class that interacts with the SageMaker training service."""
 
-    def __init__(self, job_name: str, s3_uri: str, sagemaker_session: Session, hmac_key: str):
+    def __init__(self, job_name: str, s3_uri: str, sagemaker_session: Session):
         """Initialize a _Job object.
 
         Args:
             job_name (str): The training job name.
             s3_uri (str): The training job output S3 uri.
             sagemaker_session (Session): SageMaker boto session.
-            hmac_key (str): Remote function secret key.
         """
         self.job_name = job_name
         self.s3_uri = s3_uri
         self.sagemaker_session = sagemaker_session
-        self.hmac_key = hmac_key
         self._last_describe_response = None
 
     @staticmethod
@@ -867,9 +859,8 @@ class _Job:
         """
         job_name = describe_training_job_response["TrainingJobName"]
         s3_uri = describe_training_job_response["OutputDataConfig"]["S3OutputPath"]
-        hmac_key = describe_training_job_response["Environment"]["REMOTE_FUNCTION_SECRET_KEY"]
 
-        job = _Job(job_name, s3_uri, sagemaker_session, hmac_key)
+        job = _Job(job_name, s3_uri, sagemaker_session)
         job._last_describe_response = describe_training_job_response
         return job
 
@@ -907,7 +898,6 @@ class _Job:
             job_name,
             s3_base_uri,
             job_settings.sagemaker_session,
-            training_job_request["Environment"]["REMOTE_FUNCTION_SECRET_KEY"],
         )
 
     @staticmethod
@@ -935,18 +925,11 @@ class _Job:
 
         jobs_container_entrypoint = JOBS_CONTAINER_ENTRYPOINT[:]
 
-        # generate hmac key for integrity check
-        if step_compilation_context is None:
-            hmac_key = secrets.token_hex(32)
-        else:
-            hmac_key = step_compilation_context.function_step_secret_token
-
         # serialize function and arguments
         if step_compilation_context is None:
             stored_function = StoredFunction(
                 sagemaker_session=job_settings.sagemaker_session,
                 s3_base_uri=s3_base_uri,
-                hmac_key=hmac_key,
                 s3_kms_key=job_settings.s3_kms_key,
             )
             stored_function.save(func, *func_args, **func_kwargs)
@@ -954,7 +937,6 @@ class _Job:
             stored_function = StoredFunction(
                 sagemaker_session=job_settings.sagemaker_session,
                 s3_base_uri=s3_base_uri,
-                hmac_key=hmac_key,
                 s3_kms_key=job_settings.s3_kms_key,
                 context=Context(
                     step_name=step_compilation_context.step_name,
@@ -1114,7 +1096,6 @@ class _Job:
         request_dict["EnableManagedSpotTraining"] = job_settings.use_spot_instances
 
         request_dict["Environment"] = job_settings.environment_variables
-        request_dict["Environment"].update({"REMOTE_FUNCTION_SECRET_KEY": hmac_key})
 
         extended_request = _extend_spark_config_to_request(request_dict, job_settings, s3_base_uri)
         extended_request = _extend_mpirun_to_request(extended_request, job_settings)

sagemaker/core/telemetry/constants.py

@@ -22,14 +22,12 @@ DEFAULT_AWS_REGION = "us-west-2"
 class Feature(Enum):
     """Enumeration of feature names used in telemetry."""
 
-    SDK_DEFAULTS = 1
-    LOCAL_MODE = 2
-    REMOTE_FUNCTION = 3
-    MODEL_TRAINER = 4
-    ESTIMATOR = 5
-    HYPERPOD = 6  # Added to support telemetry in sagemaker-hyperpod-cli
-    HYPERPOD_CLI = 7  # Added to support telemetry in sagemaker-hyperpod-cli
-    MODEL_CUSTOMIZATION = 8
+    SDK_DEFAULTS = 11
+    LOCAL_MODE = 12
+    REMOTE_FUNCTION = 13
+    MODEL_TRAINER = 14
+    MODEL_CUSTOMIZATION = 15
+    MLOPS = 16
 
     def __str__(self):  # pylint: disable=E0307
         """Return the feature name."""

sagemaker/core/telemetry/telemetry_logging.py

@@ -50,11 +50,12 @@ TELEMETRY_OPT_OUT_MESSAGING = (
 )
 
 FEATURE_TO_CODE = {
-    str(Feature.SDK_DEFAULTS): 1,
-    str(Feature.LOCAL_MODE): 2,
-    str(Feature.REMOTE_FUNCTION): 3,
-    str(Feature.MODEL_TRAINER): 4,
-    str(Feature.MODEL_CUSTOMIZATION): 8,
+    str(Feature.SDK_DEFAULTS): 11,
+    str(Feature.LOCAL_MODE): 12,
+    str(Feature.REMOTE_FUNCTION): 13,
+    str(Feature.MODEL_TRAINER): 14,
+    str(Feature.MODEL_CUSTOMIZATION): 15,
+    str(Feature.MLOPS): 16,
 }
 
 STATUS_TO_CODE = {

sagemaker/core/training/configs.py

@@ -21,7 +21,7 @@ For more documentation on ``sagemaker.core.shapes``, see:
 
 from __future__ import absolute_import
 
-from typing import Optional, Union
+from typing import Optional, Union, List
 from pydantic import BaseModel, model_validator, ConfigDict
 
 import sagemaker.core.shapes as shapes
@@ -106,18 +106,29 @@ class SourceCode(BaseConfig):
         command (Optional[StrPipeVar]):
             The command(s) to execute in the training job container. Example: "python my_script.py".
             If not specified, entry_script must be provided.
+        ignore_patterns: (Optional[List[str]]) :
+            The ignore patterns to ignore specific files/folders when uploading to S3. If not specified,
+            default to: ['.env', '.git', '__pycache__', '.DS_Store', '.cache', '.ipynb_checkpoints'].
     """
 
     source_dir: Optional[StrPipeVar] = None
     requirements: Optional[StrPipeVar] = None
     entry_script: Optional[StrPipeVar] = None
     command: Optional[StrPipeVar] = None
-
+    ignore_patterns: Optional[List[str]] = [
+        ".env",
+        ".git",
+        "__pycache__",
+        ".DS_Store",
+        ".cache",
+        ".ipynb_checkpoints",
+    ]
 
 class OutputDataConfig(shapes.OutputDataConfig):
     """OutputDataConfig.
 
-    Provides the configuration for the output data location of the training job.
+    Provides the configuration for the output data location of the training job 
+    (will not be carried over to any model repository or deployment).
 
     Parameters:
         s3_output_path (Optional[StrPipeVar]):
@@ -274,7 +285,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[StrPipeVar]):

sagemaker/core/workflow/utilities.py

@@ -21,7 +21,14 @@ from typing import List, Sequence, Union, Set, TYPE_CHECKING, Optional
 import hashlib
 from urllib.parse import unquote, urlparse
 from contextlib import contextmanager
-from _hashlib import HASH as Hash
+
+try:
+    # _hashlib is an internal python module, and is not present in
+    # statically linked interpreters.
+    from _hashlib import HASH as Hash
+except ImportError:
+    import typing
+    Hash = typing.Any
 
 from sagemaker.core.common_utils import base_from_name
 from sagemaker.core.workflow.parameters import Parameter
@@ -165,9 +172,9 @@ def get_code_hash(step: Entity) -> str:
         source_code = model_trainer.source_code
         if source_code:
             source_dir = source_code.source_dir
-            dependencies = source_code.dependencies
+            requirements = source_code.requirements
             entry_point = source_code.entry_script
-            return get_training_code_hash(entry_point, source_dir, dependencies)
+            return get_training_code_hash(entry_point, source_dir, requirements)
     return None
 
 

{sagemaker_core-2.1.1.dist-info → sagemaker_core-2.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 2.1.1
+Version: 2.3.1
 Summary: An python package for sagemaker core functionalities
 Author-email: AWS <sagemaker-interests@amazon.com>
 Project-URL: Homepage, https://github.com/aws/sagemaker-python-sdk