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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.4
+Version: 1.0.6
 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

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

{sagemaker_core-1.0.4 → sagemaker_core-1.0.6}/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"},
@@ -3426,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"},
@@ -6184,6 +6281,7 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "FeatureNames": {"member_shape": "FeatureName", "member_type": "string", "type": "list"},
     "FeatureParameter": {
         "members": [
             {"name": "Key", "shape": "FeatureParameterKey", "type": "string"},
@@ -6206,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"},
@@ -6350,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"}],
@@ -7477,6 +7599,7 @@ SHAPE_DAG = {
                 "shape": "InferenceComponentHeader",
                 "type": "string",
             },
+            {"name": "SessionId", "shape": "SessionIdOrNewSessionConstantHeader", "type": "string"},
         ],
         "type": "structure",
     },
@@ -7486,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",
     },
@@ -7508,6 +7633,7 @@ SHAPE_DAG = {
                 "shape": "InferenceComponentHeader",
                 "type": "string",
             },
+            {"name": "SessionId", "shape": "SessionIdHeader", "type": "string"},
         ],
         "type": "structure",
     },
@@ -11756,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"},
@@ -11857,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"},
@@ -12053,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"},
@@ -13020,6 +13171,7 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "TargetStores": {"member_shape": "TargetStore", "member_type": "string", "type": "list"},
     "TargetTrackingScalingPolicyConfiguration": {
         "members": [
             {"name": "MetricSpecification", "shape": "MetricSpecification", "type": "structure"},
@@ -13717,6 +13869,11 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "UnprocessedIdentifiers": {
+        "member_shape": "BatchGetRecordIdentifier",
+        "member_type": "structure",
+        "type": "list",
+    },
     "UpdateActionRequest": {
         "members": [
             {"name": "ActionName", "shape": "ExperimentEntityName", "type": "string"},
@@ -14489,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"}