mypy-boto3-sagemaker 1.40.2__py3-none-any.whl → 1.40.3__py3-none-any.whl

mypy_boto3_sagemaker/type_defs.py

@@ -55,18 +55,21 @@ from .literals import (
     AutoMLSortOrderType,
     AutoMountHomeEFSType,
     AwsManagedHumanLoopRequestSourceType,
+    BatchAddClusterNodesErrorCodeType,
     BatchDeleteClusterNodesErrorCodeType,
     BatchStrategyType,
     BooleanOperatorType,
     CandidateSortByType,
     CandidateStatusType,
     CandidateStepTypeType,
+    CapacityReservationTypeType,
     CapacitySizeTypeType,
     CaptureModeType,
     CaptureStatusType,
     ClarifyFeatureTypeType,
     ClarifyTextGranularityType,
     ClarifyTextLanguageType,
+    ClusterEventResourceTypeType,
     ClusterInstanceStatusType,
     ClusterInstanceTypeType,
     ClusterNodeRecoveryType,
@@ -331,8 +334,10 @@ __all__ = (
     "ActionSummaryTypeDef",
     "AddAssociationRequestTypeDef",
     "AddAssociationResponseTypeDef",
+    "AddClusterNodeSpecificationTypeDef",
     "AddTagsInputTypeDef",
     "AddTagsOutputTypeDef",
+    "AdditionalEnisTypeDef",
     "AdditionalInferenceSpecificationDefinitionOutputTypeDef",
     "AdditionalInferenceSpecificationDefinitionTypeDef",
     "AdditionalInferenceSpecificationDefinitionUnionTypeDef",
@@ -415,7 +420,11 @@ __all__ = (
     "AutoRollbackConfigOutputTypeDef",
     "AutoRollbackConfigTypeDef",
     "AutotuneTypeDef",
+    "BatchAddClusterNodesErrorTypeDef",
+    "BatchAddClusterNodesRequestTypeDef",
+    "BatchAddClusterNodesResponseTypeDef",
     "BatchDataCaptureConfigTypeDef",
+    "BatchDeleteClusterNodeLogicalIdsErrorTypeDef",
     "BatchDeleteClusterNodesErrorTypeDef",
     "BatchDeleteClusterNodesRequestTypeDef",
     "BatchDeleteClusterNodesResponseTypeDef",
@@ -436,6 +445,7 @@ __all__ = (
     "CandidatePropertiesTypeDef",
     "CanvasAppSettingsOutputTypeDef",
     "CanvasAppSettingsTypeDef",
+    "CapacityReservationTypeDef",
     "CapacitySizeConfigTypeDef",
     "CapacitySizeTypeDef",
     "CaptureContentTypeHeaderOutputTypeDef",
@@ -470,12 +480,15 @@ __all__ = (
     "ClarifyShapConfigTypeDef",
     "ClarifyTextConfigTypeDef",
     "ClusterEbsVolumeConfigTypeDef",
+    "ClusterEventDetailTypeDef",
+    "ClusterEventSummaryTypeDef",
     "ClusterInstanceGroupDetailsTypeDef",
     "ClusterInstanceGroupSpecificationTypeDef",
     "ClusterInstancePlacementTypeDef",
     "ClusterInstanceStatusDetailsTypeDef",
     "ClusterInstanceStorageConfigTypeDef",
     "ClusterLifeCycleConfigTypeDef",
+    "ClusterMetadataTypeDef",
     "ClusterNodeDetailsTypeDef",
     "ClusterNodeSummaryTypeDef",
     "ClusterOrchestratorEksConfigTypeDef",
@@ -773,6 +786,8 @@ __all__ = (
     "DescribeAutoMLJobResponseTypeDef",
     "DescribeAutoMLJobV2RequestTypeDef",
     "DescribeAutoMLJobV2ResponseTypeDef",
+    "DescribeClusterEventRequestTypeDef",
+    "DescribeClusterEventResponseTypeDef",
     "DescribeClusterNodeRequestTypeDef",
     "DescribeClusterNodeResponseTypeDef",
     "DescribeClusterRequestTypeDef",
@@ -973,6 +988,8 @@ __all__ = (
     "EnvironmentParameterRangesTypeDef",
     "EnvironmentParameterTypeDef",
     "ErrorInfoTypeDef",
+    "EventDetailsTypeDef",
+    "EventMetadataTypeDef",
     "ExperimentConfigTypeDef",
     "ExperimentSourceTypeDef",
     "ExperimentSummaryTypeDef",
@@ -1098,8 +1115,11 @@ __all__ = (
     "InferenceSpecificationUnionTypeDef",
     "InfraCheckConfigTypeDef",
     "InputConfigTypeDef",
+    "InstanceGroupMetadataTypeDef",
+    "InstanceGroupScalingMetadataTypeDef",
     "InstanceGroupTypeDef",
     "InstanceMetadataServiceConfigurationTypeDef",
+    "InstanceMetadataTypeDef",
     "IntegerParameterRangeSpecificationTypeDef",
     "IntegerParameterRangeTypeDef",
     "JupyterLabAppImageConfigOutputTypeDef",
@@ -1166,6 +1186,9 @@ __all__ = (
     "ListCandidatesForAutoMLJobRequestPaginateTypeDef",
     "ListCandidatesForAutoMLJobRequestTypeDef",
     "ListCandidatesForAutoMLJobResponseTypeDef",
+    "ListClusterEventsRequestPaginateTypeDef",
+    "ListClusterEventsRequestTypeDef",
+    "ListClusterEventsResponseTypeDef",
     "ListClusterNodesRequestPaginateTypeDef",
     "ListClusterNodesRequestTypeDef",
     "ListClusterNodesResponseTypeDef",
@@ -1515,6 +1538,7 @@ __all__ = (
     "NetworkConfigOutputTypeDef",
     "NetworkConfigTypeDef",
     "NetworkConfigUnionTypeDef",
+    "NodeAdditionResultTypeDef",
     "NotebookInstanceLifecycleConfigSummaryTypeDef",
     "NotebookInstanceLifecycleHookTypeDef",
     "NotebookInstanceSummaryTypeDef",
@@ -1980,11 +2004,20 @@ class ResponseMetadataTypeDef(TypedDict):
     HostId: NotRequired[str]
 
 
+class AddClusterNodeSpecificationTypeDef(TypedDict):
+    InstanceGroupName: str
+    IncrementTargetCountBy: int
+
+
 class TagTypeDef(TypedDict):
     Key: str
     Value: str
 
 
+class AdditionalEnisTypeDef(TypedDict):
+    EfaEnis: NotRequired[List[str]]
+
+
 class AdditionalS3DataSourceTypeDef(TypedDict):
     S3DataType: AdditionalS3DataSourceDataTypeType
     S3Uri: str
@@ -2207,12 +2240,31 @@ class AutotuneTypeDef(TypedDict):
     Mode: Literal["Enabled"]
 
 
+class BatchAddClusterNodesErrorTypeDef(TypedDict):
+    InstanceGroupName: str
+    ErrorCode: BatchAddClusterNodesErrorCodeType
+    FailedCount: int
+    Message: NotRequired[str]
+
+
+class NodeAdditionResultTypeDef(TypedDict):
+    NodeLogicalId: str
+    InstanceGroupName: str
+    Status: ClusterInstanceStatusType
+
+
 class BatchDataCaptureConfigTypeDef(TypedDict):
     DestinationS3Uri: str
     KmsKeyId: NotRequired[str]
     GenerateInferenceId: NotRequired[bool]
 
 
+class BatchDeleteClusterNodeLogicalIdsErrorTypeDef(TypedDict):
+    Code: BatchDeleteClusterNodesErrorCodeType
+    Message: str
+    NodeLogicalId: str
+
+
 class BatchDeleteClusterNodesErrorTypeDef(TypedDict):
     Code: BatchDeleteClusterNodesErrorCodeType
     Message: str
@@ -2222,6 +2274,7 @@ class BatchDeleteClusterNodesErrorTypeDef(TypedDict):
 class BatchDeleteClusterNodesRequestTypeDef(TypedDict):
     ClusterName: str
     NodeIds: NotRequired[Sequence[str]]
+    NodeLogicalIds: NotRequired[Sequence[str]]
 
 
 class BatchDescribeModelPackageErrorTypeDef(TypedDict):
@@ -2307,6 +2360,13 @@ class WorkspaceSettingsTypeDef(TypedDict):
     S3KmsKeyId: NotRequired[str]
 
 
+CapacityReservationTypeDef = TypedDict(
+    "CapacityReservationTypeDef",
+    {
+        "Arn": NotRequired[str],
+        "Type": NotRequired[CapacityReservationTypeType],
+    },
+)
 CapacitySizeConfigTypeDef = TypedDict(
     "CapacitySizeConfigTypeDef",
     {
@@ -2467,6 +2527,17 @@ class ClusterEbsVolumeConfigTypeDef(TypedDict):
     VolumeSizeInGB: NotRequired[int]
 
 
+class ClusterEventSummaryTypeDef(TypedDict):
+    EventId: str
+    ClusterArn: str
+    ClusterName: str
+    ResourceType: ClusterEventResourceTypeType
+    EventTime: datetime
+    InstanceGroupName: NotRequired[str]
+    InstanceId: NotRequired[str]
+    Description: NotRequired[str]
+
+
 class ClusterLifeCycleConfigTypeDef(TypedDict):
     SourceS3Uri: str
     OnCreate: str
@@ -2482,6 +2553,12 @@ class ClusterInstanceStatusDetailsTypeDef(TypedDict):
     Message: NotRequired[str]
 
 
+class ClusterMetadataTypeDef(TypedDict):
+    FailureMessage: NotRequired[str]
+    EksRoleAccessEntries: NotRequired[List[str]]
+    SlrAccessEntry: NotRequired[str]
+
+
 class ClusterOrchestratorEksConfigTypeDef(TypedDict):
     ClusterArn: str
 
@@ -3378,9 +3455,15 @@ class DescribeAutoMLJobV2RequestTypeDef(TypedDict):
     AutoMLJobName: str
 
 
+class DescribeClusterEventRequestTypeDef(TypedDict):
+    EventId: str
+    ClusterName: str
+
+
 class DescribeClusterNodeRequestTypeDef(TypedDict):
     ClusterName: str
     NodeId: NotRequired[str]
+    NodeLogicalId: NotRequired[str]
 
 
 class DescribeClusterRequestTypeDef(TypedDict):
@@ -4125,6 +4208,12 @@ class EnvironmentParameterTypeDef(TypedDict):
     Value: str
 
 
+class InstanceGroupScalingMetadataTypeDef(TypedDict):
+    InstanceCount: NotRequired[int]
+    TargetCount: NotRequired[int]
+    FailureMessage: NotRequired[str]
+
+
 class FailStepMetadataTypeDef(TypedDict):
     ErrorMessage: NotRequired[str]
 
@@ -6278,6 +6367,12 @@ class UpdateUserProfileResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
+class BatchAddClusterNodesRequestTypeDef(TypedDict):
+    ClusterName: str
+    NodesToAdd: Sequence[AddClusterNodeSpecificationTypeDef]
+    ClientToken: NotRequired[str]
+
+
 class AddTagsInputTypeDef(TypedDict):
     ResourceArn: str
     Tags: Sequence[TagTypeDef]
@@ -6598,9 +6693,17 @@ class NetworkConfigTypeDef(TypedDict):
 VpcConfigUnionTypeDef = Union[VpcConfigTypeDef, VpcConfigOutputTypeDef]
 
 
+class BatchAddClusterNodesResponseTypeDef(TypedDict):
+    Successful: List[NodeAdditionResultTypeDef]
+    Failed: List[BatchAddClusterNodesErrorTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class BatchDeleteClusterNodesResponseTypeDef(TypedDict):
     Failed: List[BatchDeleteClusterNodesErrorTypeDef]
     Successful: List[str]
+    FailedNodeLogicalIds: List[BatchDeleteClusterNodeLogicalIdsErrorTypeDef]
+    SuccessfulNodeLogicalIds: List[str]
     ResponseMetadata: ResponseMetadataTypeDef
 
 
@@ -6678,6 +6781,24 @@ class CanvasAppSettingsTypeDef(TypedDict):
     EmrServerlessSettings: NotRequired[EmrServerlessSettingsTypeDef]
 
 
+class InstanceGroupMetadataTypeDef(TypedDict):
+    FailureMessage: NotRequired[str]
+    AvailabilityZoneId: NotRequired[str]
+    CapacityReservation: NotRequired[CapacityReservationTypeDef]
+    SubnetId: NotRequired[str]
+    SecurityGroupIds: NotRequired[List[str]]
+    AmiOverride: NotRequired[str]
+
+
+class InstanceMetadataTypeDef(TypedDict):
+    CustomerEni: NotRequired[str]
+    AdditionalEnis: NotRequired[AdditionalEnisTypeDef]
+    CapacityReservation: NotRequired[CapacityReservationTypeDef]
+    FailureMessage: NotRequired[str]
+    LcsExecutionState: NotRequired[str]
+    NodeLogicalId: NotRequired[str]
+
+
 class RollingDeploymentPolicyTypeDef(TypedDict):
     MaximumBatchSize: CapacitySizeConfigTypeDef
     RollbackMaximumBatchSize: NotRequired[CapacitySizeConfigTypeDef]
@@ -6777,12 +6898,19 @@ class ClusterInstanceStorageConfigTypeDef(TypedDict):
     EbsVolumeConfig: NotRequired[ClusterEbsVolumeConfigTypeDef]
 
 
+class ListClusterEventsResponseTypeDef(TypedDict):
+    Events: List[ClusterEventSummaryTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+    NextToken: NotRequired[str]
+
+
 class ClusterNodeSummaryTypeDef(TypedDict):
     InstanceGroupName: str
     InstanceId: str
     InstanceType: ClusterInstanceTypeType
     LaunchTime: datetime
     InstanceStatus: ClusterInstanceStatusDetailsTypeDef
+    NodeLogicalId: NotRequired[str]
     LastSoftwareUpdateTime: NotRequired[datetime]
 
 
@@ -7450,6 +7578,31 @@ class ListAutoMLJobsRequestTypeDef(TypedDict):
     NextToken: NotRequired[str]
 
 
+class ListClusterEventsRequestPaginateTypeDef(TypedDict):
+    ClusterName: str
+    InstanceGroupName: NotRequired[str]
+    NodeId: NotRequired[str]
+    EventTimeAfter: NotRequired[TimestampTypeDef]
+    EventTimeBefore: NotRequired[TimestampTypeDef]
+    SortBy: NotRequired[Literal["EventTime"]]
+    SortOrder: NotRequired[SortOrderType]
+    ResourceType: NotRequired[ClusterEventResourceTypeType]
+    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
+
+
+class ListClusterEventsRequestTypeDef(TypedDict):
+    ClusterName: str
+    InstanceGroupName: NotRequired[str]
+    NodeId: NotRequired[str]
+    EventTimeAfter: NotRequired[TimestampTypeDef]
+    EventTimeBefore: NotRequired[TimestampTypeDef]
+    SortBy: NotRequired[Literal["EventTime"]]
+    SortOrder: NotRequired[SortOrderType]
+    ResourceType: NotRequired[ClusterEventResourceTypeType]
+    MaxResults: NotRequired[int]
+    NextToken: NotRequired[str]
+
+
 class ListClusterNodesRequestPaginateTypeDef(TypedDict):
     ClusterName: str
     CreationTimeAfter: NotRequired[TimestampTypeDef]
@@ -7457,6 +7610,7 @@ class ListClusterNodesRequestPaginateTypeDef(TypedDict):
     InstanceGroupNameContains: NotRequired[str]
     SortBy: NotRequired[ClusterSortByType]
     SortOrder: NotRequired[SortOrderType]
+    IncludeNodeLogicalIds: NotRequired[bool]
     PaginationConfig: NotRequired[PaginatorConfigTypeDef]
 
 
@@ -7469,6 +7623,7 @@ class ListClusterNodesRequestTypeDef(TypedDict):
     NextToken: NotRequired[str]
     SortBy: NotRequired[ClusterSortByType]
     SortOrder: NotRequired[SortOrderType]
+    IncludeNodeLogicalIds: NotRequired[bool]
 
 
 class ListClusterSchedulerConfigsRequestPaginateTypeDef(TypedDict):
@@ -10404,6 +10559,13 @@ class AutoMLCandidateTypeDef(TypedDict):
     ]
 
 
+class EventMetadataTypeDef(TypedDict):
+    Cluster: NotRequired[ClusterMetadataTypeDef]
+    InstanceGroup: NotRequired[InstanceGroupMetadataTypeDef]
+    InstanceGroupScaling: NotRequired[InstanceGroupScalingMetadataTypeDef]
+    Instance: NotRequired[InstanceMetadataTypeDef]
+
+
 class DeploymentConfigurationOutputTypeDef(TypedDict):
     RollingUpdatePolicy: NotRequired[RollingDeploymentPolicyTypeDef]
     WaitIntervalInSeconds: NotRequired[int]
@@ -10476,6 +10638,7 @@ class ClarifyExplainerConfigTypeDef(TypedDict):
 class ClusterNodeDetailsTypeDef(TypedDict):
     InstanceGroupName: NotRequired[str]
     InstanceId: NotRequired[str]
+    NodeLogicalId: NotRequired[str]
     InstanceStatus: NotRequired[ClusterInstanceStatusDetailsTypeDef]
     InstanceType: NotRequired[ClusterInstanceTypeType]
     LaunchTime: NotRequired[datetime]
@@ -10488,6 +10651,8 @@ class ClusterNodeDetailsTypeDef(TypedDict):
     PrivatePrimaryIpv6: NotRequired[str]
     PrivateDnsHostname: NotRequired[str]
     Placement: NotRequired[ClusterInstancePlacementTypeDef]
+    CurrentImageId: NotRequired[str]
+    DesiredImageId: NotRequired[str]
 
 
 class ListClusterNodesResponseTypeDef(TypedDict):
@@ -11740,6 +11905,10 @@ class ListCandidatesForAutoMLJobResponseTypeDef(TypedDict):
     NextToken: NotRequired[str]
 
 
+class EventDetailsTypeDef(TypedDict):
+    EventMetadata: NotRequired[EventMetadataTypeDef]
+
+
 class ScheduledUpdateConfigOutputTypeDef(TypedDict):
     ScheduleExpression: str
     DeploymentConfig: NotRequired[DeploymentConfigurationOutputTypeDef]
@@ -12712,6 +12881,18 @@ class ListPipelineExecutionStepsResponseTypeDef(TypedDict):
     NextToken: NotRequired[str]
 
 
+class ClusterEventDetailTypeDef(TypedDict):
+    EventId: str
+    ClusterArn: str
+    ClusterName: str
+    ResourceType: ClusterEventResourceTypeType
+    EventTime: datetime
+    InstanceGroupName: NotRequired[str]
+    InstanceId: NotRequired[str]
+    EventDetails: NotRequired[EventDetailsTypeDef]
+    Description: NotRequired[str]
+
+
 class ClusterInstanceGroupDetailsTypeDef(TypedDict):
     CurrentCount: NotRequired[int]
     TargetCount: NotRequired[int]
@@ -12727,6 +12908,8 @@ class ClusterInstanceGroupDetailsTypeDef(TypedDict):
     TrainingPlanStatus: NotRequired[str]
     OverrideVpcConfig: NotRequired[VpcConfigOutputTypeDef]
     ScheduledUpdateConfig: NotRequired[ScheduledUpdateConfigOutputTypeDef]
+    CurrentImageId: NotRequired[str]
+    DesiredImageId: NotRequired[str]
 
 
 class ClusterRestrictedInstanceGroupDetailsTypeDef(TypedDict):
@@ -12755,6 +12938,7 @@ class UpdateClusterSoftwareRequestTypeDef(TypedDict):
     ClusterName: str
     InstanceGroups: NotRequired[Sequence[UpdateClusterSoftwareInstanceGroupSpecificationTypeDef]]
     DeploymentConfig: NotRequired[DeploymentConfigurationUnionTypeDef]
+    ImageId: NotRequired[str]
 
 
 DeploymentConfigUnionTypeDef = Union[DeploymentConfigTypeDef, DeploymentConfigOutputTypeDef]
@@ -13322,6 +13506,11 @@ class CreateAutoMLJobV2RequestTypeDef(TypedDict):
     AutoMLComputeConfig: NotRequired[AutoMLComputeConfigTypeDef]
 
 
+class DescribeClusterEventResponseTypeDef(TypedDict):
+    EventDetails: ClusterEventDetailTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class DescribeClusterResponseTypeDef(TypedDict):
     ClusterArn: str
     ClusterName: str
@@ -13333,6 +13522,7 @@ class DescribeClusterResponseTypeDef(TypedDict):
     VpcConfig: VpcConfigOutputTypeDef
     Orchestrator: ClusterOrchestratorTypeDef
     NodeRecovery: ClusterNodeRecoveryType
+    NodeProvisioningMode: Literal["Continuous"]
     ResponseMetadata: ResponseMetadataTypeDef
 
 
@@ -13612,6 +13802,7 @@ class ClusterInstanceGroupSpecificationTypeDef(TypedDict):
     TrainingPlanArn: NotRequired[str]
     OverrideVpcConfig: NotRequired[VpcConfigUnionTypeDef]
     ScheduledUpdateConfig: NotRequired[ScheduledUpdateConfigUnionTypeDef]
+    ImageId: NotRequired[str]
 
 
 class ClusterRestrictedInstanceGroupSpecificationTypeDef(TypedDict):
@@ -13814,6 +14005,7 @@ class CreateClusterRequestTypeDef(TypedDict):
     Tags: NotRequired[Sequence[TagTypeDef]]
     Orchestrator: NotRequired[ClusterOrchestratorTypeDef]
     NodeRecovery: NotRequired[ClusterNodeRecoveryType]
+    NodeProvisioningMode: NotRequired[Literal["Continuous"]]
 
 
 class UpdateClusterRequestTypeDef(TypedDict):