sagemaker-core 1.0.29__tar.gz → 1.0.31__tar.gz

{sagemaker_core-1.0.29/src/sagemaker_core.egg-info → sagemaker_core-1.0.31}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.29
+Version: 1.0.31
 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.31/VERSION

@@ -0,0 +1 @@
+1.0.31

{sagemaker_core-1.0.29 → sagemaker_core-1.0.31}/pyproject.toml

@@ -49,4 +49,4 @@ line-length = 100
 exclude = '\.ipynb$'
 
 [tool.setuptools.dynamic]
-version = { attr = "sagemaker_core._version.__version__"}
+version = { file = "VERSION"}

sagemaker_core-1.0.31/src/sagemaker_core/_version.py

@@ -0,0 +1,3 @@
+import importlib.metadata
+
+__version__ = importlib.metadata.version("sagemaker_core")

{sagemaker_core-1.0.29 → sagemaker_core-1.0.31}/src/sagemaker_core/main/code_injection/shape_dag.py

@@ -124,6 +124,10 @@ SHAPE_DAG = {
         "members": [{"name": "AlarmName", "shape": "AlarmName", "type": "string"}],
         "type": "structure",
     },
+    "AlarmDetails": {
+        "members": [{"name": "AlarmName", "shape": "AlarmName", "type": "string"}],
+        "type": "structure",
+    },
     "AlarmList": {"member_shape": "Alarm", "member_type": "structure", "type": "list"},
     "AlgorithmSpecification": {
         "members": [
@@ -734,6 +738,11 @@ SHAPE_DAG = {
         "type": "structure",
     },
     "AutoParameters": {"member_shape": "AutoParameter", "member_type": "structure", "type": "list"},
+    "AutoRollbackAlarms": {
+        "member_shape": "AlarmDetails",
+        "member_type": "structure",
+        "type": "list",
+    },
     "AutoRollbackConfig": {
         "members": [{"name": "Alarms", "shape": "AlarmList", "type": "list"}],
         "type": "structure",
@@ -1069,6 +1078,13 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "CapacitySizeConfig": {
+        "members": [
+            {"name": "Type", "shape": "NodeUnavailabilityType", "type": "string"},
+            {"name": "Value", "shape": "NodeUnavailabilityValue", "type": "integer"},
+        ],
+        "type": "structure",
+    },
     "CaptureContentTypeHeader": {
         "members": [
             {"name": "CsvContentTypes", "shape": "CsvContentTypes", "type": "list"},
@@ -1272,6 +1288,11 @@ SHAPE_DAG = {
                 "type": "string",
             },
             {"name": "OverrideVpcConfig", "shape": "VpcConfig", "type": "structure"},
+            {
+                "name": "ScheduledUpdateConfig",
+                "shape": "ScheduledUpdateConfig",
+                "type": "structure",
+            },
         ],
         "type": "structure",
     },
@@ -1296,6 +1317,11 @@ SHAPE_DAG = {
             {"name": "OnStartDeepHealthChecks", "shape": "OnStartDeepHealthChecks", "type": "list"},
             {"name": "TrainingPlanArn", "shape": "TrainingPlanArn", "type": "string"},
             {"name": "OverrideVpcConfig", "shape": "VpcConfig", "type": "structure"},
+            {
+                "name": "ScheduledUpdateConfig",
+                "shape": "ScheduledUpdateConfig",
+                "type": "structure",
+            },
         ],
         "type": "structure",
     },
@@ -1352,6 +1378,7 @@ SHAPE_DAG = {
             },
             {"name": "InstanceType", "shape": "ClusterInstanceType", "type": "string"},
             {"name": "LaunchTime", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "LastSoftwareUpdateTime", "shape": "Timestamp", "type": "timestamp"},
             {"name": "LifeCycleConfig", "shape": "ClusterLifeCycleConfig", "type": "structure"},
             {"name": "OverrideVpcConfig", "shape": "VpcConfig", "type": "structure"},
             {"name": "ThreadsPerCore", "shape": "ClusterThreadsPerCore", "type": "integer"},
@@ -1379,6 +1406,7 @@ SHAPE_DAG = {
             {"name": "InstanceId", "shape": "String", "type": "string"},
             {"name": "InstanceType", "shape": "ClusterInstanceType", "type": "string"},
             {"name": "LaunchTime", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "LastSoftwareUpdateTime", "shape": "Timestamp", "type": "timestamp"},
             {
                 "name": "InstanceStatus",
                 "shape": "ClusterInstanceStatusDetails",
@@ -3889,6 +3917,22 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "DeploymentConfiguration": {
+        "members": [
+            {
+                "name": "RollingUpdatePolicy",
+                "shape": "RollingDeploymentPolicy",
+                "type": "structure",
+            },
+            {
+                "name": "WaitIntervalInSeconds",
+                "shape": "WaitTimeIntervalInSeconds",
+                "type": "integer",
+            },
+            {"name": "AutoRollbackConfiguration", "shape": "AutoRollbackAlarms", "type": "list"},
+        ],
+        "type": "structure",
+    },
     "DeploymentRecommendation": {
         "members": [
             {"name": "RecommendationStatus", "shape": "RecommendationStatus", "type": "string"},
@@ -10885,6 +10929,7 @@ SHAPE_DAG = {
             {"name": "CreationTime", "shape": "CreationTime", "type": "timestamp"},
             {"name": "ModelPackageStatus", "shape": "ModelPackageStatus", "type": "string"},
             {"name": "ModelApprovalStatus", "shape": "ModelApprovalStatus", "type": "string"},
+            {"name": "ModelLifeCycle", "shape": "ModelLifeCycle", "type": "structure"},
         ],
         "type": "structure",
     },
@@ -11153,7 +11198,12 @@ SHAPE_DAG = {
     "MonitoringDatasetFormat": {
         "members": [
             {"name": "Csv", "shape": "MonitoringCsvDatasetFormat", "type": "structure"},
-            {"name": "Json", "shape": "MonitoringJsonDatasetFormat", "type": "structure"},
+            {
+                "alias": "json",
+                "name": "JsonFormat",
+                "shape": "MonitoringJsonDatasetFormat",
+                "type": "structure",
+            },
             {"name": "Parquet", "shape": "MonitoringParquetDatasetFormat", "type": "structure"},
         ],
         "type": "structure",
@@ -13168,6 +13218,17 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "RollingDeploymentPolicy": {
+        "members": [
+            {"name": "MaximumBatchSize", "shape": "CapacitySizeConfig", "type": "structure"},
+            {
+                "name": "RollbackMaximumBatchSize",
+                "shape": "CapacitySizeConfig",
+                "type": "structure",
+            },
+        ],
+        "type": "structure",
+    },
     "RollingUpdatePolicy": {
         "members": [
             {"name": "MaximumBatchSize", "shape": "CapacitySize", "type": "structure"},
@@ -13274,6 +13335,13 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "ScheduledUpdateConfig": {
+        "members": [
+            {"name": "ScheduleExpression", "shape": "CronScheduleExpression", "type": "string"},
+            {"name": "DeploymentConfig", "shape": "DeploymentConfiguration", "type": "structure"},
+        ],
+        "type": "structure",
+    },
     "SchedulerConfig": {
         "members": [
             {"name": "PriorityClasses", "shape": "PriorityClassList", "type": "list"},
@@ -14850,8 +14918,27 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "UpdateClusterSoftwareInstanceGroupSpecification": {
+        "members": [
+            {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"}
+        ],
+        "type": "structure",
+    },
+    "UpdateClusterSoftwareInstanceGroups": {
+        "member_shape": "UpdateClusterSoftwareInstanceGroupSpecification",
+        "member_type": "structure",
+        "type": "list",
+    },
     "UpdateClusterSoftwareRequest": {
-        "members": [{"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"}],
+        "members": [
+            {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
+            {
+                "name": "InstanceGroups",
+                "shape": "UpdateClusterSoftwareInstanceGroups",
+                "type": "list",
+            },
+            {"name": "DeploymentConfig", "shape": "DeploymentConfiguration", "type": "structure"},
+        ],
         "type": "structure",
     },
     "UpdateClusterSoftwareResponse": {