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

mypy_boto3_sagemaker/type_defs.py

@@ -236,6 +236,7 @@ from .literals import (
     RepositoryAccessModeType,
     ReservedCapacityInstanceTypeType,
     ReservedCapacityStatusType,
+    ReservedCapacityTypeType,
     ResourceCatalogSortOrderType,
     ResourceSharingStrategyType,
     ResourceTypeType,
@@ -306,6 +307,7 @@ from .literals import (
     TransformJobStatusType,
     TrialComponentPrimaryStatusType,
     TtlDurationUnitType,
+    UltraServerHealthStatusType,
     UserProfileSortKeyType,
     UserProfileStatusType,
     VariantPropertyTypeType,
@@ -897,6 +899,8 @@ __all__ = (
     "DescribeProcessingJobResponseTypeDef",
     "DescribeProjectInputTypeDef",
     "DescribeProjectOutputTypeDef",
+    "DescribeReservedCapacityRequestTypeDef",
+    "DescribeReservedCapacityResponseTypeDef",
     "DescribeSpaceRequestTypeDef",
     "DescribeSpaceResponseTypeDef",
     "DescribeStudioLifecycleConfigRequestTypeDef",
@@ -1120,6 +1124,9 @@ __all__ = (
     "InstanceGroupTypeDef",
     "InstanceMetadataServiceConfigurationTypeDef",
     "InstanceMetadataTypeDef",
+    "InstancePlacementConfigOutputTypeDef",
+    "InstancePlacementConfigTypeDef",
+    "InstancePlacementConfigUnionTypeDef",
     "IntegerParameterRangeSpecificationTypeDef",
     "IntegerParameterRangeTypeDef",
     "JupyterLabAppImageConfigOutputTypeDef",
@@ -1395,6 +1402,9 @@ __all__ = (
     "ListTrialsRequestPaginateTypeDef",
     "ListTrialsRequestTypeDef",
     "ListTrialsResponseTypeDef",
+    "ListUltraServersByReservedCapacityRequestPaginateTypeDef",
+    "ListUltraServersByReservedCapacityRequestTypeDef",
+    "ListUltraServersByReservedCapacityResponseTypeDef",
     "ListUserProfilesRequestPaginateTypeDef",
     "ListUserProfilesRequestTypeDef",
     "ListUserProfilesResponseTypeDef",
@@ -1599,6 +1609,7 @@ __all__ = (
     "PipelineTypeDef",
     "PipelineVersionSummaryTypeDef",
     "PipelineVersionTypeDef",
+    "PlacementSpecificationTypeDef",
     "PredefinedMetricSpecificationTypeDef",
     "PresignedUrlAccessConfigTypeDef",
     "PriorityClassTypeDef",
@@ -1868,6 +1879,7 @@ __all__ = (
     "TrialSourceTypeDef",
     "TrialSummaryTypeDef",
     "TrialTypeDef",
+    "TrustedIdentityPropagationSettingsTypeDef",
     "TtlDurationTypeDef",
     "TuningJobCompletionCriteriaTypeDef",
     "TuningJobStepMetaDataTypeDef",
@@ -1875,6 +1887,9 @@ __all__ = (
     "UiConfigTypeDef",
     "UiTemplateInfoTypeDef",
     "UiTemplateTypeDef",
+    "UltraServerInfoTypeDef",
+    "UltraServerSummaryTypeDef",
+    "UltraServerTypeDef",
     "UnifiedStudioSettingsTypeDef",
     "UpdateActionRequestTypeDef",
     "UpdateActionResponseTypeDef",
@@ -2559,6 +2574,10 @@ class ClusterMetadataTypeDef(TypedDict):
     SlrAccessEntry: NotRequired[str]
 
 
+class UltraServerInfoTypeDef(TypedDict):
+    Id: NotRequired[str]
+
+
 class ClusterOrchestratorEksConfigTypeDef(TypedDict):
     ClusterArn: str
 
@@ -3866,6 +3885,18 @@ class ServiceCatalogProvisionedProductDetailsTypeDef(TypedDict):
     ProvisionedProductStatusMessage: NotRequired[str]
 
 
+class DescribeReservedCapacityRequestTypeDef(TypedDict):
+    ReservedCapacityArn: str
+
+
+class UltraServerSummaryTypeDef(TypedDict):
+    UltraServerType: str
+    InstanceType: ReservedCapacityInstanceTypeType
+    UltraServerCount: NotRequired[int]
+    AvailableSpareInstanceCount: NotRequired[int]
+    UnhealthyInstanceCount: NotRequired[int]
+
+
 class DescribeSpaceRequestTypeDef(TypedDict):
     DomainId: str
     SpaceName: str
@@ -3944,6 +3975,9 @@ class ReservedCapacitySummaryTypeDef(TypedDict):
     InstanceType: ReservedCapacityInstanceTypeType
     TotalInstanceCount: int
     Status: ReservedCapacityStatusType
+    ReservedCapacityType: NotRequired[ReservedCapacityTypeType]
+    UltraServerType: NotRequired[str]
+    UltraServerCount: NotRequired[int]
     AvailabilityZone: NotRequired[str]
     DurationHours: NotRequired[int]
     DurationMinutes: NotRequired[int]
@@ -4078,6 +4112,10 @@ class DomainDetailsTypeDef(TypedDict):
     Url: NotRequired[str]
 
 
+class TrustedIdentityPropagationSettingsTypeDef(TypedDict):
+    Status: FeatureStatusType
+
+
 class UnifiedStudioSettingsTypeDef(TypedDict):
     StudioWebPortalAccess: NotRequired[FeatureStatusType]
     DomainAccountId: NotRequired[str]
@@ -4472,6 +4510,11 @@ class InstanceGroupTypeDef(TypedDict):
     InstanceGroupName: str
 
 
+class PlacementSpecificationTypeDef(TypedDict):
+    InstanceCount: int
+    UltraServerId: NotRequired[str]
+
+
 class IntegerParameterRangeSpecificationTypeDef(TypedDict):
     MinValue: str
     MaxValue: str
@@ -4834,6 +4877,26 @@ class TransformJobSummaryTypeDef(TypedDict):
     FailureReason: NotRequired[str]
 
 
+class ListUltraServersByReservedCapacityRequestTypeDef(TypedDict):
+    ReservedCapacityArn: str
+    MaxResults: NotRequired[int]
+    NextToken: NotRequired[str]
+
+
+class UltraServerTypeDef(TypedDict):
+    UltraServerId: str
+    UltraServerType: str
+    AvailabilityZone: str
+    InstanceType: ReservedCapacityInstanceTypeType
+    TotalInstanceCount: int
+    ConfiguredSpareInstanceCount: NotRequired[int]
+    AvailableInstanceCount: NotRequired[int]
+    InUseInstanceCount: NotRequired[int]
+    AvailableSpareInstanceCount: NotRequired[int]
+    UnhealthyInstanceCount: NotRequired[int]
+    HealthStatus: NotRequired[UltraServerHealthStatusType]
+
+
 class ListUserProfilesRequestTypeDef(TypedDict):
     NextToken: NotRequired[str]
     MaxResults: NotRequired[int]
@@ -5292,6 +5355,9 @@ class RenderingErrorTypeDef(TypedDict):
 class ReservedCapacityOfferingTypeDef(TypedDict):
     InstanceType: ReservedCapacityInstanceTypeType
     InstanceCount: int
+    ReservedCapacityType: NotRequired[ReservedCapacityTypeType]
+    UltraServerType: NotRequired[str]
+    UltraServerCount: NotRequired[int]
     AvailabilityZone: NotRequired[str]
     DurationHours: NotRequired[int]
     DurationMinutes: NotRequired[int]
@@ -6433,6 +6499,7 @@ class CreateStudioLifecycleConfigRequestTypeDef(TypedDict):
 class CreateTrainingPlanRequestTypeDef(TypedDict):
     TrainingPlanName: str
     TrainingPlanOfferingId: str
+    SpareInstanceCountPerUltraServer: NotRequired[int]
     Tags: NotRequired[Sequence[TagTypeDef]]
 
 
@@ -6520,6 +6587,7 @@ class DescribeAppResponseTypeDef(TypedDict):
     UserProfileName: str
     SpaceName: str
     Status: AppStatusType
+    EffectiveTrustedIdentityPropagationStatus: FeatureStatusType
     RecoveryMode: bool
     LastHealthCheckTimestamp: datetime
     LastUserActivityTimestamp: datetime
@@ -6912,6 +6980,7 @@ class ClusterNodeSummaryTypeDef(TypedDict):
     InstanceStatus: ClusterInstanceStatusDetailsTypeDef
     NodeLogicalId: NotRequired[str]
     LastSoftwareUpdateTime: NotRequired[datetime]
+    UltraServerInfo: NotRequired[UltraServerInfoTypeDef]
 
 
 class ClusterOrchestratorTypeDef(TypedDict):
@@ -7215,6 +7284,11 @@ class ListTrainingJobsForHyperParameterTuningJobRequestPaginateTypeDef(TypedDict
     PaginationConfig: NotRequired[PaginatorConfigTypeDef]
 
 
+class ListUltraServersByReservedCapacityRequestPaginateTypeDef(TypedDict):
+    ReservedCapacityArn: str
+    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
+
+
 class ListUserProfilesRequestPaginateTypeDef(TypedDict):
     SortOrder: NotRequired[SortOrderType]
     SortBy: NotRequired[UserProfileSortKeyType]
@@ -8884,6 +8958,8 @@ class SearchTrainingPlanOfferingsRequestTypeDef(TypedDict):
     TargetResources: Sequence[SageMakerResourceNameType]
     InstanceType: NotRequired[ReservedCapacityInstanceTypeType]
     InstanceCount: NotRequired[int]
+    UltraServerType: NotRequired[str]
+    UltraServerCount: NotRequired[int]
     StartTimeAfter: NotRequired[TimestampTypeDef]
     EndTimeBefore: NotRequired[TimestampTypeDef]
 
@@ -9244,6 +9320,23 @@ DescribePartnerAppResponseTypeDef = TypedDict(
 )
 
 
+class DescribeReservedCapacityResponseTypeDef(TypedDict):
+    ReservedCapacityArn: str
+    ReservedCapacityType: ReservedCapacityTypeType
+    Status: ReservedCapacityStatusType
+    AvailabilityZone: str
+    DurationHours: int
+    DurationMinutes: int
+    StartTime: datetime
+    EndTime: datetime
+    InstanceType: ReservedCapacityInstanceTypeType
+    TotalInstanceCount: int
+    AvailableInstanceCount: int
+    InUseInstanceCount: int
+    UltraServerSummary: UltraServerSummaryTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class DescribeSubscribedWorkteamResponseTypeDef(TypedDict):
     SubscribedWorkteam: SubscribedWorkteamTypeDef
     ResponseMetadata: ResponseMetadataTypeDef
@@ -9281,6 +9374,9 @@ class DescribeTrainingPlanResponseTypeDef(TypedDict):
     TotalInstanceCount: int
     AvailableInstanceCount: int
     InUseInstanceCount: int
+    UnhealthyInstanceCount: int
+    AvailableSpareInstanceCount: int
+    TotalUltraServerCount: int
     TargetResources: List[SageMakerResourceNameType]
     ReservedCapacitySummaries: List[ReservedCapacitySummaryTypeDef]
     ResponseMetadata: ResponseMetadataTypeDef
@@ -9300,6 +9396,7 @@ class TrainingPlanSummaryTypeDef(TypedDict):
     TotalInstanceCount: NotRequired[int]
     AvailableInstanceCount: NotRequired[int]
     InUseInstanceCount: NotRequired[int]
+    TotalUltraServerCount: NotRequired[int]
     TargetResources: NotRequired[List[SageMakerResourceNameType]]
     ReservedCapacitySummaries: NotRequired[List[ReservedCapacitySummaryTypeDef]]
 
@@ -9651,24 +9748,14 @@ class ListInferenceRecommendationsJobsResponseTypeDef(TypedDict):
     NextToken: NotRequired[str]
 
 
-class ResourceConfigOutputTypeDef(TypedDict):
-    VolumeSizeInGB: int
-    InstanceType: NotRequired[TrainingInstanceTypeType]
-    InstanceCount: NotRequired[int]
-    VolumeKmsKeyId: NotRequired[str]
-    KeepAlivePeriodInSeconds: NotRequired[int]
-    InstanceGroups: NotRequired[List[InstanceGroupTypeDef]]
-    TrainingPlanArn: NotRequired[str]
+class InstancePlacementConfigOutputTypeDef(TypedDict):
+    EnableMultipleJobs: NotRequired[bool]
+    PlacementSpecifications: NotRequired[List[PlacementSpecificationTypeDef]]
 
 
-class ResourceConfigTypeDef(TypedDict):
-    VolumeSizeInGB: int
-    InstanceType: NotRequired[TrainingInstanceTypeType]
-    InstanceCount: NotRequired[int]
-    VolumeKmsKeyId: NotRequired[str]
-    KeepAlivePeriodInSeconds: NotRequired[int]
-    InstanceGroups: NotRequired[Sequence[InstanceGroupTypeDef]]
-    TrainingPlanArn: NotRequired[str]
+class InstancePlacementConfigTypeDef(TypedDict):
+    EnableMultipleJobs: NotRequired[bool]
+    PlacementSpecifications: NotRequired[Sequence[PlacementSpecificationTypeDef]]
 
 
 class ParameterRangeOutputTypeDef(TypedDict):
@@ -9899,6 +9986,12 @@ class ListTransformJobsResponseTypeDef(TypedDict):
     NextToken: NotRequired[str]
 
 
+class ListUltraServersByReservedCapacityResponseTypeDef(TypedDict):
+    UltraServers: List[UltraServerTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+    NextToken: NotRequired[str]
+
+
 class ListUserProfilesResponseTypeDef(TypedDict):
     UserProfiles: List[UserProfileDetailsTypeDef]
     ResponseMetadata: ResponseMetadataTypeDef
@@ -10331,6 +10424,7 @@ class DomainSettingsOutputTypeDef(TypedDict):
     SecurityGroupIds: NotRequired[List[str]]
     RStudioServerProDomainSettings: NotRequired[RStudioServerProDomainSettingsTypeDef]
     ExecutionRoleIdentityConfig: NotRequired[ExecutionRoleIdentityConfigType]
+    TrustedIdentityPropagationSettings: NotRequired[TrustedIdentityPropagationSettingsTypeDef]
     DockerSettings: NotRequired[DockerSettingsOutputTypeDef]
     AmazonQSettings: NotRequired[AmazonQSettingsTypeDef]
     UnifiedStudioSettings: NotRequired[UnifiedStudioSettingsTypeDef]
@@ -10340,6 +10434,7 @@ class DomainSettingsTypeDef(TypedDict):
     SecurityGroupIds: NotRequired[Sequence[str]]
     RStudioServerProDomainSettings: NotRequired[RStudioServerProDomainSettingsTypeDef]
     ExecutionRoleIdentityConfig: NotRequired[ExecutionRoleIdentityConfigType]
+    TrustedIdentityPropagationSettings: NotRequired[TrustedIdentityPropagationSettingsTypeDef]
     DockerSettings: NotRequired[DockerSettingsTypeDef]
     AmazonQSettings: NotRequired[AmazonQSettingsTypeDef]
     UnifiedStudioSettings: NotRequired[UnifiedStudioSettingsTypeDef]
@@ -10653,6 +10748,7 @@ class ClusterNodeDetailsTypeDef(TypedDict):
     Placement: NotRequired[ClusterInstancePlacementTypeDef]
     CurrentImageId: NotRequired[str]
     DesiredImageId: NotRequired[str]
+    UltraServerInfo: NotRequired[UltraServerInfoTypeDef]
 
 
 class ListClusterNodesResponseTypeDef(TypedDict):
@@ -10824,6 +10920,7 @@ class DomainSettingsForUpdateTypeDef(TypedDict):
     ]
     ExecutionRoleIdentityConfig: NotRequired[ExecutionRoleIdentityConfigType]
     SecurityGroupIds: NotRequired[Sequence[str]]
+    TrustedIdentityPropagationSettings: NotRequired[TrustedIdentityPropagationSettingsTypeDef]
     DockerSettings: NotRequired[DockerSettingsUnionTypeDef]
     AmazonQSettings: NotRequired[AmazonQSettingsTypeDef]
     UnifiedStudioSettings: NotRequired[UnifiedStudioSettingsTypeDef]
@@ -11246,7 +11343,20 @@ class CreateInferenceComponentInputTypeDef(TypedDict):
     Tags: NotRequired[Sequence[TagTypeDef]]
 
 
-ResourceConfigUnionTypeDef = Union[ResourceConfigTypeDef, ResourceConfigOutputTypeDef]
+class ResourceConfigOutputTypeDef(TypedDict):
+    VolumeSizeInGB: int
+    InstanceType: NotRequired[TrainingInstanceTypeType]
+    InstanceCount: NotRequired[int]
+    VolumeKmsKeyId: NotRequired[str]
+    KeepAlivePeriodInSeconds: NotRequired[int]
+    InstanceGroups: NotRequired[List[InstanceGroupTypeDef]]
+    TrainingPlanArn: NotRequired[str]
+    InstancePlacementConfig: NotRequired[InstancePlacementConfigOutputTypeDef]
+
+
+InstancePlacementConfigUnionTypeDef = Union[
+    InstancePlacementConfigTypeDef, InstancePlacementConfigOutputTypeDef
+]
 HyperParameterSpecificationOutputTypeDef = TypedDict(
     "HyperParameterSpecificationOutputTypeDef",
     {
@@ -12229,6 +12339,17 @@ InferenceComponentDeploymentConfigUnionTypeDef = Union[
 ]
 
 
+class ResourceConfigTypeDef(TypedDict):
+    VolumeSizeInGB: int
+    InstanceType: NotRequired[TrainingInstanceTypeType]
+    InstanceCount: NotRequired[int]
+    VolumeKmsKeyId: NotRequired[str]
+    KeepAlivePeriodInSeconds: NotRequired[int]
+    InstanceGroups: NotRequired[Sequence[InstanceGroupTypeDef]]
+    TrainingPlanArn: NotRequired[str]
+    InstancePlacementConfig: NotRequired[InstancePlacementConfigUnionTypeDef]
+
+
 class TrainingSpecificationOutputTypeDef(TypedDict):
     TrainingImage: str
     SupportedTrainingInstanceTypes: List[TrainingInstanceTypeType]
@@ -13042,6 +13163,7 @@ class UpdateInferenceComponentInputTypeDef(TypedDict):
     DeploymentConfig: NotRequired[InferenceComponentDeploymentConfigUnionTypeDef]
 
 
+ResourceConfigUnionTypeDef = Union[ResourceConfigTypeDef, ResourceConfigOutputTypeDef]
 TrainingSpecificationUnionTypeDef = Union[
     TrainingSpecificationTypeDef, TrainingSpecificationOutputTypeDef
 ]