sagemaker-core 1.0.6__tar.gz → 1.0.7__tar.gz

{sagemaker_core-1.0.6/src/sagemaker_core.egg-info → sagemaker_core-1.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.6
+Version: 1.0.7
 Summary: An python package for sagemaker core functionalities
 Author-email: AWS <sagemaker-interests@amazon.com>
 Project-URL: Repository, https://github.com/aws/sagemaker-core.git

sagemaker_core-1.0.7/VERSION

@@ -0,0 +1 @@
+1.0.7

{sagemaker_core-1.0.6 → sagemaker_core-1.0.7}/src/sagemaker_core/main/code_injection/shape_dag.py

@@ -797,6 +797,16 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "BatchGetMetricsRequest": {
+        "members": [{"name": "MetricQueries", "shape": "MetricQueryList", "type": "list"}],
+        "type": "structure",
+    },
+    "BatchGetMetricsResponse": {
+        "members": [
+            {"name": "MetricQueryResults", "shape": "MetricQueryResultList", "type": "list"}
+        ],
+        "type": "structure",
+    },
     "BatchGetRecordError": {
         "members": [
             {"name": "FeatureGroupName", "shape": "ValueAsString", "type": "string"},
@@ -9659,6 +9669,33 @@ SHAPE_DAG = {
         "member_type": "structure",
         "type": "list",
     },
+    "MetricQuery": {
+        "members": [
+            {"name": "MetricName", "shape": "MetricName", "type": "string"},
+            {"name": "ResourceArn", "shape": "SageMakerResourceArn", "type": "string"},
+            {"name": "MetricStat", "shape": "MetricStatistic", "type": "string"},
+            {"name": "Period", "shape": "Period", "type": "string"},
+            {"name": "XAxisType", "shape": "XAxisType", "type": "string"},
+            {"name": "Start", "shape": "Long", "type": "long"},
+            {"name": "End", "shape": "Long", "type": "long"},
+        ],
+        "type": "structure",
+    },
+    "MetricQueryList": {"member_shape": "MetricQuery", "member_type": "structure", "type": "list"},
+    "MetricQueryResult": {
+        "members": [
+            {"name": "Status", "shape": "MetricQueryResultStatus", "type": "string"},
+            {"name": "Message", "shape": "Message", "type": "string"},
+            {"name": "XAxisValues", "shape": "XAxisValues", "type": "list"},
+            {"name": "MetricValues", "shape": "MetricValues", "type": "list"},
+        ],
+        "type": "structure",
+    },
+    "MetricQueryResultList": {
+        "member_shape": "MetricQueryResult",
+        "member_type": "structure",
+        "type": "list",
+    },
     "MetricSpecification": {
         "members": [
             {"name": "Predefined", "shape": "PredefinedMetricSpecification", "type": "structure"},
@@ -9666,6 +9703,7 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "MetricValues": {"member_shape": "Double", "member_type": "double", "type": "list"},
     "MetricsSource": {
         "members": [
             {"name": "ContentType", "shape": "ContentType", "type": "string"},
@@ -12459,6 +12497,7 @@ SHAPE_DAG = {
             {"name": "CompressionType", "shape": "ModelCompressionType", "type": "string"},
             {"name": "ModelAccessConfig", "shape": "ModelAccessConfig", "type": "structure"},
             {"name": "HubAccessConfig", "shape": "InferenceHubAccessConfig", "type": "structure"},
+            {"name": "ManifestS3Uri", "shape": "S3ModelUri", "type": "string"},
         ],
         "type": "structure",
     },
@@ -14796,4 +14835,5 @@ SHAPE_DAG = {
         "type": "structure",
     },
     "Workteams": {"member_shape": "Workteam", "member_type": "structure", "type": "list"},
+    "XAxisValues": {"member_shape": "Long", "member_type": "long", "type": "list"},
 }

{sagemaker_core-1.0.6 → sagemaker_core-1.0.7}/src/sagemaker_core/main/config_schema.py

@@ -469,6 +469,7 @@ SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA = {
                                                 "s3_data_source": {
                                                     "s3_uri": {"type": "string"},
                                                     "s3_data_type": {"type": "string"},
+                                                    "manifest_s3_uri": {"type": "string"},
                                                 }
                                             }
                                         },

{sagemaker_core-1.0.6 → sagemaker_core-1.0.7}/src/sagemaker_core/main/shapes.py

@@ -362,6 +362,50 @@ class ResourceNotFound(Base):
     message: Optional[str] = Unassigned()
 
 
+class MetricQuery(Base):
+    """
+    MetricQuery
+      Specifies a query to retrieve training metrics from SageMaker.
+
+    Attributes
+    ----------------------
+    metric_name: The name of the metric to retrieve.
+    resource_arn: The ARN of the SageMaker resource to retrieve metrics for.
+    metric_stat: The metrics stat type of metrics to retrieve.
+    period: The time period of metrics to retrieve.
+    x_axis_type: The x-axis type of metrics to retrieve.
+    start: The start time of metrics to retrieve.
+    end: The end time of metrics to retrieve.
+    """
+
+    metric_name: str
+    resource_arn: str
+    metric_stat: str
+    period: str
+    x_axis_type: str
+    start: Optional[int] = Unassigned()
+    end: Optional[int] = Unassigned()
+
+
+class MetricQueryResult(Base):
+    """
+    MetricQueryResult
+      The result of a query to retrieve training metrics from SageMaker.
+
+    Attributes
+    ----------------------
+    status: The status of the metric query.
+    message: A message describing the status of the metric query.
+    x_axis_values: The values for the x-axis of the metrics.
+    metric_values: The metric values retrieved by the query.
+    """
+
+    status: str
+    x_axis_values: List[int]
+    metric_values: List[float]
+    message: Optional[str] = Unassigned()
+
+
 class BatchPutMetricsError(Base):
     """
     BatchPutMetricsError
@@ -491,6 +535,7 @@ class S3ModelDataSource(Base):
     compression_type: Specifies how the ML model data is prepared. If you choose Gzip and choose S3Object as the value of S3DataType, S3Uri identifies an object that is a gzip-compressed TAR archive. SageMaker will attempt to decompress and untar the object during model deployment. If you choose None and chooose S3Object as the value of S3DataType, S3Uri identifies an object that represents an uncompressed ML model to deploy. If you choose None and choose S3Prefix as the value of S3DataType, S3Uri identifies a key name prefix, under which all objects represents the uncompressed ML model to deploy. If you choose None, then SageMaker will follow rules below when creating model data files under /opt/ml/model directory for use by your inference code:   If you choose S3Object as the value of S3DataType, then SageMaker will split the key of the S3 object referenced by S3Uri by slash (/), and use the last part as the filename of the file holding the content of the S3 object.   If you choose S3Prefix as the value of S3DataType, then for each S3 object under the key name pefix referenced by S3Uri, SageMaker will trim its key by the prefix, and use the remainder as the path (relative to /opt/ml/model) of the file holding the content of the S3 object. SageMaker will split the remainder by slash (/), using intermediate parts as directory names and the last part as filename of the file holding the content of the S3 object.   Do not use any of the following as file names or directory names:   An empty or blank string   A string which contains null bytes   A string longer than 255 bytes   A single dot (.)   A double dot (..)     Ambiguous file names will result in model deployment failure. For example, if your uncompressed ML model consists of two S3 objects s3://mybucket/model/weights and s3://mybucket/model/weights/part1 and you specify s3://mybucket/model/ as the value of S3Uri and S3Prefix as the value of S3DataType, then it will result in name clash between /opt/ml/model/weights (a regular file) and /opt/ml/model/weights/ (a directory).   Do not organize the model artifacts in S3 console using folders. When you create a folder in S3 console, S3 creates a 0-byte object with a key set to the folder name you provide. They key of the 0-byte object ends with a slash (/) which violates SageMaker restrictions on model artifact file names, leading to model deployment failure.
     model_access_config: Specifies the access configuration file for the ML model. You can explicitly accept the model end-user license agreement (EULA) within the ModelAccessConfig. You are responsible for reviewing and complying with any applicable license terms and making sure they are acceptable for your use case before downloading or using a model.
     hub_access_config: Configuration information for hub access.
+    manifest_s3_uri: The Amazon S3 URI of the manifest file. The manifest file is a CSV file that stores the artifact locations.
     """
 
     s3_uri: str
@@ -498,6 +543,7 @@ class S3ModelDataSource(Base):
     compression_type: str
     model_access_config: Optional[ModelAccessConfig] = Unassigned()
     hub_access_config: Optional[InferenceHubAccessConfig] = Unassigned()
+    manifest_s3_uri: Optional[str] = Unassigned()
 
 
 class ModelDataSource(Base):

{sagemaker_core-1.0.6 → sagemaker_core-1.0.7/src/sagemaker_core.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.6
+Version: 1.0.7
 Summary: An python package for sagemaker core functionalities
 Author-email: AWS <sagemaker-interests@amazon.com>
 Project-URL: Repository, https://github.com/aws/sagemaker-core.git

sagemaker_core-1.0.6/VERSION

@@ -1 +0,0 @@
-1.0.6