sagemaker-core 1.0.29__py3-none-any.whl → 1.0.31__py3-none-any.whl

sagemaker_core/_version.py

@@ -1,11 +1,3 @@
-import os
+import importlib.metadata
 
-script_dir = os.path.dirname(os.path.abspath(__file__))
-
-# Get the root directory of the project
-root_dir = os.path.abspath(os.path.join(script_dir, "..", ".."))
-
-version_file_path = os.path.join(root_dir, "VERSION")
-
-with open(version_file_path) as version_file:
-    __version__ = version_file.read().strip()
+__version__ = importlib.metadata.version("sagemaker_core")

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": {