sagemaker-core 1.0.3__tar.gz → 1.0.5__tar.gz

{sagemaker_core-1.0.3/src/sagemaker_core.egg-info → sagemaker_core-1.0.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.3
+Version: 1.0.5
 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.5/VERSION

@@ -0,0 +1 @@
+1.0.5

{sagemaker_core-1.0.3 → sagemaker_core-1.0.5}/src/sagemaker_core/main/code_injection/shape_dag.py

@@ -1,4 +1,8 @@
 SHAPE_DAG = {
+    "AccessForbidden": {
+        "members": [{"name": "Message", "shape": "Message", "type": "string"}],
+        "type": "structure",
+    },
     "ActionSource": {
         "members": [
             {"name": "SourceUri", "shape": "SourceUri", "type": "string"},
@@ -793,6 +797,89 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "BatchGetRecordError": {
+        "members": [
+            {"name": "FeatureGroupName", "shape": "ValueAsString", "type": "string"},
+            {"name": "RecordIdentifierValueAsString", "shape": "ValueAsString", "type": "string"},
+            {"name": "ErrorCode", "shape": "ValueAsString", "type": "string"},
+            {"name": "ErrorMessage", "shape": "Message", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "BatchGetRecordErrors": {
+        "member_shape": "BatchGetRecordError",
+        "member_type": "structure",
+        "type": "list",
+    },
+    "BatchGetRecordIdentifier": {
+        "members": [
+            {"name": "FeatureGroupName", "shape": "FeatureGroupNameOrArn", "type": "string"},
+            {
+                "name": "RecordIdentifiersValueAsString",
+                "shape": "RecordIdentifiers",
+                "type": "list",
+            },
+            {"name": "FeatureNames", "shape": "FeatureNames", "type": "list"},
+        ],
+        "type": "structure",
+    },
+    "BatchGetRecordIdentifiers": {
+        "member_shape": "BatchGetRecordIdentifier",
+        "member_type": "structure",
+        "type": "list",
+    },
+    "BatchGetRecordRequest": {
+        "members": [
+            {"name": "Identifiers", "shape": "BatchGetRecordIdentifiers", "type": "list"},
+            {"name": "ExpirationTimeResponse", "shape": "ExpirationTimeResponse", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "BatchGetRecordResponse": {
+        "members": [
+            {"name": "Records", "shape": "BatchGetRecordResultDetails", "type": "list"},
+            {"name": "Errors", "shape": "BatchGetRecordErrors", "type": "list"},
+            {"name": "UnprocessedIdentifiers", "shape": "UnprocessedIdentifiers", "type": "list"},
+        ],
+        "type": "structure",
+    },
+    "BatchGetRecordResultDetail": {
+        "members": [
+            {"name": "FeatureGroupName", "shape": "ValueAsString", "type": "string"},
+            {"name": "RecordIdentifierValueAsString", "shape": "ValueAsString", "type": "string"},
+            {"name": "Record", "shape": "Record", "type": "list"},
+            {"name": "ExpiresAt", "shape": "ExpiresAt", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "BatchGetRecordResultDetails": {
+        "member_shape": "BatchGetRecordResultDetail",
+        "member_type": "structure",
+        "type": "list",
+    },
+    "BatchPutMetricsError": {
+        "members": [
+            {"name": "Code", "shape": "PutMetricsErrorCode", "type": "string"},
+            {"name": "MetricIndex", "shape": "Integer", "type": "integer"},
+        ],
+        "type": "structure",
+    },
+    "BatchPutMetricsErrorList": {
+        "member_shape": "BatchPutMetricsError",
+        "member_type": "structure",
+        "type": "list",
+    },
+    "BatchPutMetricsRequest": {
+        "members": [
+            {"name": "TrialComponentName", "shape": "ExperimentEntityName", "type": "string"},
+            {"name": "MetricData", "shape": "RawMetricDataList", "type": "list"},
+        ],
+        "type": "structure",
+    },
+    "BatchPutMetricsResponse": {
+        "members": [{"name": "Errors", "shape": "BatchPutMetricsErrorList", "type": "list"}],
+        "type": "structure",
+    },
     "BatchTransformInput": {
         "members": [
             {"name": "DataCapturedDestinationS3Uri", "shape": "DestinationS3Uri", "type": "string"},
@@ -1131,13 +1218,14 @@ SHAPE_DAG = {
             {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
             {"name": "InstanceType", "shape": "ClusterInstanceType", "type": "string"},
             {"name": "LifeCycleConfig", "shape": "ClusterLifeCycleConfig", "type": "structure"},
-            {"name": "ExecutionRole", "shape": "IAMRoleArn", "type": "string"},
+            {"name": "ExecutionRole", "shape": "RoleArn", "type": "string"},
             {"name": "ThreadsPerCore", "shape": "ClusterThreadsPerCore", "type": "integer"},
             {
                 "name": "InstanceStorageConfigs",
                 "shape": "ClusterInstanceStorageConfigs",
                 "type": "list",
             },
+            {"name": "OnStartDeepHealthChecks", "shape": "OnStartDeepHealthChecks", "type": "list"},
         ],
         "type": "structure",
     },
@@ -1152,13 +1240,14 @@ SHAPE_DAG = {
             {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
             {"name": "InstanceType", "shape": "ClusterInstanceType", "type": "string"},
             {"name": "LifeCycleConfig", "shape": "ClusterLifeCycleConfig", "type": "structure"},
-            {"name": "ExecutionRole", "shape": "IAMRoleArn", "type": "string"},
+            {"name": "ExecutionRole", "shape": "RoleArn", "type": "string"},
             {"name": "ThreadsPerCore", "shape": "ClusterThreadsPerCore", "type": "integer"},
             {
                 "name": "InstanceStorageConfigs",
                 "shape": "ClusterInstanceStorageConfigs",
                 "type": "list",
             },
+            {"name": "OnStartDeepHealthChecks", "shape": "OnStartDeepHealthChecks", "type": "list"},
         ],
         "type": "structure",
     },
@@ -1242,6 +1331,14 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "ClusterOrchestrator": {
+        "members": [{"name": "Eks", "shape": "ClusterOrchestratorEksConfig", "type": "structure"}],
+        "type": "structure",
+    },
+    "ClusterOrchestratorEksConfig": {
+        "members": [{"name": "ClusterArn", "shape": "EksClusterArn", "type": "string"}],
+        "type": "structure",
+    },
     "ClusterSummaries": {
         "member_shape": "ClusterSummary",
         "member_type": "structure",
@@ -1659,6 +1756,8 @@ SHAPE_DAG = {
             },
             {"name": "VpcConfig", "shape": "VpcConfig", "type": "structure"},
             {"name": "Tags", "shape": "TagList", "type": "list"},
+            {"name": "Orchestrator", "shape": "ClusterOrchestrator", "type": "structure"},
+            {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
         ],
         "type": "structure",
     },
@@ -3414,6 +3513,16 @@ SHAPE_DAG = {
         "members": [{"name": "ProjectName", "shape": "ProjectEntityName", "type": "string"}],
         "type": "structure",
     },
+    "DeleteRecordRequest": {
+        "members": [
+            {"name": "FeatureGroupName", "shape": "FeatureGroupNameOrArn", "type": "string"},
+            {"name": "RecordIdentifierValueAsString", "shape": "ValueAsString", "type": "string"},
+            {"name": "EventTime", "shape": "ValueAsString", "type": "string"},
+            {"name": "TargetStores", "shape": "TargetStores", "type": "list"},
+            {"name": "DeletionMode", "shape": "DeletionMode", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "DeleteSpaceRequest": {
         "members": [
             {"name": "DomainId", "shape": "DomainId", "type": "string"},
@@ -3818,6 +3927,8 @@ SHAPE_DAG = {
             {"name": "FailureMessage", "shape": "String", "type": "string"},
             {"name": "InstanceGroups", "shape": "ClusterInstanceGroupDetailsList", "type": "list"},
             {"name": "VpcConfig", "shape": "VpcConfig", "type": "structure"},
+            {"name": "Orchestrator", "shape": "ClusterOrchestrator", "type": "structure"},
+            {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
         ],
         "type": "structure",
     },
@@ -6170,6 +6281,7 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "FeatureNames": {"member_shape": "FeatureName", "member_type": "string", "type": "list"},
     "FeatureParameter": {
         "members": [
             {"name": "Key", "shape": "FeatureParameterKey", "type": "string"},
@@ -6192,6 +6304,14 @@ SHAPE_DAG = {
         "member_type": "structure",
         "type": "list",
     },
+    "FeatureValue": {
+        "members": [
+            {"name": "FeatureName", "shape": "FeatureName", "type": "string"},
+            {"name": "ValueAsString", "shape": "ValueAsString", "type": "string"},
+            {"name": "ValueAsStringList", "shape": "ValueAsStringList", "type": "list"},
+        ],
+        "type": "structure",
+    },
     "FileSource": {
         "members": [
             {"name": "ContentType", "shape": "ContentType", "type": "string"},
@@ -6336,6 +6456,22 @@ SHAPE_DAG = {
         "members": [{"name": "ResourcePolicy", "shape": "PolicyString", "type": "string"}],
         "type": "structure",
     },
+    "GetRecordRequest": {
+        "members": [
+            {"name": "FeatureGroupName", "shape": "FeatureGroupNameOrArn", "type": "string"},
+            {"name": "RecordIdentifierValueAsString", "shape": "ValueAsString", "type": "string"},
+            {"name": "FeatureNames", "shape": "FeatureNames", "type": "list"},
+            {"name": "ExpirationTimeResponse", "shape": "ExpirationTimeResponse", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "GetRecordResponse": {
+        "members": [
+            {"name": "Record", "shape": "Record", "type": "list"},
+            {"name": "ExpiresAt", "shape": "ExpiresAt", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "GetSagemakerServicecatalogPortfolioStatusInput": {"members": [], "type": "structure"},
     "GetSagemakerServicecatalogPortfolioStatusOutput": {
         "members": [{"name": "Status", "shape": "SagemakerServicecatalogStatus", "type": "string"}],
@@ -7463,6 +7599,7 @@ SHAPE_DAG = {
                 "shape": "InferenceComponentHeader",
                 "type": "string",
             },
+            {"name": "SessionId", "shape": "SessionIdOrNewSessionConstantHeader", "type": "string"},
         ],
         "type": "structure",
     },
@@ -7472,6 +7609,8 @@ SHAPE_DAG = {
             {"name": "ContentType", "shape": "Header", "type": "string"},
             {"name": "InvokedProductionVariant", "shape": "Header", "type": "string"},
             {"name": "CustomAttributes", "shape": "CustomAttributesHeader", "type": "string"},
+            {"name": "NewSessionId", "shape": "NewSessionResponseHeader", "type": "string"},
+            {"name": "ClosedSessionId", "shape": "SessionIdHeader", "type": "string"},
         ],
         "type": "structure",
     },
@@ -7494,6 +7633,7 @@ SHAPE_DAG = {
                 "shape": "InferenceComponentHeader",
                 "type": "string",
             },
+            {"name": "SessionId", "shape": "SessionIdHeader", "type": "string"},
         ],
         "type": "structure",
     },
@@ -10775,6 +10915,11 @@ SHAPE_DAG = {
         "members": [{"name": "Groups", "shape": "Groups", "type": "list"}],
         "type": "structure",
     },
+    "OnStartDeepHealthChecks": {
+        "member_shape": "DeepHealthCheckType",
+        "member_type": "string",
+        "type": "list",
+    },
     "OnlineStoreConfig": {
         "members": [
             {"name": "SecurityConfig", "shape": "OnlineStoreSecurityConfig", "type": "structure"},
@@ -11737,6 +11882,15 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "PutRecordRequest": {
+        "members": [
+            {"name": "FeatureGroupName", "shape": "FeatureGroupNameOrArn", "type": "string"},
+            {"name": "Record", "shape": "Record", "type": "list"},
+            {"name": "TargetStores", "shape": "TargetStores", "type": "list"},
+            {"name": "TtlDuration", "shape": "TtlDuration", "type": "structure"},
+        ],
+        "type": "structure",
+    },
     "QualityCheckStepMetadata": {
         "members": [
             {"name": "CheckType", "shape": "String256", "type": "string"},
@@ -11838,6 +11992,20 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "RawMetricData": {
+        "members": [
+            {"name": "MetricName", "shape": "MetricName", "type": "string"},
+            {"name": "Timestamp", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "Step", "shape": "Step", "type": "integer"},
+            {"name": "Value", "shape": "Double", "type": "double"},
+        ],
+        "type": "structure",
+    },
+    "RawMetricDataList": {
+        "member_shape": "RawMetricData",
+        "member_type": "structure",
+        "type": "list",
+    },
     "RealTimeInferenceConfig": {
         "members": [
             {"name": "InstanceType", "shape": "InstanceType", "type": "string"},
@@ -12034,6 +12202,8 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "Record": {"member_shape": "FeatureValue", "member_type": "structure", "type": "list"},
+    "RecordIdentifiers": {"member_shape": "ValueAsString", "member_type": "string", "type": "list"},
     "RedshiftDatasetDefinition": {
         "members": [
             {"name": "ClusterId", "shape": "RedshiftClusterId", "type": "string"},
@@ -13001,6 +13171,7 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "TargetStores": {"member_shape": "TargetStore", "member_type": "string", "type": "list"},
     "TargetTrackingScalingPolicyConfiguration": {
         "members": [
             {"name": "MetricSpecification", "shape": "MetricSpecification", "type": "structure"},
@@ -13698,6 +13869,11 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "UnprocessedIdentifiers": {
+        "member_shape": "BatchGetRecordIdentifier",
+        "member_type": "structure",
+        "type": "list",
+    },
     "UpdateActionRequest": {
         "members": [
             {"name": "ActionName", "shape": "ExperimentEntityName", "type": "string"},
@@ -13766,6 +13942,7 @@ SHAPE_DAG = {
                 "shape": "ClusterInstanceGroupSpecifications",
                 "type": "list",
             },
+            {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
         ],
         "type": "structure",
     },
@@ -14469,6 +14646,7 @@ SHAPE_DAG = {
         "members": [{"name": "Message", "shape": "Message", "type": "string"}],
         "type": "structure",
     },
+    "ValueAsStringList": {"member_shape": "ValueAsString", "member_type": "string", "type": "list"},
     "VariantProperty": {
         "members": [
             {"name": "VariantPropertyType", "shape": "VariantPropertyType", "type": "string"}