sagemaker 2.239.1__tar.gz → 2.239.2__tar.gz

{sagemaker-2.239.1 → sagemaker-2.239.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: sagemaker
-Version: 2.239.1
+Version: 2.239.2
 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.239.2/VERSION

@@ -0,0 +1 @@
+2.239.2

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/amazon/common.py

@@ -13,7 +13,6 @@
 """Placeholder docstring"""
 from __future__ import absolute_import
 
-import io
 import logging
 import struct
 import sys
@@ -21,76 +20,9 @@ import sys
 import numpy as np
 
 from sagemaker.amazon.record_pb2 import Record
-from sagemaker.deprecations import deprecated_class
-from sagemaker.deserializers import SimpleBaseDeserializer
-from sagemaker.serializers import SimpleBaseSerializer
 from sagemaker.utils import DeferredError
 
 
-class RecordSerializer(SimpleBaseSerializer):
-    """Serialize a NumPy array for an inference request."""
-
-    def __init__(self, content_type="application/x-recordio-protobuf"):
-        """Initialize a ``RecordSerializer`` instance.
-
-        Args:
-            content_type (str): The MIME type to signal to the inference endpoint when sending
-                request data (default: "application/x-recordio-protobuf").
-        """
-        super(RecordSerializer, self).__init__(content_type=content_type)
-
-    def serialize(self, data):
-        """Serialize a NumPy array into a buffer containing RecordIO records.
-
-        Args:
-            data (numpy.ndarray): The data to serialize.
-
-        Returns:
-            io.BytesIO: A buffer containing the data serialized as records.
-        """
-        if len(data.shape) == 1:
-            data = data.reshape(1, data.shape[0])
-
-        if len(data.shape) != 2:
-            raise ValueError(
-                "Expected a 1D or 2D array, but got a %dD array instead." % len(data.shape)
-            )
-
-        buffer = io.BytesIO()
-        write_numpy_to_dense_tensor(buffer, data)
-        buffer.seek(0)
-
-        return buffer
-
-
-class RecordDeserializer(SimpleBaseDeserializer):
-    """Deserialize RecordIO Protobuf data from an inference endpoint."""
-
-    def __init__(self, accept="application/x-recordio-protobuf"):
-        """Initialize a ``RecordDeserializer`` instance.
-
-        Args:
-            accept (union[str, tuple[str]]): The MIME type (or tuple of allowable MIME types) that
-                is expected from the inference endpoint (default:
-                "application/x-recordio-protobuf").
-        """
-        super(RecordDeserializer, self).__init__(accept=accept)
-
-    def deserialize(self, data, content_type):
-        """Deserialize RecordIO Protobuf data from an inference endpoint.
-
-        Args:
-            data (object): The protobuf message to deserialize.
-            content_type (str): The MIME type of the data.
-        Returns:
-            list: A list of records.
-        """
-        try:
-            return read_records(data)
-        finally:
-            data.close()
-
-
 def _write_feature_tensor(resolved_type, record, vector):
     """Placeholder Docstring"""
     if resolved_type == "Int32":
@@ -288,7 +220,3 @@ def _resolve_type(dtype):
     if dtype == np.dtype("float32"):
         return "Float32"
     raise ValueError("Unsupported dtype {} on array".format(dtype))
-
-
-numpy_to_record_serializer = deprecated_class(RecordSerializer, "numpy_to_record_serializer")
-record_deserializer = deprecated_class(RecordDeserializer, "record_deserializer")

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/amazon/factorization_machines.py

@@ -17,11 +17,12 @@ from typing import Union, Optional
 
 from sagemaker import image_uris
 from sagemaker.amazon.amazon_estimator import AmazonAlgorithmEstimatorBase
-from sagemaker.amazon.common import RecordSerializer, RecordDeserializer
 from sagemaker.amazon.hyperparameter import Hyperparameter as hp  # noqa
 from sagemaker.amazon.validation import gt, isin, ge
+from sagemaker.deserializers import RecordDeserializer
 from sagemaker.predictor import Predictor
 from sagemaker.model import Model
+from sagemaker.serializers import RecordSerializer
 from sagemaker.session import Session
 from sagemaker.utils import pop_out_unused_kwarg
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/amazon/kmeans.py

@@ -17,11 +17,12 @@ from typing import Union, Optional, List
 
 from sagemaker import image_uris
 from sagemaker.amazon.amazon_estimator import AmazonAlgorithmEstimatorBase
-from sagemaker.amazon.common import RecordSerializer, RecordDeserializer
 from sagemaker.amazon.hyperparameter import Hyperparameter as hp  # noqa
 from sagemaker.amazon.validation import gt, isin, ge, le
+from sagemaker.deserializers import RecordDeserializer
 from sagemaker.predictor import Predictor
 from sagemaker.model import Model
+from sagemaker.serializers import RecordSerializer
 from sagemaker.session import Session
 from sagemaker.utils import pop_out_unused_kwarg
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/amazon/knn.py

@@ -17,11 +17,12 @@ from typing import Union, Optional
 
 from sagemaker import image_uris
 from sagemaker.amazon.amazon_estimator import AmazonAlgorithmEstimatorBase
-from sagemaker.amazon.common import RecordSerializer, RecordDeserializer
 from sagemaker.amazon.hyperparameter import Hyperparameter as hp  # noqa
 from sagemaker.amazon.validation import ge, isin
+from sagemaker.deserializers import RecordDeserializer
 from sagemaker.predictor import Predictor
 from sagemaker.model import Model
+from sagemaker.serializers import RecordSerializer
 from sagemaker.session import Session
 from sagemaker.utils import pop_out_unused_kwarg
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/amazon/lda.py

@@ -18,11 +18,12 @@ from typing import Union, Optional
 
 from sagemaker import image_uris
 from sagemaker.amazon.amazon_estimator import AmazonAlgorithmEstimatorBase
-from sagemaker.amazon.common import RecordSerializer, RecordDeserializer
+from sagemaker.deserializers import RecordDeserializer
 from sagemaker.amazon.hyperparameter import Hyperparameter as hp  # noqa
 from sagemaker.amazon.validation import gt
 from sagemaker.predictor import Predictor
 from sagemaker.model import Model
+from sagemaker.serializers import RecordSerializer
 from sagemaker.session import Session
 from sagemaker.utils import pop_out_unused_kwarg
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/amazon/linear_learner.py

@@ -18,11 +18,12 @@ from typing import Union, Optional
 
 from sagemaker import image_uris
 from sagemaker.amazon.amazon_estimator import AmazonAlgorithmEstimatorBase
-from sagemaker.amazon.common import RecordSerializer, RecordDeserializer
+from sagemaker.deserializers import RecordDeserializer
 from sagemaker.amazon.hyperparameter import Hyperparameter as hp  # noqa
 from sagemaker.amazon.validation import isin, gt, lt, ge, le
 from sagemaker.predictor import Predictor
 from sagemaker.model import Model
+from sagemaker.serializers import RecordSerializer
 from sagemaker.session import Session
 from sagemaker.utils import pop_out_unused_kwarg
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/amazon/ntm.py

@@ -17,11 +17,12 @@ from typing import Optional, Union, List
 
 from sagemaker import image_uris
 from sagemaker.amazon.amazon_estimator import AmazonAlgorithmEstimatorBase
-from sagemaker.amazon.common import RecordSerializer, RecordDeserializer
 from sagemaker.amazon.hyperparameter import Hyperparameter as hp  # noqa
 from sagemaker.amazon.validation import ge, le, isin
+from sagemaker.deserializers import RecordDeserializer
 from sagemaker.predictor import Predictor
 from sagemaker.model import Model
+from sagemaker.serializers import RecordSerializer
 from sagemaker.session import Session
 from sagemaker.utils import pop_out_unused_kwarg
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/amazon/pca.py

@@ -17,11 +17,12 @@ from typing import Union, Optional
 
 from sagemaker import image_uris
 from sagemaker.amazon.amazon_estimator import AmazonAlgorithmEstimatorBase
-from sagemaker.amazon.common import RecordSerializer, RecordDeserializer
 from sagemaker.amazon.hyperparameter import Hyperparameter as hp  # noqa
 from sagemaker.amazon.validation import gt, isin
+from sagemaker.deserializers import RecordDeserializer
 from sagemaker.predictor import Predictor
 from sagemaker.model import Model
+from sagemaker.serializers import RecordSerializer
 from sagemaker.session import Session
 from sagemaker.utils import pop_out_unused_kwarg
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/amazon/randomcutforest.py

@@ -17,11 +17,12 @@ from typing import Optional, Union, List
 
 from sagemaker import image_uris
 from sagemaker.amazon.amazon_estimator import AmazonAlgorithmEstimatorBase
-from sagemaker.amazon.common import RecordSerializer, RecordDeserializer
 from sagemaker.amazon.hyperparameter import Hyperparameter as hp  # noqa
 from sagemaker.amazon.validation import ge, le
+from sagemaker.deserializers import RecordDeserializer
 from sagemaker.predictor import Predictor
 from sagemaker.model import Model
+from sagemaker.serializers import RecordSerializer
 from sagemaker.session import Session
 from sagemaker.utils import pop_out_unused_kwarg
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/base_deserializers.py

@@ -23,6 +23,7 @@ import json
 import numpy as np
 from six import with_metaclass
 
+from sagemaker.amazon.common import read_records
 from sagemaker.utils import DeferredError
 
 try:
@@ -388,3 +389,31 @@ class TorchTensorDeserializer(SimpleBaseDeserializer):
                 "Unable to deserialize your data to torch.Tensor.\
                     Please provide custom deserializer in InferenceSpec."
             )
+
+
+class RecordDeserializer(SimpleBaseDeserializer):
+    """Deserialize RecordIO Protobuf data from an inference endpoint."""
+
+    def __init__(self, accept="application/x-recordio-protobuf"):
+        """Initialize a ``RecordDeserializer`` instance.
+
+        Args:
+            accept (union[str, tuple[str]]): The MIME type (or tuple of allowable MIME types) that
+                is expected from the inference endpoint (default:
+                "application/x-recordio-protobuf").
+        """
+        super(RecordDeserializer, self).__init__(accept=accept)
+
+    def deserialize(self, data, content_type):
+        """Deserialize RecordIO Protobuf data from an inference endpoint.
+
+        Args:
+            data (object): The protobuf message to deserialize.
+            content_type (str): The MIME type of the data.
+        Returns:
+            list: A list of records.
+        """
+        try:
+            return read_records(data)
+        finally:
+            data.close()

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/base_serializers.py

@@ -22,6 +22,7 @@ import numpy as np
 from pandas import DataFrame
 from six import with_metaclass
 
+from sagemaker.amazon.common import write_numpy_to_dense_tensor
 from sagemaker.utils import DeferredError
 
 try:
@@ -466,3 +467,39 @@ class TorchTensorSerializer(SimpleBaseSerializer):
                 )
 
         raise ValueError("Object of type %s is not a torch.Tensor" % type(data))
+
+
+class RecordSerializer(SimpleBaseSerializer):
+    """Serialize a NumPy array for an inference request."""
+
+    def __init__(self, content_type="application/x-recordio-protobuf"):
+        """Initialize a ``RecordSerializer`` instance.
+
+        Args:
+            content_type (str): The MIME type to signal to the inference endpoint when sending
+                request data (default: "application/x-recordio-protobuf").
+        """
+        super(RecordSerializer, self).__init__(content_type=content_type)
+
+    def serialize(self, data):
+        """Serialize a NumPy array into a buffer containing RecordIO records.
+
+        Args:
+            data (numpy.ndarray): The data to serialize.
+
+        Returns:
+            io.BytesIO: A buffer containing the data serialized as records.
+        """
+        if len(data.shape) == 1:
+            data = data.reshape(1, data.shape[0])
+
+        if len(data.shape) != 2:
+            raise ValueError(
+                "Expected a 1D or 2D array, but got a %dD array instead." % len(data.shape)
+            )
+
+        buffer = io.BytesIO()
+        write_numpy_to_dense_tensor(buffer, data)
+        buffer.seek(0)
+
+        return buffer

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/cli/compatibility/v2/modifiers/serde.py

@@ -51,8 +51,8 @@ NEW_CLASS_NAME_TO_NAMESPACES = {
     "StreamDeserializer": ("sagemaker.deserializers",),
     "NumpyDeserializer": ("sagemaker.deserializers",),
     "JSONDeserializer": ("sagemaker.deserializers",),
-    "RecordSerializer ": ("sagemaker.amazon.common",),
-    "RecordDeserializer": ("sagemaker.amazon.common",),
+    "RecordSerializer ": ("sagemaker.serializers",),
+    "RecordDeserializer": ("sagemaker.deserializers",),
 }
 
 OLD_CLASS_NAME_TO_NEW_CLASS_NAME = {
@@ -101,8 +101,8 @@ class SerdeConstructorRenamer(Modifier):
         - ``sagemaker.predictor.StreamDeserializer``
         - ``sagemaker.predictor._NumpyDeserializer``
         - ``sagemaker.predictor._JsonDeserializer``
-        - ``sagemaker.amazon.common.numpy_to_record_serializer``
-        - ``sagemaker.amazon.common.record_deserializer``
+        - ``sagemaker.serializers.numpy_to_record_serializer``
+        - ``sagemaker.deserializers.record_deserializer``
 
         Args:
             node (ast.Call): a node that represents a function call. For more,
@@ -128,8 +128,8 @@ class SerdeConstructorRenamer(Modifier):
         - ``sagemaker.deserializers.StreamDeserializer``
         - ``sagemaker.deserializers.NumpyDeserializer``
         - ``sagemaker.deserializers._JsonDeserializer``
-        - ``sagemaker.amazon.common.RecordSerializer``
-        - ``sagemaker.amazon.common.RecordDeserializer``
+        - ``sagemaker.serializers.RecordSerializer``
+        - ``sagemaker.deserializers.RecordDeserializer``
 
         Args:
             node (ast.Call): a node that represents a SerDe constructor.
@@ -303,8 +303,8 @@ class SerdeImportFromAmazonCommonRenamer(Modifier):
         """Checks if the import statement imports a SerDe from the ``sagemaker.amazon.common``.
 
         This checks for:
-        - ``sagemaker.amazon.common.numpy_to_record_serializer``
-        - ``sagemaker.amazon.common.record_deserializer``
+        - ``sagemaker.serializers.numpy_to_record_serializer``
+        - ``sagemaker.deserializers.record_deserializer``
 
         Args:
             node (ast.ImportFrom): a node that represents a ``from ... import ... `` statement.
@@ -322,8 +322,8 @@ class SerdeImportFromAmazonCommonRenamer(Modifier):
         """Upgrades the ``numpy_to_record_serializer`` and ``record_deserializer`` imports.
 
         This upgrades the classes to (if applicable):
-        - ``sagemaker.amazon.common.RecordSerializer``
-        - ``sagemaker.amazon.common.RecordDeserializer``
+        - ``sagemaker.serializers.RecordSerializer``
+        - ``sagemaker.deserializers.RecordDeserializer``
 
         Args:
             node (ast.ImportFrom): a node that represents a ``from ... import ... `` statement.

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/deserializers.py

@@ -31,8 +31,10 @@ from sagemaker.base_deserializers import (  # noqa: F401 # pylint: disable=W0611
     StreamDeserializer,
     StringDeserializer,
     TorchTensorDeserializer,
+    RecordDeserializer,
 )
 
+from sagemaker.deprecations import deprecated_class
 from sagemaker.jumpstart import artifacts, utils as jumpstart_utils
 from sagemaker.jumpstart.constants import DEFAULT_JUMPSTART_SAGEMAKER_SESSION
 from sagemaker.jumpstart.enums import JumpStartModelType
@@ -150,3 +152,6 @@ def retrieve_default(
         model_type=model_type,
         config_name=config_name,
     )
+
+
+record_deserializer = deprecated_class(RecordDeserializer, "record_deserializer")

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/model.py

@@ -1492,6 +1492,9 @@ api/latest/reference/services/sagemaker.html#SageMaker.Client.add_tags>`_
                     }
             model_reference_arn (Optional [str]): Hub Content Arn of a Model Reference type
                 content (default: None).
+            inference_ami_version (Optional [str]): Specifies an option from a collection of preconfigured
+             Amazon Machine Image (AMI) images. For a full list of options, see:
+             https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_ProductionVariant.html
         Raises:
              ValueError: If arguments combination check failed in these circumstances:
                 - If no role is specified or
@@ -1743,6 +1746,7 @@ api/latest/reference/services/sagemaker.html#SageMaker.Client.add_tags>`_
                 model_data_download_timeout=model_data_download_timeout,
                 container_startup_health_check_timeout=container_startup_health_check_timeout,
                 routing_config=routing_config,
+                inference_ami_version=inference_ami_version,
             )
             if endpoint_name:
                 self.endpoint_name = endpoint_name

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/serializers.py

@@ -30,8 +30,10 @@ from sagemaker.base_serializers import (  # noqa: F401 # pylint: disable=W0611
     SparseMatrixSerializer,
     TorchTensorSerializer,
     StringSerializer,
+    RecordSerializer,
 )
 
+from sagemaker.deprecations import deprecated_class
 from sagemaker.jumpstart import artifacts, utils as jumpstart_utils
 from sagemaker.jumpstart.constants import DEFAULT_JUMPSTART_SAGEMAKER_SESSION
 from sagemaker.jumpstart.enums import JumpStartModelType
@@ -152,3 +154,6 @@ def retrieve_default(
         model_type=model_type,
         config_name=config_name,
     )
+
+
+numpy_to_record_serializer = deprecated_class(RecordSerializer, "numpy_to_record_serializer")

{sagemaker-2.239.1 → sagemaker-2.239.2}/src/sagemaker/tuner.py

@@ -18,21 +18,20 @@ import importlib
 import inspect
 import json
 import logging
-
 from enum import Enum
-from typing import Union, Dict, Optional, List, Set
+from typing import Dict, List, Optional, Set, Union
 
 import sagemaker
 from sagemaker.amazon.amazon_estimator import (
-    RecordSet,
     AmazonAlgorithmEstimatorBase,
     FileSystemRecordSet,
+    RecordSet,
 )
 from sagemaker.amazon.hyperparameter import Hyperparameter as hp  # noqa
 from sagemaker.analytics import HyperparameterTuningJobAnalytics
 from sagemaker.deprecations import removed_function
-from sagemaker.estimator import Framework, EstimatorBase
-from sagemaker.inputs import TrainingInput, FileSystemInput
+from sagemaker.estimator import EstimatorBase, Framework
+from sagemaker.inputs import FileSystemInput, TrainingInput
 from sagemaker.job import _Job
 from sagemaker.jumpstart.utils import (
     add_jumpstart_uri_tags,
@@ -44,18 +43,17 @@ from sagemaker.parameter import (
     IntegerParameter,
     ParameterRange,
 )
-from sagemaker.workflow.entities import PipelineVariable
-from sagemaker.workflow.pipeline_context import runnable_by_pipeline
-
 from sagemaker.session import Session
 from sagemaker.utils import (
+    Tags,
     base_from_name,
     base_name_from_image,
+    format_tags,
     name_from_base,
     to_string,
-    format_tags,
-    Tags,
 )
+from sagemaker.workflow.entities import PipelineVariable
+from sagemaker.workflow.pipeline_context import runnable_by_pipeline
 
 AMAZON_ESTIMATOR_MODULE = "sagemaker"
 AMAZON_ESTIMATOR_CLS_NAMES = {
@@ -133,15 +131,12 @@ class WarmStartConfig(object):
 
         if warm_start_type not in list(WarmStartTypes):
             raise ValueError(
-                "Invalid type: {}, valid warm start types are: {}".format(
-                    warm_start_type, list(WarmStartTypes)
-                )
+                f"Invalid type: {warm_start_type}, "
+                f"valid warm start types are: {list(WarmStartTypes)}"
             )
 
         if not parents:
-            raise ValueError(
-                "Invalid parents: {}, parents should not be None/empty".format(parents)
-            )
+            raise ValueError(f"Invalid parents: {parents}, parents should not be None/empty")
 
         self.type = warm_start_type
         self.parents = set(parents)
@@ -1455,9 +1450,7 @@ class HyperparameterTuner(object):
             return tuning_job_describe_result["BestTrainingJob"]
         except KeyError:
             raise Exception(
-                "Best training job not available for tuning job: {}".format(
-                    self.latest_tuning_job.name
-                )
+                f"Best training job not available for tuning job: {self.latest_tuning_job.name}"
             )
 
     def _ensure_last_tuning_job(self):
@@ -1920,8 +1913,11 @@ class HyperparameterTuner(object):
                 :meth:`~sagemaker.tuner.HyperparameterTuner.fit` method launches.
                 If not specified, a default job name is generated,
                 based on the training image name and current timestamp.
-            strategy (str): Strategy to be used for hyperparameter estimations
-                (default: 'Bayesian').
+            strategy (str or PipelineVariable): Strategy to be used for hyperparameter estimations.
+                More information about different strategies:
+                https://docs.aws.amazon.com/sagemaker/latest/dg/automatic-model-tuning-how-it-works.html.
+                Available options are: 'Bayesian', 'Random', 'Hyperband',
+                'Grid' (default: 'Bayesian')
             strategy_config (dict): The configuration for a training job launched by a
                 hyperparameter tuning job.
             completion_criteria_config (dict): The configuration for tuning job completion criteria.
@@ -2080,21 +2076,19 @@ class HyperparameterTuner(object):
             return
 
         if not isinstance(value, dict):
-            raise ValueError(
-                "Argument '{}' must be a dictionary using {} as keys".format(name, allowed_keys)
-            )
+            raise ValueError(f"Argument '{name}' must be a dictionary using {allowed_keys} as keys")
 
         value_keys = sorted(value.keys())
 
         if require_same_keys:
             if value_keys != allowed_keys:
                 raise ValueError(
-                    "The keys of argument '{}' must be the same as {}".format(name, allowed_keys)
+                    f"The keys of argument '{name}' must be the same as {allowed_keys}"
                 )
         else:
             if not set(value_keys).issubset(set(allowed_keys)):
                 raise ValueError(
-                    "The keys of argument '{}' must be a subset of {}".format(name, allowed_keys)
+                    f"The keys of argument '{name}' must be a subset of {allowed_keys}"
                 )
 
     def _add_estimator(

sagemaker-2.239.1/VERSION

@@ -1 +0,0 @@
-2.239.1