sagemaker-core 1.0.42__tar.gz → 1.0.44__tar.gz

{sagemaker_core-1.0.42/src/sagemaker_core.egg-info → sagemaker_core-1.0.44}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.42
+Version: 1.0.44
 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
@@ -22,7 +22,7 @@ Requires-Dist: jsonschema<5.0.0
 Requires-Dist: platformdirs<5.0.0,>=4.0.0
 Requires-Dist: rich<15.0.0,>=14.0.0
 Requires-Dist: mock<5.0,>4.0
-Requires-Dist: importlib-metadata<7.0,>=1.4.0
+Requires-Dist: importlib-metadata<=9.0,>=1.4.0
 Provides-Extra: codegen
 Requires-Dist: black<25.0.0,>=24.3.0; extra == "codegen"
 Requires-Dist: pandas<3.0.0,>=2.0.0; extra == "codegen"

sagemaker_core-1.0.44/VERSION

@@ -0,0 +1 @@
+1.0.44

{sagemaker_core-1.0.42 → sagemaker_core-1.0.44}/pyproject.toml

@@ -19,7 +19,7 @@ dependencies = [
     "platformdirs>=4.0.0, <5.0.0",
     "rich>=14.0.0, <15.0.0",
     "mock>4.0, <5.0",
-    "importlib-metadata<7.0,>=1.4.0",
+    "importlib-metadata<=9.0,>=1.4.0",
 ]
 requires-python = ">=3.8"
 classifiers = [

{sagemaker_core-1.0.42 → sagemaker_core-1.0.44}/src/sagemaker_core/main/code_injection/shape_dag.py

@@ -1506,6 +1506,71 @@ SHAPE_DAG = {
         "members": [{"name": "ClusterArn", "shape": "EksClusterArn", "type": "string"}],
         "type": "structure",
     },
+    "ClusterRestrictedInstanceGroupDetails": {
+        "members": [
+            {"name": "CurrentCount", "shape": "ClusterNonNegativeInstanceCount", "type": "integer"},
+            {"name": "TargetCount", "shape": "ClusterInstanceCount", "type": "integer"},
+            {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
+            {"name": "InstanceType", "shape": "ClusterInstanceType", "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"},
+            {"name": "Status", "shape": "InstanceGroupStatus", "type": "string"},
+            {"name": "TrainingPlanArn", "shape": "TrainingPlanArn", "type": "string"},
+            {
+                "name": "TrainingPlanStatus",
+                "shape": "InstanceGroupTrainingPlanStatus",
+                "type": "string",
+            },
+            {"name": "OverrideVpcConfig", "shape": "VpcConfig", "type": "structure"},
+            {
+                "name": "ScheduledUpdateConfig",
+                "shape": "ScheduledUpdateConfig",
+                "type": "structure",
+            },
+            {"name": "EnvironmentConfig", "shape": "EnvironmentConfigDetails", "type": "structure"},
+        ],
+        "type": "structure",
+    },
+    "ClusterRestrictedInstanceGroupDetailsList": {
+        "member_shape": "ClusterRestrictedInstanceGroupDetails",
+        "member_type": "structure",
+        "type": "list",
+    },
+    "ClusterRestrictedInstanceGroupSpecification": {
+        "members": [
+            {"name": "InstanceCount", "shape": "ClusterInstanceCount", "type": "integer"},
+            {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
+            {"name": "InstanceType", "shape": "ClusterInstanceType", "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"},
+            {"name": "TrainingPlanArn", "shape": "TrainingPlanArn", "type": "string"},
+            {"name": "OverrideVpcConfig", "shape": "VpcConfig", "type": "structure"},
+            {
+                "name": "ScheduledUpdateConfig",
+                "shape": "ScheduledUpdateConfig",
+                "type": "structure",
+            },
+            {"name": "EnvironmentConfig", "shape": "EnvironmentConfig", "type": "structure"},
+        ],
+        "type": "structure",
+    },
+    "ClusterRestrictedInstanceGroupSpecifications": {
+        "member_shape": "ClusterRestrictedInstanceGroupSpecification",
+        "member_type": "structure",
+        "type": "list",
+    },
     "ClusterSchedulerConfigSummary": {
         "members": [
             {
@@ -2010,6 +2075,11 @@ SHAPE_DAG = {
                 "shape": "ClusterInstanceGroupSpecifications",
                 "type": "list",
             },
+            {
+                "name": "RestrictedInstanceGroups",
+                "shape": "ClusterRestrictedInstanceGroupSpecifications",
+                "type": "list",
+            },
             {"name": "VpcConfig", "shape": "VpcConfig", "type": "structure"},
             {"name": "Tags", "shape": "TagList", "type": "list"},
             {"name": "Orchestrator", "shape": "ClusterOrchestrator", "type": "structure"},
@@ -4372,6 +4442,11 @@ SHAPE_DAG = {
             {"name": "CreationTime", "shape": "Timestamp", "type": "timestamp"},
             {"name": "FailureMessage", "shape": "String", "type": "string"},
             {"name": "InstanceGroups", "shape": "ClusterInstanceGroupDetailsList", "type": "list"},
+            {
+                "name": "RestrictedInstanceGroups",
+                "shape": "ClusterRestrictedInstanceGroupDetailsList",
+                "type": "list",
+            },
             {"name": "VpcConfig", "shape": "VpcConfig", "type": "structure"},
             {"name": "Orchestrator", "shape": "ClusterOrchestrator", "type": "structure"},
             {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
@@ -5743,11 +5818,15 @@ SHAPE_DAG = {
                 "shape": "SelectiveExecutionConfig",
                 "type": "structure",
             },
+            {"name": "PipelineVersionId", "shape": "PipelineVersionId", "type": "long"},
         ],
         "type": "structure",
     },
     "DescribePipelineRequest": {
-        "members": [{"name": "PipelineName", "shape": "PipelineNameOrArn", "type": "string"}],
+        "members": [
+            {"name": "PipelineName", "shape": "PipelineNameOrArn", "type": "string"},
+            {"name": "PipelineVersionId", "shape": "PipelineVersionId", "type": "long"},
+        ],
         "type": "structure",
     },
     "DescribePipelineResponse": {
@@ -5769,6 +5848,16 @@ SHAPE_DAG = {
                 "shape": "ParallelismConfiguration",
                 "type": "structure",
             },
+            {
+                "name": "PipelineVersionDisplayName",
+                "shape": "PipelineVersionName",
+                "type": "string",
+            },
+            {
+                "name": "PipelineVersionDescription",
+                "shape": "PipelineVersionDescription",
+                "type": "string",
+            },
         ],
         "type": "structure",
     },
@@ -6734,6 +6823,17 @@ SHAPE_DAG = {
         "type": "list",
     },
     "Endpoints": {"member_shape": "EndpointInfo", "member_type": "structure", "type": "list"},
+    "EnvironmentConfig": {
+        "members": [{"name": "FSxLustreConfig", "shape": "FSxLustreConfig", "type": "structure"}],
+        "type": "structure",
+    },
+    "EnvironmentConfigDetails": {
+        "members": [
+            {"name": "FSxLustreConfig", "shape": "FSxLustreConfig", "type": "structure"},
+            {"name": "S3OutputPath", "shape": "S3Uri", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "EnvironmentMap": {
         "key_shape": "EnvironmentKey",
         "key_type": "string",
@@ -6833,6 +6933,17 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "FSxLustreConfig": {
+        "members": [
+            {"name": "SizeInGiB", "shape": "FSxLustreSizeInGiB", "type": "integer"},
+            {
+                "name": "PerUnitStorageThroughput",
+                "shape": "FSxLustrePerUnitStorageThroughput",
+                "type": "integer",
+            },
+        ],
+        "type": "structure",
+    },
     "FSxLustreFileSystem": {
         "members": [{"name": "FileSystemId", "shape": "FileSystemId", "type": "string"}],
         "type": "structure",
@@ -10037,6 +10148,28 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "ListPipelineVersionsRequest": {
+        "members": [
+            {"name": "PipelineName", "shape": "PipelineNameOrArn", "type": "string"},
+            {"name": "CreatedAfter", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "CreatedBefore", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "SortOrder", "shape": "SortOrder", "type": "string"},
+            {"name": "NextToken", "shape": "NextToken", "type": "string"},
+            {"name": "MaxResults", "shape": "MaxResults", "type": "integer"},
+        ],
+        "type": "structure",
+    },
+    "ListPipelineVersionsResponse": {
+        "members": [
+            {
+                "name": "PipelineVersionSummaries",
+                "shape": "PipelineVersionSummaryList",
+                "type": "list",
+            },
+            {"name": "NextToken", "shape": "NextToken", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "ListPipelinesRequest": {
         "members": [
             {"name": "PipelineNamePrefix", "shape": "PipelineName", "type": "string"},
@@ -12207,6 +12340,12 @@ SHAPE_DAG = {
                 "type": "structure",
             },
             {"name": "PipelineParameters", "shape": "ParameterList", "type": "list"},
+            {"name": "PipelineVersionId", "shape": "PipelineVersionId", "type": "long"},
+            {
+                "name": "PipelineVersionDisplayName",
+                "shape": "PipelineVersionName",
+                "type": "string",
+            },
         ],
         "type": "structure",
     },
@@ -12309,6 +12448,70 @@ SHAPE_DAG = {
         "member_type": "structure",
         "type": "list",
     },
+    "PipelineVersion": {
+        "members": [
+            {"name": "PipelineArn", "shape": "PipelineArn", "type": "string"},
+            {"name": "PipelineVersionId", "shape": "PipelineVersionId", "type": "long"},
+            {
+                "name": "PipelineVersionDisplayName",
+                "shape": "PipelineVersionName",
+                "type": "string",
+            },
+            {
+                "name": "PipelineVersionDescription",
+                "shape": "PipelineVersionDescription",
+                "type": "string",
+            },
+            {"name": "CreationTime", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "LastModifiedTime", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "CreatedBy", "shape": "UserContext", "type": "structure"},
+            {"name": "LastModifiedBy", "shape": "UserContext", "type": "structure"},
+            {
+                "name": "LastExecutedPipelineExecutionArn",
+                "shape": "PipelineExecutionArn",
+                "type": "string",
+            },
+            {
+                "name": "LastExecutedPipelineExecutionDisplayName",
+                "shape": "PipelineExecutionName",
+                "type": "string",
+            },
+            {
+                "name": "LastExecutedPipelineExecutionStatus",
+                "shape": "PipelineExecutionStatus",
+                "type": "string",
+            },
+        ],
+        "type": "structure",
+    },
+    "PipelineVersionSummary": {
+        "members": [
+            {"name": "PipelineArn", "shape": "PipelineArn", "type": "string"},
+            {"name": "PipelineVersionId", "shape": "PipelineVersionId", "type": "long"},
+            {"name": "CreationTime", "shape": "Timestamp", "type": "timestamp"},
+            {
+                "name": "PipelineVersionDescription",
+                "shape": "PipelineVersionDescription",
+                "type": "string",
+            },
+            {
+                "name": "PipelineVersionDisplayName",
+                "shape": "PipelineVersionName",
+                "type": "string",
+            },
+            {
+                "name": "LastExecutionPipelineExecutionArn",
+                "shape": "PipelineExecutionArn",
+                "type": "string",
+            },
+        ],
+        "type": "structure",
+    },
+    "PipelineVersionSummaryList": {
+        "member_shape": "PipelineVersionSummary",
+        "member_type": "structure",
+        "type": "list",
+    },
     "PredefinedMetricSpecification": {
         "members": [{"name": "PredefinedMetricType", "shape": "String", "type": "string"}],
         "type": "structure",
@@ -13582,6 +13785,7 @@ SHAPE_DAG = {
             {"name": "ModelPackageGroup", "shape": "ModelPackageGroup", "type": "structure"},
             {"name": "Pipeline", "shape": "Pipeline", "type": "structure"},
             {"name": "PipelineExecution", "shape": "PipelineExecution", "type": "structure"},
+            {"name": "PipelineVersion", "shape": "PipelineVersion", "type": "structure"},
             {"name": "FeatureGroup", "shape": "FeatureGroup", "type": "structure"},
             {"name": "FeatureMetadata", "shape": "FeatureMetadata", "type": "structure"},
             {"name": "Project", "shape": "Project", "type": "structure"},
@@ -13997,6 +14201,7 @@ SHAPE_DAG = {
                 "shape": "SelectiveExecutionConfig",
                 "type": "structure",
             },
+            {"name": "PipelineVersionId", "shape": "PipelineVersionId", "type": "long"},
         ],
         "type": "structure",
     },
@@ -15138,6 +15343,11 @@ SHAPE_DAG = {
                 "shape": "ClusterInstanceGroupSpecifications",
                 "type": "list",
             },
+            {
+                "name": "RestrictedInstanceGroups",
+                "shape": "ClusterRestrictedInstanceGroupSpecifications",
+                "type": "list",
+            },
             {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
             {
                 "name": "InstanceGroupsToDelete",
@@ -15749,7 +15959,34 @@ SHAPE_DAG = {
         "type": "structure",
     },
     "UpdatePipelineResponse": {
-        "members": [{"name": "PipelineArn", "shape": "PipelineArn", "type": "string"}],
+        "members": [
+            {"name": "PipelineArn", "shape": "PipelineArn", "type": "string"},
+            {"name": "PipelineVersionId", "shape": "PipelineVersionId", "type": "long"},
+        ],
+        "type": "structure",
+    },
+    "UpdatePipelineVersionRequest": {
+        "members": [
+            {"name": "PipelineArn", "shape": "PipelineArn", "type": "string"},
+            {"name": "PipelineVersionId", "shape": "PipelineVersionId", "type": "long"},
+            {
+                "name": "PipelineVersionDisplayName",
+                "shape": "PipelineVersionName",
+                "type": "string",
+            },
+            {
+                "name": "PipelineVersionDescription",
+                "shape": "PipelineVersionDescription",
+                "type": "string",
+            },
+        ],
+        "type": "structure",
+    },
+    "UpdatePipelineVersionResponse": {
+        "members": [
+            {"name": "PipelineArn", "shape": "PipelineArn", "type": "string"},
+            {"name": "PipelineVersionId", "shape": "PipelineVersionId", "type": "long"},
+        ],
         "type": "structure",
     },
     "UpdateProjectInput": {