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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.42
+Version: 1.0.43
 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.43/VERSION

@@ -0,0 +1 @@
+1.0.43

{sagemaker_core-1.0.42 → sagemaker_core-1.0.43}/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": {

{sagemaker_core-1.0.42 → sagemaker_core-1.0.43}/src/sagemaker_core/main/resources.py

@@ -3219,6 +3219,7 @@ class Cluster(Base):
         cluster_name: The name of the SageMaker HyperPod cluster.
         creation_time: The time when the SageMaker Cluster is created.
         failure_message: The failure message of the SageMaker HyperPod cluster.
+        restricted_instance_groups: The specialized instance groups for training models like Amazon Nova to be created in the SageMaker HyperPod cluster.
         vpc_config:
         orchestrator: The type of orchestrator used for the SageMaker HyperPod cluster.
         node_recovery: The node recovery mode configured for the SageMaker HyperPod cluster.
@@ -3231,6 +3232,9 @@ class Cluster(Base):
     creation_time: Optional[datetime.datetime] = Unassigned()
     failure_message: Optional[str] = Unassigned()
     instance_groups: Optional[List[shapes.ClusterInstanceGroupDetails]] = Unassigned()
+    restricted_instance_groups: Optional[List[shapes.ClusterRestrictedInstanceGroupDetails]] = (
+        Unassigned()
+    )
     vpc_config: Optional[shapes.VpcConfig] = Unassigned()
     orchestrator: Optional[shapes.ClusterOrchestrator] = Unassigned()
     node_recovery: Optional[str] = Unassigned()
@@ -3276,6 +3280,9 @@ class Cluster(Base):
         cls,
         cluster_name: str,
         instance_groups: Optional[List[shapes.ClusterInstanceGroupSpecification]] = Unassigned(),
+        restricted_instance_groups: Optional[
+            List[shapes.ClusterRestrictedInstanceGroupSpecification]
+        ] = Unassigned(),
         vpc_config: Optional[shapes.VpcConfig] = Unassigned(),
         tags: Optional[List[shapes.Tag]] = Unassigned(),
         orchestrator: Optional[shapes.ClusterOrchestrator] = Unassigned(),
@@ -3289,6 +3296,7 @@ class Cluster(Base):
         Parameters:
             cluster_name: The name for the new SageMaker HyperPod cluster.
             instance_groups: The instance groups to be created in the SageMaker HyperPod cluster.
+            restricted_instance_groups: The specialized instance groups for training models like Amazon Nova to be created in the SageMaker HyperPod cluster.
             vpc_config: Specifies the Amazon Virtual Private Cloud (VPC) that is associated with the Amazon SageMaker HyperPod cluster. You can control access to and from your resources by configuring your VPC. For more information, see Give SageMaker access to resources in your Amazon VPC.  When your Amazon VPC and subnets support IPv6, network communications differ based on the cluster orchestration platform:   Slurm-orchestrated clusters automatically configure nodes with dual IPv6 and IPv4 addresses, allowing immediate IPv6 network communications.   In Amazon EKS-orchestrated clusters, nodes receive dual-stack addressing, but pods can only use IPv6 when the Amazon EKS cluster is explicitly IPv6-enabled. For information about deploying an IPv6 Amazon EKS cluster, see Amazon EKS IPv6 Cluster Deployment.   Additional resources for IPv6 configuration:   For information about adding IPv6 support to your VPC, see to IPv6 Support for VPC.   For information about creating a new IPv6-compatible VPC, see Amazon VPC Creation Guide.   To configure SageMaker HyperPod with a custom Amazon VPC, see Custom Amazon VPC Setup for SageMaker HyperPod.
             tags: Custom tags for managing the SageMaker HyperPod cluster as an Amazon Web Services resource. You can add tags to your cluster in the same way you add them in other Amazon Web Services services that support tagging. To learn more about tagging Amazon Web Services resources in general, see Tagging Amazon Web Services Resources User Guide.
             orchestrator: The type of orchestrator to use for the SageMaker HyperPod cluster. Currently, the only supported value is "eks", which is to use an Amazon Elastic Kubernetes Service (EKS) cluster as the orchestrator.
@@ -3324,6 +3332,7 @@ class Cluster(Base):
         operation_input_args = {
             "ClusterName": cluster_name,
             "InstanceGroups": instance_groups,
+            "RestrictedInstanceGroups": restricted_instance_groups,
             "VpcConfig": vpc_config,
             "Tags": tags,
             "Orchestrator": orchestrator,
@@ -3438,6 +3447,9 @@ class Cluster(Base):
     def update(
         self,
         instance_groups: Optional[List[shapes.ClusterInstanceGroupSpecification]] = Unassigned(),
+        restricted_instance_groups: Optional[
+            List[shapes.ClusterRestrictedInstanceGroupSpecification]
+        ] = Unassigned(),
         node_recovery: Optional[str] = Unassigned(),
         instance_groups_to_delete: Optional[List[str]] = Unassigned(),
     ) -> Optional["Cluster"]:
@@ -3471,6 +3483,7 @@ class Cluster(Base):
         operation_input_args = {
             "ClusterName": self.cluster_name,
             "InstanceGroups": instance_groups,
+            "RestrictedInstanceGroups": restricted_instance_groups,
             "NodeRecovery": node_recovery,
             "InstanceGroupsToDelete": instance_groups_to_delete,
         }
@@ -3717,7 +3730,7 @@ class Cluster(Base):
     @Base.add_validate_call
     def get_node(
         self,
-        node_id: str,
+        node_id: Optional[str] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> Optional[shapes.ClusterNodeDetails]:
@@ -3879,7 +3892,7 @@ class Cluster(Base):
     @Base.add_validate_call
     def batch_delete_nodes(
         self,
-        node_ids: List[str],
+        node_ids: Optional[List[str]] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> Optional[shapes.BatchDeleteClusterNodesResponse]:
@@ -24538,6 +24551,8 @@ class Pipeline(Base):
         created_by:
         last_modified_by:
         parallelism_configuration: Lists the parallelism configuration applied to the pipeline.
+        pipeline_version_display_name: The display name of the pipeline version.
+        pipeline_version_description: The description of the pipeline version.
 
     """
 
@@ -24554,6 +24569,8 @@ class Pipeline(Base):
     created_by: Optional[shapes.UserContext] = Unassigned()
     last_modified_by: Optional[shapes.UserContext] = Unassigned()
     parallelism_configuration: Optional[shapes.ParallelismConfiguration] = Unassigned()
+    pipeline_version_display_name: Optional[str] = Unassigned()
+    pipeline_version_description: Optional[str] = Unassigned()
 
     def get_name(self) -> str:
         attributes = vars(self)
@@ -24677,6 +24694,7 @@ class Pipeline(Base):
     def get(
         cls,
         pipeline_name: str,
+        pipeline_version_id: Optional[int] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> Optional["Pipeline"]:
@@ -24685,6 +24703,7 @@ class Pipeline(Base):
 
         Parameters:
             pipeline_name: The name or Amazon Resource Name (ARN) of the pipeline to describe.
+            pipeline_version_id: The ID of the pipeline version to describe.
             session: Boto3 session.
             region: Region name.
 
@@ -24706,6 +24725,7 @@ class Pipeline(Base):
 
         operation_input_args = {
             "PipelineName": pipeline_name,
+            "PipelineVersionId": pipeline_version_id,
         }
         # serialize the input request
         operation_input_args = serialize(operation_input_args)
@@ -24726,6 +24746,7 @@ class Pipeline(Base):
     @Base.add_validate_call
     def refresh(
         self,
+        pipeline_version_id: Optional[int] = Unassigned(),
     ) -> Optional["Pipeline"]:
         """
         Refresh a Pipeline resource
@@ -24748,6 +24769,7 @@ class Pipeline(Base):
 
         operation_input_args = {
             "PipelineName": self.pipeline_name,
+            "PipelineVersionId": pipeline_version_id,
         }
         # serialize the input request
         operation_input_args = serialize(operation_input_args)
@@ -25053,6 +25075,7 @@ class PipelineExecution(Base):
         last_modified_by:
         parallelism_configuration: The parallelism configuration applied to the pipeline.
         selective_execution_config: The selective execution configuration applied to the pipeline run.
+        pipeline_version_id: The ID of the pipeline version.
 
     """
 
@@ -25069,6 +25092,7 @@ class PipelineExecution(Base):
     last_modified_by: Optional[shapes.UserContext] = Unassigned()
     parallelism_configuration: Optional[shapes.ParallelismConfiguration] = Unassigned()
     selective_execution_config: Optional[shapes.SelectiveExecutionConfig] = Unassigned()
+    pipeline_version_id: Optional[int] = Unassigned()
 
     def get_name(self) -> str:
         attributes = vars(self)

{sagemaker_core-1.0.42 → sagemaker_core-1.0.43}/src/sagemaker_core/main/shapes.py

@@ -1516,7 +1516,7 @@ class IamIdentity(Base):
 class UserContext(Base):
     """
     UserContext
-      Information about the user who created or modified an experiment, trial, trial component, lineage group, project, or model card.
+      Information about the user who created or modified a SageMaker resource.
 
     Attributes
     ----------------------
@@ -3552,6 +3552,121 @@ class ClusterOrchestrator(Base):
     eks: ClusterOrchestratorEksConfig
 
 
+class FSxLustreConfig(Base):
+    """
+    FSxLustreConfig
+      Configuration settings for an Amazon FSx for Lustre file system to be used with the cluster.
+
+    Attributes
+    ----------------------
+    size_in_gi_b: The storage capacity of the Amazon FSx for Lustre file system, specified in gibibytes (GiB).
+    per_unit_storage_throughput: The throughput capacity of the Amazon FSx for Lustre file system, measured in MB/s per TiB of storage.
+    """
+
+    size_in_gi_b: int
+    per_unit_storage_throughput: int
+
+
+class EnvironmentConfigDetails(Base):
+    """
+    EnvironmentConfigDetails
+      The configuration details for the restricted instance groups (RIG) environment.
+
+    Attributes
+    ----------------------
+    f_sx_lustre_config: Configuration settings for an Amazon FSx for Lustre file system to be used with the cluster.
+    s3_output_path: The Amazon S3 path where output data from the restricted instance group (RIG) environment will be stored.
+    """
+
+    f_sx_lustre_config: Optional[FSxLustreConfig] = Unassigned()
+    s3_output_path: Optional[str] = Unassigned()
+
+
+class ClusterRestrictedInstanceGroupDetails(Base):
+    """
+    ClusterRestrictedInstanceGroupDetails
+      The instance group details of the restricted instance group (RIG).
+
+    Attributes
+    ----------------------
+    current_count: The number of instances that are currently in the restricted instance group of a SageMaker HyperPod cluster.
+    target_count: The number of instances you specified to add to the restricted instance group of a SageMaker HyperPod cluster.
+    instance_group_name: The name of the restricted instance group of a SageMaker HyperPod cluster.
+    instance_type: The instance type of the restricted instance group of a SageMaker HyperPod cluster.
+    execution_role: The execution role for the restricted instance group to assume.
+    threads_per_core: The number you specified to TreadsPerCore in CreateCluster for enabling or disabling multithreading. For instance types that support multithreading, you can specify 1 for disabling multithreading and 2 for enabling multithreading. For more information, see the reference table of CPU cores and threads per CPU core per instance type in the Amazon Elastic Compute Cloud User Guide.
+    instance_storage_configs: The additional storage configurations for the instances in the SageMaker HyperPod cluster restricted instance group.
+    on_start_deep_health_checks: A flag indicating whether deep health checks should be performed when the cluster's restricted instance group is created or updated.
+    status: The current status of the cluster's restricted instance group.    InService: The restricted instance group is active and healthy.    Creating: The restricted instance group is being provisioned.    Updating: The restricted instance group is being updated.    Failed: The restricted instance group has failed to provision or is no longer healthy.    Degraded: The restricted instance group is degraded, meaning that some instances have failed to provision or are no longer healthy.    Deleting: The restricted instance group is being deleted.
+    training_plan_arn: The Amazon Resource Name (ARN) of the training plan to filter clusters by. For more information about reserving GPU capacity for your SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see  CreateTrainingPlan .
+    training_plan_status: The current status of the training plan associated with this cluster restricted instance group.
+    override_vpc_config
+    scheduled_update_config
+    environment_config: The configuration for the restricted instance groups (RIG) environment.
+    """
+
+    current_count: Optional[int] = Unassigned()
+    target_count: Optional[int] = Unassigned()
+    instance_group_name: Optional[str] = Unassigned()
+    instance_type: Optional[str] = Unassigned()
+    execution_role: Optional[str] = Unassigned()
+    threads_per_core: Optional[int] = Unassigned()
+    instance_storage_configs: Optional[List[ClusterInstanceStorageConfig]] = Unassigned()
+    on_start_deep_health_checks: Optional[List[str]] = Unassigned()
+    status: Optional[str] = Unassigned()
+    training_plan_arn: Optional[str] = Unassigned()
+    training_plan_status: Optional[str] = Unassigned()
+    override_vpc_config: Optional[VpcConfig] = Unassigned()
+    scheduled_update_config: Optional[ScheduledUpdateConfig] = Unassigned()
+    environment_config: Optional[EnvironmentConfigDetails] = Unassigned()
+
+
+class EnvironmentConfig(Base):
+    """
+    EnvironmentConfig
+      The configuration for the restricted instance groups (RIG) environment.
+
+    Attributes
+    ----------------------
+    f_sx_lustre_config: Configuration settings for an Amazon FSx for Lustre file system to be used with the cluster.
+    """
+
+    f_sx_lustre_config: Optional[FSxLustreConfig] = Unassigned()
+
+
+class ClusterRestrictedInstanceGroupSpecification(Base):
+    """
+    ClusterRestrictedInstanceGroupSpecification
+      The specifications of a restricted instance group that you need to define.
+
+    Attributes
+    ----------------------
+    instance_count: Specifies the number of instances to add to the restricted instance group of a SageMaker HyperPod cluster.
+    instance_group_name: Specifies the name of the restricted instance group.
+    instance_type: Specifies the instance type of the restricted instance group.
+    execution_role: Specifies an IAM execution role to be assumed by the restricted instance group.
+    threads_per_core: The number you specified to TreadsPerCore in CreateCluster for enabling or disabling multithreading. For instance types that support multithreading, you can specify 1 for disabling multithreading and 2 for enabling multithreading. For more information, see the reference table of CPU cores and threads per CPU core per instance type in the Amazon Elastic Compute Cloud User Guide.
+    instance_storage_configs: Specifies the additional storage configurations for the instances in the SageMaker HyperPod cluster restricted instance group.
+    on_start_deep_health_checks: A flag indicating whether deep health checks should be performed when the cluster restricted instance group is created or updated.
+    training_plan_arn: The Amazon Resource Name (ARN) of the training plan to filter clusters by. For more information about reserving GPU capacity for your SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see  CreateTrainingPlan .
+    override_vpc_config
+    scheduled_update_config
+    environment_config: The configuration for the restricted instance groups (RIG) environment.
+    """
+
+    instance_count: int
+    instance_group_name: str
+    instance_type: str
+    execution_role: str
+    environment_config: EnvironmentConfig
+    threads_per_core: Optional[int] = Unassigned()
+    instance_storage_configs: Optional[List[ClusterInstanceStorageConfig]] = Unassigned()
+    on_start_deep_health_checks: Optional[List[str]] = Unassigned()
+    training_plan_arn: Optional[str] = Unassigned()
+    override_vpc_config: Optional[VpcConfig] = Unassigned()
+    scheduled_update_config: Optional[ScheduledUpdateConfig] = Unassigned()
+
+
 class ClusterSchedulerConfigSummary(Base):
     """
     ClusterSchedulerConfigSummary
@@ -11500,6 +11615,29 @@ class Parameter(Base):
     value: str
 
 
+class PipelineVersionSummary(Base):
+    """
+    PipelineVersionSummary
+      The summary of the pipeline version.
+
+    Attributes
+    ----------------------
+    pipeline_arn: The Amazon Resource Name (ARN) of the pipeline.
+    pipeline_version_id: The ID of the pipeline version.
+    creation_time: The creation time of the pipeline version.
+    pipeline_version_description: The description of the pipeline version.
+    pipeline_version_display_name: The display name of the pipeline version.
+    last_execution_pipeline_execution_arn: The Amazon Resource Name (ARN) of the most recent pipeline execution created from this pipeline version.
+    """
+
+    pipeline_arn: Optional[str] = Unassigned()
+    pipeline_version_id: Optional[int] = Unassigned()
+    creation_time: Optional[datetime.datetime] = Unassigned()
+    pipeline_version_description: Optional[str] = Unassigned()
+    pipeline_version_display_name: Optional[str] = Unassigned()
+    last_execution_pipeline_execution_arn: Optional[str] = Unassigned()
+
+
 class PipelineSummary(Base):
     """
     PipelineSummary
@@ -12316,6 +12454,8 @@ class PipelineExecution(Base):
     parallelism_configuration: The parallelism configuration applied to the pipeline execution.
     selective_execution_config: The selective execution configuration applied to the pipeline run.
     pipeline_parameters: Contains a list of pipeline parameters. This list can be empty.
+    pipeline_version_id: The ID of the pipeline version that started this execution.
+    pipeline_version_display_name: The display name of the pipeline version that started this execution.
     """
 
     pipeline_arn: Optional[str] = Unassigned()
@@ -12332,6 +12472,41 @@ class PipelineExecution(Base):
     parallelism_configuration: Optional[ParallelismConfiguration] = Unassigned()
     selective_execution_config: Optional[SelectiveExecutionConfig] = Unassigned()
     pipeline_parameters: Optional[List[Parameter]] = Unassigned()
+    pipeline_version_id: Optional[int] = Unassigned()
+    pipeline_version_display_name: Optional[str] = Unassigned()
+
+
+class PipelineVersion(Base):
+    """
+    PipelineVersion
+      The version of the pipeline.
+
+    Attributes
+    ----------------------
+    pipeline_arn: The Amazon Resource Name (ARN) of the pipeline.
+    pipeline_version_id: The ID of the pipeline version.
+    pipeline_version_display_name: The display name of the pipeline version.
+    pipeline_version_description: The description of the pipeline version.
+    creation_time: The creation time of the pipeline version.
+    last_modified_time: The time when the pipeline version was last modified.
+    created_by
+    last_modified_by
+    last_executed_pipeline_execution_arn: The Amazon Resource Name (ARN) of the most recent pipeline execution created from this pipeline version.
+    last_executed_pipeline_execution_display_name: The display name of the most recent pipeline execution created from this pipeline version.
+    last_executed_pipeline_execution_status: The status of the most recent pipeline execution created from this pipeline version.
+    """
+
+    pipeline_arn: Optional[str] = Unassigned()
+    pipeline_version_id: Optional[int] = Unassigned()
+    pipeline_version_display_name: Optional[str] = Unassigned()
+    pipeline_version_description: Optional[str] = Unassigned()
+    creation_time: Optional[datetime.datetime] = Unassigned()
+    last_modified_time: Optional[datetime.datetime] = Unassigned()
+    created_by: Optional[UserContext] = Unassigned()
+    last_modified_by: Optional[UserContext] = Unassigned()
+    last_executed_pipeline_execution_arn: Optional[str] = Unassigned()
+    last_executed_pipeline_execution_display_name: Optional[str] = Unassigned()
+    last_executed_pipeline_execution_status: Optional[str] = Unassigned()
 
 
 class ProcessingJob(Base):
@@ -12844,6 +13019,7 @@ class SearchRecord(Base):
     model_package_group
     pipeline
     pipeline_execution
+    pipeline_version: The version of the pipeline.
     feature_group
     feature_metadata: The feature metadata used to search through the features.
     project: The properties of a project.
@@ -12861,6 +13037,7 @@ class SearchRecord(Base):
     model_package_group: Optional[ModelPackageGroup] = Unassigned()
     pipeline: Optional[Pipeline] = Unassigned()
     pipeline_execution: Optional[PipelineExecution] = Unassigned()
+    pipeline_version: Optional[PipelineVersion] = Unassigned()
     feature_group: Optional[FeatureGroup] = Unassigned()
     feature_metadata: Optional[FeatureMetadata] = Unassigned()
     project: Optional[Project] = Unassigned()

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.42
+Version: 1.0.43
 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.42/VERSION

@@ -1 +0,0 @@
-1.0.42