mypy-boto3-sagemaker 1.35.61__py3-none-any.whl → 1.35.75__py3-none-any.whl

mypy_boto3_sagemaker/type_defs.py

@@ -10,6 +10,8 @@ Usage::
 
     data: ActionSourceTypeDef = ...
     ```
+
+Copyright 2024 Vlad Emelianov
 """
 
 import sys
@@ -18,6 +20,7 @@ from typing import Any, Dict, List, Mapping, Sequence, Union
 
 from .literals import (
     ActionStatusType,
+    ActivationStateType,
     AdditionalS3DataSourceDataTypeType,
     AggregationTransformationValueType,
     AlgorithmSortByType,
@@ -96,6 +99,7 @@ from .literals import (
     ExecutionRoleIdentityConfigType,
     ExecutionStatusType,
     FailureHandlingPolicyType,
+    FairShareType,
     FeatureGroupSortByType,
     FeatureGroupSortOrderType,
     FeatureGroupStatusType,
@@ -132,6 +136,7 @@ from .literals import (
     InferenceExperimentStatusType,
     InferenceExperimentStopDesiredStateType,
     InputModeType,
+    InstanceGroupStatusType,
     InstanceTypeType,
     IsTrackingServerActiveType,
     JobTypeType,
@@ -192,8 +197,11 @@ from .literals import (
     OrderKeyType,
     OutputCompressionTypeType,
     ParameterTypeType,
+    PartnerAppStatusType,
+    PartnerAppTypeType,
     PipelineExecutionStatusType,
     PipelineStatusType,
+    PreemptTeamTasksType,
     ProblemTypeType,
     ProcessingInstanceTypeType,
     ProcessingJobStatusType,
@@ -217,7 +225,10 @@ from .literals import (
     RedshiftResultCompressionTypeType,
     RedshiftResultFormatType,
     RepositoryAccessModeType,
+    ReservedCapacityInstanceTypeType,
+    ReservedCapacityStatusType,
     ResourceCatalogSortOrderType,
+    ResourceSharingStrategyType,
     ResourceTypeType,
     RetentionTypeType,
     RootAccessType,
@@ -228,7 +239,9 @@ from .literals import (
     S3DataDistributionType,
     S3DataTypeType,
     S3ModelDataTypeType,
+    SageMakerResourceNameType,
     SagemakerServicecatalogStatusType,
+    SchedulerResourceStatusType,
     ScheduleStatusType,
     SearchSortOrderType,
     SecondaryStatusType,
@@ -237,6 +250,7 @@ from .literals import (
     SortActionsByType,
     SortAssociationsByType,
     SortByType,
+    SortClusterSchedulerConfigByType,
     SortContextsByType,
     SortExperimentsByType,
     SortInferenceExperimentsByType,
@@ -244,6 +258,7 @@ from .literals import (
     SortOrderType,
     SortPipelineExecutionsByType,
     SortPipelinesByType,
+    SortQuotaByType,
     SortTrackingServerByType,
     SortTrialComponentsByType,
     SortTrialsByType,
@@ -273,6 +288,9 @@ from .literals import (
     TrainingJobEarlyStoppingTypeType,
     TrainingJobSortByOptionsType,
     TrainingJobStatusType,
+    TrainingPlanSortByType,
+    TrainingPlanSortOrderType,
+    TrainingPlanStatusType,
     TrainingRepositoryAccessModeType,
     TransformInstanceTypeType,
     TransformJobStatusType,
@@ -443,6 +461,7 @@ __all__ = (
     "ClusterNodeSummaryTypeDef",
     "ClusterOrchestratorEksConfigTypeDef",
     "ClusterOrchestratorTypeDef",
+    "ClusterSchedulerConfigSummaryTypeDef",
     "ClusterSummaryTypeDef",
     "CodeEditorAppImageConfigOutputTypeDef",
     "CodeEditorAppImageConfigTypeDef",
@@ -458,6 +477,11 @@ __all__ = (
     "CollectionConfigurationTypeDef",
     "CollectionConfigurationUnionTypeDef",
     "CompilationJobSummaryTypeDef",
+    "ComputeQuotaConfigOutputTypeDef",
+    "ComputeQuotaConfigTypeDef",
+    "ComputeQuotaResourceConfigTypeDef",
+    "ComputeQuotaSummaryTypeDef",
+    "ComputeQuotaTargetTypeDef",
     "ConditionStepMetadataTypeDef",
     "ContainerConfigOutputTypeDef",
     "ContainerConfigTypeDef",
@@ -486,10 +510,14 @@ __all__ = (
     "CreateAutoMLJobV2ResponseTypeDef",
     "CreateClusterRequestRequestTypeDef",
     "CreateClusterResponseTypeDef",
+    "CreateClusterSchedulerConfigRequestRequestTypeDef",
+    "CreateClusterSchedulerConfigResponseTypeDef",
     "CreateCodeRepositoryInputRequestTypeDef",
     "CreateCodeRepositoryOutputTypeDef",
     "CreateCompilationJobRequestRequestTypeDef",
     "CreateCompilationJobResponseTypeDef",
+    "CreateComputeQuotaRequestRequestTypeDef",
+    "CreateComputeQuotaResponseTypeDef",
     "CreateContextRequestRequestTypeDef",
     "CreateContextResponseTypeDef",
     "CreateDataQualityJobDefinitionRequestRequestTypeDef",
@@ -557,6 +585,10 @@ __all__ = (
     "CreateNotebookInstanceOutputTypeDef",
     "CreateOptimizationJobRequestRequestTypeDef",
     "CreateOptimizationJobResponseTypeDef",
+    "CreatePartnerAppPresignedUrlRequestRequestTypeDef",
+    "CreatePartnerAppPresignedUrlResponseTypeDef",
+    "CreatePartnerAppRequestRequestTypeDef",
+    "CreatePartnerAppResponseTypeDef",
     "CreatePipelineRequestRequestTypeDef",
     "CreatePipelineResponseTypeDef",
     "CreatePresignedDomainUrlRequestRequestTypeDef",
@@ -575,6 +607,8 @@ __all__ = (
     "CreateStudioLifecycleConfigResponseTypeDef",
     "CreateTrainingJobRequestRequestTypeDef",
     "CreateTrainingJobResponseTypeDef",
+    "CreateTrainingPlanRequestRequestTypeDef",
+    "CreateTrainingPlanResponseTypeDef",
     "CreateTransformJobRequestRequestTypeDef",
     "CreateTransformJobResponseTypeDef",
     "CreateTrialComponentRequestRequestTypeDef",
@@ -627,8 +661,10 @@ __all__ = (
     "DeleteAssociationResponseTypeDef",
     "DeleteClusterRequestRequestTypeDef",
     "DeleteClusterResponseTypeDef",
+    "DeleteClusterSchedulerConfigRequestRequestTypeDef",
     "DeleteCodeRepositoryInputRequestTypeDef",
     "DeleteCompilationJobRequestRequestTypeDef",
+    "DeleteComputeQuotaRequestRequestTypeDef",
     "DeleteContextRequestRequestTypeDef",
     "DeleteContextResponseTypeDef",
     "DeleteDataQualityJobDefinitionRequestRequestTypeDef",
@@ -666,6 +702,8 @@ __all__ = (
     "DeleteNotebookInstanceInputRequestTypeDef",
     "DeleteNotebookInstanceLifecycleConfigInputRequestTypeDef",
     "DeleteOptimizationJobRequestRequestTypeDef",
+    "DeletePartnerAppRequestRequestTypeDef",
+    "DeletePartnerAppResponseTypeDef",
     "DeletePipelineRequestRequestTypeDef",
     "DeletePipelineResponseTypeDef",
     "DeleteProjectInputRequestTypeDef",
@@ -706,10 +744,14 @@ __all__ = (
     "DescribeClusterNodeResponseTypeDef",
     "DescribeClusterRequestRequestTypeDef",
     "DescribeClusterResponseTypeDef",
+    "DescribeClusterSchedulerConfigRequestRequestTypeDef",
+    "DescribeClusterSchedulerConfigResponseTypeDef",
     "DescribeCodeRepositoryInputRequestTypeDef",
     "DescribeCodeRepositoryOutputTypeDef",
     "DescribeCompilationJobRequestRequestTypeDef",
     "DescribeCompilationJobResponseTypeDef",
+    "DescribeComputeQuotaRequestRequestTypeDef",
+    "DescribeComputeQuotaResponseTypeDef",
     "DescribeContextRequestRequestTypeDef",
     "DescribeContextResponseTypeDef",
     "DescribeDataQualityJobDefinitionRequestRequestTypeDef",
@@ -794,6 +836,8 @@ __all__ = (
     "DescribeNotebookInstanceOutputTypeDef",
     "DescribeOptimizationJobRequestRequestTypeDef",
     "DescribeOptimizationJobResponseTypeDef",
+    "DescribePartnerAppRequestRequestTypeDef",
+    "DescribePartnerAppResponseTypeDef",
     "DescribePipelineDefinitionForExecutionRequestRequestTypeDef",
     "DescribePipelineDefinitionForExecutionResponseTypeDef",
     "DescribePipelineExecutionRequestRequestTypeDef",
@@ -814,6 +858,8 @@ __all__ = (
     "DescribeTrainingJobRequestRequestTypeDef",
     "DescribeTrainingJobRequestTrainingJobCompletedOrStoppedWaitTypeDef",
     "DescribeTrainingJobResponseTypeDef",
+    "DescribeTrainingPlanRequestRequestTypeDef",
+    "DescribeTrainingPlanResponseTypeDef",
     "DescribeTransformJobRequestRequestTypeDef",
     "DescribeTransformJobRequestTransformJobCompletedOrStoppedWaitTypeDef",
     "DescribeTransformJobResponseTypeDef",
@@ -890,6 +936,7 @@ __all__ = (
     "EnvironmentParameterRangesTypeDef",
     "EnvironmentParameterRangesUnionTypeDef",
     "EnvironmentParameterTypeDef",
+    "ErrorInfoTypeDef",
     "ExperimentConfigTypeDef",
     "ExperimentSourceTypeDef",
     "ExperimentSummaryTypeDef",
@@ -897,6 +944,8 @@ __all__ = (
     "ExplainabilityTypeDef",
     "ExplainerConfigOutputTypeDef",
     "ExplainerConfigTypeDef",
+    "FSxLustreFileSystemConfigTypeDef",
+    "FSxLustreFileSystemTypeDef",
     "FailStepMetadataTypeDef",
     "FeatureDefinitionTypeDef",
     "FeatureGroupSummaryTypeDef",
@@ -1072,6 +1121,9 @@ __all__ = (
     "ListClusterNodesRequestListClusterNodesPaginateTypeDef",
     "ListClusterNodesRequestRequestTypeDef",
     "ListClusterNodesResponseTypeDef",
+    "ListClusterSchedulerConfigsRequestListClusterSchedulerConfigsPaginateTypeDef",
+    "ListClusterSchedulerConfigsRequestRequestTypeDef",
+    "ListClusterSchedulerConfigsResponseTypeDef",
     "ListClustersRequestListClustersPaginateTypeDef",
     "ListClustersRequestRequestTypeDef",
     "ListClustersResponseTypeDef",
@@ -1081,6 +1133,9 @@ __all__ = (
     "ListCompilationJobsRequestListCompilationJobsPaginateTypeDef",
     "ListCompilationJobsRequestRequestTypeDef",
     "ListCompilationJobsResponseTypeDef",
+    "ListComputeQuotasRequestListComputeQuotasPaginateTypeDef",
+    "ListComputeQuotasRequestRequestTypeDef",
+    "ListComputeQuotasResponseTypeDef",
     "ListContextsRequestListContextsPaginateTypeDef",
     "ListContextsRequestRequestTypeDef",
     "ListContextsResponseTypeDef",
@@ -1210,6 +1265,9 @@ __all__ = (
     "ListOptimizationJobsRequestListOptimizationJobsPaginateTypeDef",
     "ListOptimizationJobsRequestRequestTypeDef",
     "ListOptimizationJobsResponseTypeDef",
+    "ListPartnerAppsRequestListPartnerAppsPaginateTypeDef",
+    "ListPartnerAppsRequestRequestTypeDef",
+    "ListPartnerAppsResponseTypeDef",
     "ListPipelineExecutionStepsRequestListPipelineExecutionStepsPaginateTypeDef",
     "ListPipelineExecutionStepsRequestRequestTypeDef",
     "ListPipelineExecutionStepsResponseTypeDef",
@@ -1251,6 +1309,9 @@ __all__ = (
     "ListTrainingJobsRequestListTrainingJobsPaginateTypeDef",
     "ListTrainingJobsRequestRequestTypeDef",
     "ListTrainingJobsResponseTypeDef",
+    "ListTrainingPlansRequestListTrainingPlansPaginateTypeDef",
+    "ListTrainingPlansRequestRequestTypeDef",
+    "ListTrainingPlansResponseTypeDef",
     "ListTransformJobsRequestListTransformJobsPaginateTypeDef",
     "ListTransformJobsRequestRequestTypeDef",
     "ListTransformJobsResponseTypeDef",
@@ -1346,6 +1407,9 @@ __all__ = (
     "ModelQuantizationConfigTypeDef",
     "ModelQuantizationConfigUnionTypeDef",
     "ModelRegisterSettingsTypeDef",
+    "ModelShardingConfigOutputTypeDef",
+    "ModelShardingConfigTypeDef",
+    "ModelShardingConfigUnionTypeDef",
     "ModelStepMetadataTypeDef",
     "ModelSummaryTypeDef",
     "ModelTypeDef",
@@ -1437,6 +1501,10 @@ __all__ = (
     "ParameterTypeDef",
     "ParentHyperParameterTuningJobTypeDef",
     "ParentTypeDef",
+    "PartnerAppConfigOutputTypeDef",
+    "PartnerAppConfigTypeDef",
+    "PartnerAppMaintenanceConfigTypeDef",
+    "PartnerAppSummaryTypeDef",
     "PendingDeploymentSummaryTypeDef",
     "PendingProductionVariantSummaryTypeDef",
     "PhaseTypeDef",
@@ -1449,6 +1517,7 @@ __all__ = (
     "PipelineSummaryTypeDef",
     "PipelineTypeDef",
     "PredefinedMetricSpecificationTypeDef",
+    "PriorityClassTypeDef",
     "ProcessingClusterConfigTypeDef",
     "ProcessingFeatureStoreOutputTypeDef",
     "ProcessingInputTypeDef",
@@ -1525,6 +1594,8 @@ __all__ = (
     "RenderableTaskTypeDef",
     "RenderingErrorTypeDef",
     "RepositoryAuthConfigTypeDef",
+    "ReservedCapacityOfferingTypeDef",
+    "ReservedCapacitySummaryTypeDef",
     "ResolvedAttributesTypeDef",
     "ResourceCatalogTypeDef",
     "ResourceConfigForUpdateTypeDef",
@@ -1532,6 +1603,7 @@ __all__ = (
     "ResourceConfigTypeDef",
     "ResourceConfigUnionTypeDef",
     "ResourceLimitsTypeDef",
+    "ResourceSharingConfigTypeDef",
     "ResourceSpecTypeDef",
     "ResponseMetadataTypeDef",
     "RetentionPolicyTypeDef",
@@ -1549,12 +1621,16 @@ __all__ = (
     "ScalingPolicyObjectiveTypeDef",
     "ScalingPolicyTypeDef",
     "ScheduleConfigTypeDef",
+    "SchedulerConfigOutputTypeDef",
+    "SchedulerConfigTypeDef",
     "SearchExpressionPaginatorTypeDef",
     "SearchExpressionTypeDef",
     "SearchRecordTypeDef",
     "SearchRequestRequestTypeDef",
     "SearchRequestSearchPaginateTypeDef",
     "SearchResponseTypeDef",
+    "SearchTrainingPlanOfferingsRequestRequestTypeDef",
+    "SearchTrainingPlanOfferingsResponseTypeDef",
     "SecondaryStatusTransitionTypeDef",
     "SelectedStepTypeDef",
     "SelectiveExecutionConfigOutputTypeDef",
@@ -1669,6 +1745,9 @@ __all__ = (
     "TrainingJobStepMetadataTypeDef",
     "TrainingJobSummaryTypeDef",
     "TrainingJobTypeDef",
+    "TrainingPlanFilterTypeDef",
+    "TrainingPlanOfferingTypeDef",
+    "TrainingPlanSummaryTypeDef",
     "TrainingRepositoryAuthConfigTypeDef",
     "TrainingSpecificationOutputTypeDef",
     "TrainingSpecificationTypeDef",
@@ -1710,10 +1789,14 @@ __all__ = (
     "UpdateArtifactResponseTypeDef",
     "UpdateClusterRequestRequestTypeDef",
     "UpdateClusterResponseTypeDef",
+    "UpdateClusterSchedulerConfigRequestRequestTypeDef",
+    "UpdateClusterSchedulerConfigResponseTypeDef",
     "UpdateClusterSoftwareRequestRequestTypeDef",
     "UpdateClusterSoftwareResponseTypeDef",
     "UpdateCodeRepositoryInputRequestTypeDef",
     "UpdateCodeRepositoryOutputTypeDef",
+    "UpdateComputeQuotaRequestRequestTypeDef",
+    "UpdateComputeQuotaResponseTypeDef",
     "UpdateContextRequestRequestTypeDef",
     "UpdateContextResponseTypeDef",
     "UpdateDeviceFleetRequestRequestTypeDef",
@@ -1753,6 +1836,8 @@ __all__ = (
     "UpdateMonitoringScheduleResponseTypeDef",
     "UpdateNotebookInstanceInputRequestTypeDef",
     "UpdateNotebookInstanceLifecycleConfigInputRequestTypeDef",
+    "UpdatePartnerAppRequestRequestTypeDef",
+    "UpdatePartnerAppResponseTypeDef",
     "UpdatePipelineExecutionRequestRequestTypeDef",
     "UpdatePipelineExecutionResponseTypeDef",
     "UpdatePipelineRequestRequestTypeDef",
@@ -2272,11 +2357,23 @@ class ClusterOrchestratorEksConfigTypeDef(TypedDict):
     ClusterArn: str
 
 
+class ClusterSchedulerConfigSummaryTypeDef(TypedDict):
+    ClusterSchedulerConfigArn: str
+    ClusterSchedulerConfigId: str
+    Name: str
+    CreationTime: datetime
+    Status: SchedulerResourceStatusType
+    ClusterSchedulerConfigVersion: NotRequired[int]
+    LastModifiedTime: NotRequired[datetime]
+    ClusterArn: NotRequired[str]
+
+
 class ClusterSummaryTypeDef(TypedDict):
     ClusterArn: str
     ClusterName: str
     CreationTime: datetime
     ClusterStatus: ClusterStatusType
+    TrainingPlanArns: NotRequired[List[str]]
 
 
 class ContainerConfigOutputTypeDef(TypedDict):
@@ -2346,6 +2443,21 @@ class CompilationJobSummaryTypeDef(TypedDict):
     LastModifiedTime: NotRequired[datetime]
 
 
+class ComputeQuotaResourceConfigTypeDef(TypedDict):
+    InstanceType: ClusterInstanceTypeType
+    Count: int
+
+
+class ResourceSharingConfigTypeDef(TypedDict):
+    Strategy: ResourceSharingStrategyType
+    BorrowLimit: NotRequired[int]
+
+
+class ComputeQuotaTargetTypeDef(TypedDict):
+    TeamName: str
+    FairShareWeight: NotRequired[int]
+
+
 class ConditionStepMetadataTypeDef(TypedDict):
     Outcome: NotRequired[ConditionOutcomeType]
 
@@ -2561,6 +2673,21 @@ class OptimizationVpcConfigTypeDef(TypedDict):
     Subnets: Sequence[str]
 
 
+class CreatePartnerAppPresignedUrlRequestRequestTypeDef(TypedDict):
+    Arn: str
+    ExpiresInSeconds: NotRequired[int]
+    SessionExpirationDurationInSeconds: NotRequired[int]
+
+
+class PartnerAppConfigTypeDef(TypedDict):
+    AdminUsers: NotRequired[Sequence[str]]
+    Arguments: NotRequired[Mapping[str, str]]
+
+
+class PartnerAppMaintenanceConfigTypeDef(TypedDict):
+    MaintenanceWindowStart: NotRequired[str]
+
+
 class ParallelismConfigurationTypeDef(TypedDict):
     MaxParallelExecutionSteps: int
 
@@ -2718,10 +2845,19 @@ class EFSFileSystemConfigTypeDef(TypedDict):
     FileSystemPath: NotRequired[str]
 
 
+class FSxLustreFileSystemConfigTypeDef(TypedDict):
+    FileSystemId: str
+    FileSystemPath: NotRequired[str]
+
+
 class EFSFileSystemTypeDef(TypedDict):
     FileSystemId: str
 
 
+class FSxLustreFileSystemTypeDef(TypedDict):
+    FileSystemId: str
+
+
 class CustomPosixUserConfigTypeDef(TypedDict):
     Uid: int
     Gid: int
@@ -2875,6 +3011,10 @@ class DeleteClusterRequestRequestTypeDef(TypedDict):
     ClusterName: str
 
 
+class DeleteClusterSchedulerConfigRequestRequestTypeDef(TypedDict):
+    ClusterSchedulerConfigId: str
+
+
 class DeleteCodeRepositoryInputRequestTypeDef(TypedDict):
     CodeRepositoryName: str
 
@@ -2883,6 +3023,10 @@ class DeleteCompilationJobRequestRequestTypeDef(TypedDict):
     CompilationJobName: str
 
 
+class DeleteComputeQuotaRequestRequestTypeDef(TypedDict):
+    ComputeQuotaId: str
+
+
 class DeleteContextRequestRequestTypeDef(TypedDict):
     ContextName: str
 
@@ -3023,6 +3167,11 @@ class DeleteOptimizationJobRequestRequestTypeDef(TypedDict):
     OptimizationJobName: str
 
 
+class DeletePartnerAppRequestRequestTypeDef(TypedDict):
+    Arn: str
+    ClientToken: NotRequired[str]
+
+
 class DeletePipelineRequestRequestTypeDef(TypedDict):
     PipelineName: str
     ClientRequestToken: str
@@ -3153,6 +3302,11 @@ class DescribeClusterRequestRequestTypeDef(TypedDict):
     ClusterName: str
 
 
+class DescribeClusterSchedulerConfigRequestRequestTypeDef(TypedDict):
+    ClusterSchedulerConfigId: str
+    ClusterSchedulerConfigVersion: NotRequired[int]
+
+
 class DescribeCodeRepositoryInputRequestTypeDef(TypedDict):
     CodeRepositoryName: str
 
@@ -3174,6 +3328,11 @@ class NeoVpcConfigOutputTypeDef(TypedDict):
     Subnets: List[str]
 
 
+class DescribeComputeQuotaRequestRequestTypeDef(TypedDict):
+    ComputeQuotaId: str
+    ComputeQuotaVersion: NotRequired[int]
+
+
 class DescribeContextRequestRequestTypeDef(TypedDict):
     ContextName: str
 
@@ -3494,6 +3653,20 @@ class OptimizationVpcConfigOutputTypeDef(TypedDict):
     Subnets: List[str]
 
 
+class DescribePartnerAppRequestRequestTypeDef(TypedDict):
+    Arn: str
+
+
+class ErrorInfoTypeDef(TypedDict):
+    Code: NotRequired[str]
+    Reason: NotRequired[str]
+
+
+class PartnerAppConfigOutputTypeDef(TypedDict):
+    AdminUsers: NotRequired[List[str]]
+    Arguments: NotRequired[Dict[str, str]]
+
+
 class DescribePipelineDefinitionForExecutionRequestRequestTypeDef(TypedDict):
     PipelineExecutionArn: str
 
@@ -3593,6 +3766,22 @@ class WarmPoolStatusTypeDef(TypedDict):
     ReusedByJob: NotRequired[str]
 
 
+class DescribeTrainingPlanRequestRequestTypeDef(TypedDict):
+    TrainingPlanName: str
+
+
+class ReservedCapacitySummaryTypeDef(TypedDict):
+    ReservedCapacityArn: str
+    InstanceType: ReservedCapacityInstanceTypeType
+    TotalInstanceCount: int
+    Status: ReservedCapacityStatusType
+    AvailabilityZone: NotRequired[str]
+    DurationHours: NotRequired[int]
+    DurationMinutes: NotRequired[int]
+    StartTime: NotRequired[datetime]
+    EndTime: NotRequired[datetime]
+
+
 class DescribeTransformJobRequestRequestTypeDef(TypedDict):
     TransformJobName: str
 
@@ -4293,6 +4482,23 @@ class OptimizationJobSummaryTypeDef(TypedDict):
     LastModifiedTime: NotRequired[datetime]
 
 
+class ListPartnerAppsRequestRequestTypeDef(TypedDict):
+    MaxResults: NotRequired[int]
+    NextToken: NotRequired[str]
+
+
+PartnerAppSummaryTypeDef = TypedDict(
+    "PartnerAppSummaryTypeDef",
+    {
+        "Arn": NotRequired[str],
+        "Name": NotRequired[str],
+        "Type": NotRequired[PartnerAppTypeType],
+        "Status": NotRequired[PartnerAppStatusType],
+        "CreationTime": NotRequired[datetime],
+    },
+)
+
+
 class ListPipelineExecutionStepsRequestRequestTypeDef(TypedDict):
     PipelineExecutionArn: NotRequired[str]
     NextToken: NotRequired[str]
@@ -4404,6 +4610,11 @@ class ListTrainingJobsForHyperParameterTuningJobRequestRequestTypeDef(TypedDict)
     SortOrder: NotRequired[SortOrderType]
 
 
+class TrainingPlanFilterTypeDef(TypedDict):
+    Name: Literal["Status"]
+    Value: str
+
+
 class TransformJobSummaryTypeDef(TypedDict):
     TransformJobName: str
     TransformJobArn: str
@@ -4520,6 +4731,16 @@ class ModelQuantizationConfigTypeDef(TypedDict):
     OverrideEnvironment: NotRequired[Mapping[str, str]]
 
 
+class ModelShardingConfigOutputTypeDef(TypedDict):
+    Image: NotRequired[str]
+    OverrideEnvironment: NotRequired[Dict[str, str]]
+
+
+class ModelShardingConfigTypeDef(TypedDict):
+    Image: NotRequired[str]
+    OverrideEnvironment: NotRequired[Mapping[str, str]]
+
+
 class ModelStepMetadataTypeDef(TypedDict):
     Arn: NotRequired[str]
 
@@ -4676,6 +4897,11 @@ class SelectiveExecutionResultTypeDef(TypedDict):
     SourcePipelineExecutionArn: NotRequired[str]
 
 
+class PriorityClassTypeDef(TypedDict):
+    Name: str
+    Weight: int
+
+
 class ProcessingClusterConfigTypeDef(TypedDict):
     InstanceCount: int
     InstanceType: ProcessingInstanceTypeType
@@ -4805,6 +5031,16 @@ class RenderingErrorTypeDef(TypedDict):
     Message: str
 
 
+class ReservedCapacityOfferingTypeDef(TypedDict):
+    InstanceType: ReservedCapacityInstanceTypeType
+    InstanceCount: int
+    AvailabilityZone: NotRequired[str]
+    DurationHours: NotRequired[int]
+    DurationMinutes: NotRequired[int]
+    StartTime: NotRequired[datetime]
+    EndTime: NotRequired[datetime]
+
+
 class ResourceConfigForUpdateTypeDef(TypedDict):
     KeepAlivePeriodInSeconds: int
 
@@ -5148,6 +5384,12 @@ class CreateClusterResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
+class CreateClusterSchedulerConfigResponseTypeDef(TypedDict):
+    ClusterSchedulerConfigArn: str
+    ClusterSchedulerConfigId: str
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class CreateCodeRepositoryOutputTypeDef(TypedDict):
     CodeRepositoryArn: str
     ResponseMetadata: ResponseMetadataTypeDef
@@ -5158,6 +5400,12 @@ class CreateCompilationJobResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
+class CreateComputeQuotaResponseTypeDef(TypedDict):
+    ComputeQuotaArn: str
+    ComputeQuotaId: str
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class CreateContextResponseTypeDef(TypedDict):
     ContextArn: str
     ResponseMetadata: ResponseMetadataTypeDef
@@ -5320,6 +5568,16 @@ class CreateOptimizationJobResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
+class CreatePartnerAppPresignedUrlResponseTypeDef(TypedDict):
+    Url: str
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class CreatePartnerAppResponseTypeDef(TypedDict):
+    Arn: str
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class CreatePipelineResponseTypeDef(TypedDict):
     PipelineArn: str
     ResponseMetadata: ResponseMetadataTypeDef
@@ -5366,6 +5624,11 @@ class CreateTrainingJobResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
+class CreateTrainingPlanResponseTypeDef(TypedDict):
+    TrainingPlanArn: str
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class CreateTransformJobResponseTypeDef(TypedDict):
     TransformJobArn: str
     ResponseMetadata: ResponseMetadataTypeDef
@@ -5437,6 +5700,11 @@ class DeleteMlflowTrackingServerResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
+class DeletePartnerAppResponseTypeDef(TypedDict):
+    Arn: str
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class DeletePipelineResponseTypeDef(TypedDict):
     PipelineArn: str
     ResponseMetadata: ResponseMetadataTypeDef
@@ -5614,6 +5882,12 @@ class UpdateClusterResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
+class UpdateClusterSchedulerConfigResponseTypeDef(TypedDict):
+    ClusterSchedulerConfigArn: str
+    ClusterSchedulerConfigVersion: int
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class UpdateClusterSoftwareResponseTypeDef(TypedDict):
     ClusterArn: str
     ResponseMetadata: ResponseMetadataTypeDef
@@ -5624,6 +5898,12 @@ class UpdateCodeRepositoryOutputTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
+class UpdateComputeQuotaResponseTypeDef(TypedDict):
+    ComputeQuotaArn: str
+    ComputeQuotaVersion: int
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class UpdateContextResponseTypeDef(TypedDict):
     ContextArn: str
     ResponseMetadata: ResponseMetadataTypeDef
@@ -5710,6 +5990,11 @@ class UpdateMonitoringScheduleResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
+class UpdatePartnerAppResponseTypeDef(TypedDict):
+    Arn: str
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class UpdatePipelineExecutionResponseTypeDef(TypedDict):
     PipelineExecutionArn: str
     ResponseMetadata: ResponseMetadataTypeDef
@@ -5807,6 +6092,12 @@ class CreateStudioLifecycleConfigRequestRequestTypeDef(TypedDict):
     Tags: NotRequired[Sequence[TagTypeDef]]
 
 
+class CreateTrainingPlanRequestRequestTypeDef(TypedDict):
+    TrainingPlanName: str
+    TrainingPlanOfferingId: str
+    Tags: NotRequired[Sequence[TagTypeDef]]
+
+
 class ImportHubContentRequestRequestTypeDef(TypedDict):
     HubContentName: str
     HubContentType: HubContentTypeType
@@ -6184,6 +6475,12 @@ class ClusterOrchestratorTypeDef(TypedDict):
     Eks: ClusterOrchestratorEksConfigTypeDef
 
 
+class ListClusterSchedulerConfigsResponseTypeDef(TypedDict):
+    ClusterSchedulerConfigSummaries: List[ClusterSchedulerConfigSummaryTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+    NextToken: NotRequired[str]
+
+
 class ListClustersResponseTypeDef(TypedDict):
     NextToken: str
     ClusterSummaries: List[ClusterSummaryTypeDef]
@@ -6279,6 +6576,18 @@ class ListCompilationJobsResponseTypeDef(TypedDict):
     NextToken: NotRequired[str]
 
 
+class ComputeQuotaConfigOutputTypeDef(TypedDict):
+    ComputeQuotaResources: NotRequired[List[ComputeQuotaResourceConfigTypeDef]]
+    ResourceSharingConfig: NotRequired[ResourceSharingConfigTypeDef]
+    PreemptTeamTasks: NotRequired[PreemptTeamTasksType]
+
+
+class ComputeQuotaConfigTypeDef(TypedDict):
+    ComputeQuotaResources: NotRequired[Sequence[ComputeQuotaResourceConfigTypeDef]]
+    ResourceSharingConfig: NotRequired[ResourceSharingConfigTypeDef]
+    PreemptTeamTasks: NotRequired[PreemptTeamTasksType]
+
+
 ContainerConfigUnionTypeDef = Union[ContainerConfigTypeDef, ContainerConfigOutputTypeDef]
 
 
@@ -6501,6 +6810,33 @@ class UpdateNotebookInstanceLifecycleConfigInputRequestTypeDef(TypedDict):
     OnStart: NotRequired[Sequence[NotebookInstanceLifecycleHookTypeDef]]
 
 
+CreatePartnerAppRequestRequestTypeDef = TypedDict(
+    "CreatePartnerAppRequestRequestTypeDef",
+    {
+        "Name": str,
+        "Type": PartnerAppTypeType,
+        "ExecutionRoleArn": str,
+        "Tier": str,
+        "AuthType": Literal["IAM"],
+        "MaintenanceConfig": NotRequired[PartnerAppMaintenanceConfigTypeDef],
+        "ApplicationConfig": NotRequired[PartnerAppConfigTypeDef],
+        "EnableIamSessionBasedIdentity": NotRequired[bool],
+        "ClientToken": NotRequired[str],
+        "Tags": NotRequired[Sequence[TagTypeDef]],
+    },
+)
+
+
+class UpdatePartnerAppRequestRequestTypeDef(TypedDict):
+    Arn: str
+    MaintenanceConfig: NotRequired[PartnerAppMaintenanceConfigTypeDef]
+    Tier: NotRequired[str]
+    ApplicationConfig: NotRequired[PartnerAppConfigTypeDef]
+    EnableIamSessionBasedIdentity: NotRequired[bool]
+    ClientToken: NotRequired[str]
+    Tags: NotRequired[Sequence[TagTypeDef]]
+
+
 class RetryPipelineExecutionRequestRequestTypeDef(TypedDict):
     PipelineExecutionArn: str
     ClientRequestToken: str
@@ -6623,6 +6959,18 @@ class ListClusterNodesRequestRequestTypeDef(TypedDict):
     SortOrder: NotRequired[SortOrderType]
 
 
+class ListClusterSchedulerConfigsRequestRequestTypeDef(TypedDict):
+    CreatedAfter: NotRequired[TimestampTypeDef]
+    CreatedBefore: NotRequired[TimestampTypeDef]
+    NameContains: NotRequired[str]
+    ClusterArn: NotRequired[str]
+    Status: NotRequired[SchedulerResourceStatusType]
+    SortBy: NotRequired[SortClusterSchedulerConfigByType]
+    SortOrder: NotRequired[SortOrderType]
+    NextToken: NotRequired[str]
+    MaxResults: NotRequired[int]
+
+
 class ListClustersRequestRequestTypeDef(TypedDict):
     CreationTimeAfter: NotRequired[TimestampTypeDef]
     CreationTimeBefore: NotRequired[TimestampTypeDef]
@@ -6631,6 +6979,7 @@ class ListClustersRequestRequestTypeDef(TypedDict):
     NextToken: NotRequired[str]
     SortBy: NotRequired[ClusterSortByType]
     SortOrder: NotRequired[SortOrderType]
+    TrainingPlanArn: NotRequired[str]
 
 
 class ListCodeRepositoriesInputRequestTypeDef(TypedDict):
@@ -6658,6 +7007,18 @@ class ListCompilationJobsRequestRequestTypeDef(TypedDict):
     SortOrder: NotRequired[SortOrderType]
 
 
+class ListComputeQuotasRequestRequestTypeDef(TypedDict):
+    CreatedAfter: NotRequired[TimestampTypeDef]
+    CreatedBefore: NotRequired[TimestampTypeDef]
+    NameContains: NotRequired[str]
+    Status: NotRequired[SchedulerResourceStatusType]
+    ClusterArn: NotRequired[str]
+    SortBy: NotRequired[SortQuotaByType]
+    SortOrder: NotRequired[SortOrderType]
+    NextToken: NotRequired[str]
+    MaxResults: NotRequired[int]
+
+
 class ListContextsRequestRequestTypeDef(TypedDict):
     SourceUri: NotRequired[str]
     ContextType: NotRequired[str]
@@ -7223,6 +7584,7 @@ class ListTrainingJobsRequestRequestTypeDef(TypedDict):
     SortBy: NotRequired[SortByType]
     SortOrder: NotRequired[SortOrderType]
     WarmPoolStatusEquals: NotRequired[WarmPoolResourceStatusType]
+    TrainingPlanArnEquals: NotRequired[str]
 
 
 class ListTransformJobsRequestRequestTypeDef(TypedDict):
@@ -7271,6 +7633,15 @@ class QueryFiltersTypeDef(TypedDict):
     Properties: NotRequired[Mapping[str, str]]
 
 
+class SearchTrainingPlanOfferingsRequestRequestTypeDef(TypedDict):
+    InstanceType: ReservedCapacityInstanceTypeType
+    InstanceCount: int
+    TargetResources: Sequence[SageMakerResourceNameType]
+    StartTimeAfter: NotRequired[TimestampTypeDef]
+    EndTimeBefore: NotRequired[TimestampTypeDef]
+    DurationHours: NotRequired[int]
+
+
 class CreateTrialComponentRequestRequestTypeDef(TypedDict):
     TrialComponentName: str
     DisplayName: NotRequired[str]
@@ -7316,10 +7687,12 @@ class UpdateWorkforceRequestRequestTypeDef(TypedDict):
 
 class CustomFileSystemConfigTypeDef(TypedDict):
     EFSFileSystemConfig: NotRequired[EFSFileSystemConfigTypeDef]
+    FSxLustreFileSystemConfig: NotRequired[FSxLustreFileSystemConfigTypeDef]
 
 
 class CustomFileSystemTypeDef(TypedDict):
     EFSFileSystem: NotRequired[EFSFileSystemTypeDef]
+    FSxLustreFileSystem: NotRequired[FSxLustreFileSystemTypeDef]
 
 
 class ModelBiasBaselineConfigTypeDef(TypedDict):
@@ -7614,6 +7987,28 @@ class ListMonitoringExecutionsResponseTypeDef(TypedDict):
     NextToken: NotRequired[str]
 
 
+DescribePartnerAppResponseTypeDef = TypedDict(
+    "DescribePartnerAppResponseTypeDef",
+    {
+        "Arn": str,
+        "Name": str,
+        "Type": PartnerAppTypeType,
+        "Status": PartnerAppStatusType,
+        "CreationTime": datetime,
+        "ExecutionRoleArn": str,
+        "BaseUrl": str,
+        "MaintenanceConfig": PartnerAppMaintenanceConfigTypeDef,
+        "Tier": str,
+        "Version": str,
+        "ApplicationConfig": PartnerAppConfigOutputTypeDef,
+        "AuthType": Literal["IAM"],
+        "EnableIamSessionBasedIdentity": bool,
+        "Error": ErrorInfoTypeDef,
+        "ResponseMetadata": ResponseMetadataTypeDef,
+    },
+)
+
+
 class DescribeSubscribedWorkteamResponseTypeDef(TypedDict):
     SubscribedWorkteam: SubscribedWorkteamTypeDef
     ResponseMetadata: ResponseMetadataTypeDef
@@ -7634,6 +8029,44 @@ class TrainingJobSummaryTypeDef(TypedDict):
     LastModifiedTime: NotRequired[datetime]
     SecondaryStatus: NotRequired[SecondaryStatusType]
     WarmPoolStatus: NotRequired[WarmPoolStatusTypeDef]
+    TrainingPlanArn: NotRequired[str]
+
+
+class DescribeTrainingPlanResponseTypeDef(TypedDict):
+    TrainingPlanArn: str
+    TrainingPlanName: str
+    Status: TrainingPlanStatusType
+    StatusMessage: str
+    DurationHours: int
+    DurationMinutes: int
+    StartTime: datetime
+    EndTime: datetime
+    UpfrontFee: str
+    CurrencyCode: str
+    TotalInstanceCount: int
+    AvailableInstanceCount: int
+    InUseInstanceCount: int
+    TargetResources: List[SageMakerResourceNameType]
+    ReservedCapacitySummaries: List[ReservedCapacitySummaryTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class TrainingPlanSummaryTypeDef(TypedDict):
+    TrainingPlanArn: str
+    TrainingPlanName: str
+    Status: TrainingPlanStatusType
+    StatusMessage: NotRequired[str]
+    DurationHours: NotRequired[int]
+    DurationMinutes: NotRequired[int]
+    StartTime: NotRequired[datetime]
+    EndTime: NotRequired[datetime]
+    UpfrontFee: NotRequired[str]
+    CurrencyCode: NotRequired[str]
+    TotalInstanceCount: NotRequired[int]
+    AvailableInstanceCount: NotRequired[int]
+    InUseInstanceCount: NotRequired[int]
+    TargetResources: NotRequired[List[SageMakerResourceNameType]]
+    ReservedCapacitySummaries: NotRequired[List[ReservedCapacitySummaryTypeDef]]
 
 
 class TrialSummaryTypeDef(TypedDict):
@@ -7936,10 +8369,13 @@ class ListImageVersionsResponseTypeDef(TypedDict):
 InferenceComponentSpecificationTypeDef = TypedDict(
     "InferenceComponentSpecificationTypeDef",
     {
-        "ComputeResourceRequirements": InferenceComponentComputeResourceRequirementsTypeDef,
         "ModelName": NotRequired[str],
         "Container": NotRequired[InferenceComponentContainerSpecificationTypeDef],
         "StartupParameters": NotRequired[InferenceComponentStartupParametersTypeDef],
+        "ComputeResourceRequirements": NotRequired[
+            InferenceComponentComputeResourceRequirementsTypeDef
+        ],
+        "BaseInferenceComponentName": NotRequired[str],
     },
 )
 
@@ -7963,6 +8399,7 @@ class ResourceConfigOutputTypeDef(TypedDict):
     VolumeKmsKeyId: NotRequired[str]
     KeepAlivePeriodInSeconds: NotRequired[int]
     InstanceGroups: NotRequired[List[InstanceGroupTypeDef]]
+    TrainingPlanArn: NotRequired[str]
 
 
 class ResourceConfigTypeDef(TypedDict):
@@ -7972,6 +8409,7 @@ class ResourceConfigTypeDef(TypedDict):
     VolumeKmsKeyId: NotRequired[str]
     KeepAlivePeriodInSeconds: NotRequired[int]
     InstanceGroups: NotRequired[Sequence[InstanceGroupTypeDef]]
+    TrainingPlanArn: NotRequired[str]
 
 
 class ParameterRangeOutputTypeDef(TypedDict):
@@ -8124,12 +8562,24 @@ class ListClusterNodesRequestListClusterNodesPaginateTypeDef(TypedDict):
     PaginationConfig: NotRequired[PaginatorConfigTypeDef]
 
 
+class ListClusterSchedulerConfigsRequestListClusterSchedulerConfigsPaginateTypeDef(TypedDict):
+    CreatedAfter: NotRequired[TimestampTypeDef]
+    CreatedBefore: NotRequired[TimestampTypeDef]
+    NameContains: NotRequired[str]
+    ClusterArn: NotRequired[str]
+    Status: NotRequired[SchedulerResourceStatusType]
+    SortBy: NotRequired[SortClusterSchedulerConfigByType]
+    SortOrder: NotRequired[SortOrderType]
+    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
+
+
 class ListClustersRequestListClustersPaginateTypeDef(TypedDict):
     CreationTimeAfter: NotRequired[TimestampTypeDef]
     CreationTimeBefore: NotRequired[TimestampTypeDef]
     NameContains: NotRequired[str]
     SortBy: NotRequired[ClusterSortByType]
     SortOrder: NotRequired[SortOrderType]
+    TrainingPlanArn: NotRequired[str]
     PaginationConfig: NotRequired[PaginatorConfigTypeDef]
 
 
@@ -8156,6 +8606,17 @@ class ListCompilationJobsRequestListCompilationJobsPaginateTypeDef(TypedDict):
     PaginationConfig: NotRequired[PaginatorConfigTypeDef]
 
 
+class ListComputeQuotasRequestListComputeQuotasPaginateTypeDef(TypedDict):
+    CreatedAfter: NotRequired[TimestampTypeDef]
+    CreatedBefore: NotRequired[TimestampTypeDef]
+    NameContains: NotRequired[str]
+    Status: NotRequired[SchedulerResourceStatusType]
+    ClusterArn: NotRequired[str]
+    SortBy: NotRequired[SortQuotaByType]
+    SortOrder: NotRequired[SortOrderType]
+    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
+
+
 class ListContextsRequestListContextsPaginateTypeDef(TypedDict):
     SourceUri: NotRequired[str]
     ContextType: NotRequired[str]
@@ -8591,6 +9052,10 @@ class ListOptimizationJobsRequestListOptimizationJobsPaginateTypeDef(TypedDict):
     PaginationConfig: NotRequired[PaginatorConfigTypeDef]
 
 
+class ListPartnerAppsRequestListPartnerAppsPaginateTypeDef(TypedDict):
+    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
+
+
 class ListPipelineExecutionStepsRequestListPipelineExecutionStepsPaginateTypeDef(TypedDict):
     PipelineExecutionArn: NotRequired[str]
     SortOrder: NotRequired[SortOrderType]
@@ -8700,6 +9165,7 @@ class ListTrainingJobsRequestListTrainingJobsPaginateTypeDef(TypedDict):
     SortBy: NotRequired[SortByType]
     SortOrder: NotRequired[SortOrderType]
     WarmPoolStatusEquals: NotRequired[WarmPoolResourceStatusType]
+    TrainingPlanArnEquals: NotRequired[str]
     PaginationConfig: NotRequired[PaginatorConfigTypeDef]
 
 
@@ -8860,6 +9326,12 @@ class ListOptimizationJobsResponseTypeDef(TypedDict):
     NextToken: NotRequired[str]
 
 
+class ListPartnerAppsResponseTypeDef(TypedDict):
+    Summaries: List[PartnerAppSummaryTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+    NextToken: NotRequired[str]
+
+
 class ListPipelineExecutionsResponseTypeDef(TypedDict):
     PipelineExecutionSummaries: List[PipelineExecutionSummaryTypeDef]
     ResponseMetadata: ResponseMetadataTypeDef
@@ -8902,6 +9374,25 @@ class ListStudioLifecycleConfigsResponseTypeDef(TypedDict):
     NextToken: NotRequired[str]
 
 
+class ListTrainingPlansRequestListTrainingPlansPaginateTypeDef(TypedDict):
+    StartTimeAfter: NotRequired[TimestampTypeDef]
+    StartTimeBefore: NotRequired[TimestampTypeDef]
+    SortBy: NotRequired[TrainingPlanSortByType]
+    SortOrder: NotRequired[TrainingPlanSortOrderType]
+    Filters: NotRequired[Sequence[TrainingPlanFilterTypeDef]]
+    PaginationConfig: NotRequired[PaginatorConfigTypeDef]
+
+
+class ListTrainingPlansRequestRequestTypeDef(TypedDict):
+    NextToken: NotRequired[str]
+    MaxResults: NotRequired[int]
+    StartTimeAfter: NotRequired[TimestampTypeDef]
+    StartTimeBefore: NotRequired[TimestampTypeDef]
+    SortBy: NotRequired[TrainingPlanSortByType]
+    SortOrder: NotRequired[TrainingPlanSortOrderType]
+    Filters: NotRequired[Sequence[TrainingPlanFilterTypeDef]]
+
+
 class ListTransformJobsResponseTypeDef(TypedDict):
     TransformJobSummaries: List[TransformJobSummaryTypeDef]
     ResponseMetadata: ResponseMetadataTypeDef
@@ -8982,13 +9473,19 @@ class ModelPackageStatusDetailsTypeDef(TypedDict):
     ImageScanStatuses: NotRequired[List[ModelPackageStatusItemTypeDef]]
 
 
+ModelQuantizationConfigUnionTypeDef = Union[
+    ModelQuantizationConfigTypeDef, ModelQuantizationConfigOutputTypeDef
+]
+
+
 class OptimizationConfigOutputTypeDef(TypedDict):
     ModelQuantizationConfig: NotRequired[ModelQuantizationConfigOutputTypeDef]
     ModelCompilationConfig: NotRequired[ModelCompilationConfigOutputTypeDef]
+    ModelShardingConfig: NotRequired[ModelShardingConfigOutputTypeDef]
 
 
-ModelQuantizationConfigUnionTypeDef = Union[
-    ModelQuantizationConfigTypeDef, ModelQuantizationConfigOutputTypeDef
+ModelShardingConfigUnionTypeDef = Union[
+    ModelShardingConfigTypeDef, ModelShardingConfigOutputTypeDef
 ]
 MonitoringAppSpecificationUnionTypeDef = Union[
     MonitoringAppSpecificationTypeDef, MonitoringAppSpecificationOutputTypeDef
@@ -9081,6 +9578,16 @@ class ProductionVariantSummaryTypeDef(TypedDict):
     RoutingConfig: NotRequired[ProductionVariantRoutingConfigTypeDef]
 
 
+class SchedulerConfigOutputTypeDef(TypedDict):
+    PriorityClasses: NotRequired[List[PriorityClassTypeDef]]
+    FairShare: NotRequired[FairShareType]
+
+
+class SchedulerConfigTypeDef(TypedDict):
+    PriorityClasses: NotRequired[Sequence[PriorityClassTypeDef]]
+    FairShare: NotRequired[FairShareType]
+
+
 class ProcessingResourcesTypeDef(TypedDict):
     ClusterConfig: ProcessingClusterConfigTypeDef
 
@@ -9188,6 +9695,18 @@ class RenderUiTemplateResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
+class TrainingPlanOfferingTypeDef(TypedDict):
+    TrainingPlanOfferingId: str
+    TargetResources: List[SageMakerResourceNameType]
+    RequestedStartTimeAfter: NotRequired[datetime]
+    RequestedEndTimeBefore: NotRequired[datetime]
+    DurationHours: NotRequired[int]
+    DurationMinutes: NotRequired[int]
+    UpfrontFee: NotRequired[str]
+    CurrencyCode: NotRequired[str]
+    ReservedCapacityOfferings: NotRequired[List[ReservedCapacityOfferingTypeDef]]
+
+
 class UpdateTrainingJobRequestRequestTypeDef(TypedDict):
     TrainingJobName: str
     ProfilerConfig: NotRequired[ProfilerConfigForUpdateTypeDef]
@@ -9540,17 +10059,10 @@ class ClusterInstanceGroupDetailsTypeDef(TypedDict):
     ThreadsPerCore: NotRequired[int]
     InstanceStorageConfigs: NotRequired[List[ClusterInstanceStorageConfigTypeDef]]
     OnStartDeepHealthChecks: NotRequired[List[DeepHealthCheckTypeType]]
-
-
-class ClusterInstanceGroupSpecificationTypeDef(TypedDict):
-    InstanceCount: int
-    InstanceGroupName: str
-    InstanceType: ClusterInstanceTypeType
-    LifeCycleConfig: ClusterLifeCycleConfigTypeDef
-    ExecutionRole: str
-    ThreadsPerCore: NotRequired[int]
-    InstanceStorageConfigs: NotRequired[Sequence[ClusterInstanceStorageConfigTypeDef]]
-    OnStartDeepHealthChecks: NotRequired[Sequence[DeepHealthCheckTypeType]]
+    Status: NotRequired[InstanceGroupStatusType]
+    TrainingPlanArn: NotRequired[str]
+    TrainingPlanStatus: NotRequired[str]
+    OverrideVpcConfig: NotRequired[VpcConfigOutputTypeDef]
 
 
 class ClusterNodeDetailsTypeDef(TypedDict):
@@ -9560,6 +10072,7 @@ class ClusterNodeDetailsTypeDef(TypedDict):
     InstanceType: NotRequired[ClusterInstanceTypeType]
     LaunchTime: NotRequired[datetime]
     LifeCycleConfig: NotRequired[ClusterLifeCycleConfigTypeDef]
+    OverrideVpcConfig: NotRequired[VpcConfigOutputTypeDef]
     ThreadsPerCore: NotRequired[int]
     InstanceStorageConfigs: NotRequired[List[ClusterInstanceStorageConfigTypeDef]]
     PrivatePrimaryIp: NotRequired[str]
@@ -9606,6 +10119,39 @@ class DebugHookConfigTypeDef(TypedDict):
     CollectionConfigurations: NotRequired[Sequence[CollectionConfigurationUnionTypeDef]]
 
 
+class ComputeQuotaSummaryTypeDef(TypedDict):
+    ComputeQuotaArn: str
+    ComputeQuotaId: str
+    Name: str
+    Status: SchedulerResourceStatusType
+    ComputeQuotaTarget: ComputeQuotaTargetTypeDef
+    CreationTime: datetime
+    ComputeQuotaVersion: NotRequired[int]
+    ClusterArn: NotRequired[str]
+    ComputeQuotaConfig: NotRequired[ComputeQuotaConfigOutputTypeDef]
+    ActivationState: NotRequired[ActivationStateType]
+    LastModifiedTime: NotRequired[datetime]
+
+
+class CreateComputeQuotaRequestRequestTypeDef(TypedDict):
+    Name: str
+    ClusterArn: str
+    ComputeQuotaConfig: ComputeQuotaConfigTypeDef
+    ComputeQuotaTarget: ComputeQuotaTargetTypeDef
+    Description: NotRequired[str]
+    ActivationState: NotRequired[ActivationStateType]
+    Tags: NotRequired[Sequence[TagTypeDef]]
+
+
+class UpdateComputeQuotaRequestRequestTypeDef(TypedDict):
+    ComputeQuotaId: str
+    TargetVersion: int
+    ComputeQuotaConfig: NotRequired[ComputeQuotaConfigTypeDef]
+    ComputeQuotaTarget: NotRequired[ComputeQuotaTargetTypeDef]
+    ActivationState: NotRequired[ActivationStateType]
+    Description: NotRequired[str]
+
+
 class CodeEditorAppImageConfigTypeDef(TypedDict):
     FileSystemConfig: NotRequired[FileSystemConfigTypeDef]
     ContainerConfig: NotRequired[ContainerConfigUnionTypeDef]
@@ -9628,6 +10174,19 @@ class AutoMLSecurityConfigTypeDef(TypedDict):
     VpcConfig: NotRequired[VpcConfigUnionTypeDef]
 
 
+class ClusterInstanceGroupSpecificationTypeDef(TypedDict):
+    InstanceCount: int
+    InstanceGroupName: str
+    InstanceType: ClusterInstanceTypeType
+    LifeCycleConfig: ClusterLifeCycleConfigTypeDef
+    ExecutionRole: str
+    ThreadsPerCore: NotRequired[int]
+    InstanceStorageConfigs: NotRequired[Sequence[ClusterInstanceStorageConfigTypeDef]]
+    OnStartDeepHealthChecks: NotRequired[Sequence[DeepHealthCheckTypeType]]
+    TrainingPlanArn: NotRequired[str]
+    OverrideVpcConfig: NotRequired[VpcConfigUnionTypeDef]
+
+
 class LabelingJobResourceConfigTypeDef(TypedDict):
     VolumeKmsKeyId: NotRequired[str]
     VpcConfig: NotRequired[VpcConfigUnionTypeDef]
@@ -9685,6 +10244,7 @@ InferenceComponentSpecificationSummaryTypeDef = TypedDict(
         "ComputeResourceRequirements": NotRequired[
             InferenceComponentComputeResourceRequirementsTypeDef
         ],
+        "BaseInferenceComponentName": NotRequired[str],
     },
 )
 
@@ -9728,6 +10288,12 @@ class ListTrainingJobsResponseTypeDef(TypedDict):
     NextToken: NotRequired[str]
 
 
+class ListTrainingPlansResponseTypeDef(TypedDict):
+    TrainingPlanSummaries: List[TrainingPlanSummaryTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+    NextToken: NotRequired[str]
+
+
 class ListTrialsResponseTypeDef(TypedDict):
     TrialSummaries: List[TrialSummaryTypeDef]
     ResponseMetadata: ResponseMetadataTypeDef
@@ -9892,6 +10458,25 @@ class DescribeArtifactResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
+class DescribeComputeQuotaResponseTypeDef(TypedDict):
+    ComputeQuotaArn: str
+    ComputeQuotaId: str
+    Name: str
+    Description: str
+    ComputeQuotaVersion: int
+    Status: SchedulerResourceStatusType
+    FailureReason: str
+    ClusterArn: str
+    ComputeQuotaConfig: ComputeQuotaConfigOutputTypeDef
+    ComputeQuotaTarget: ComputeQuotaTargetTypeDef
+    ActivationState: ActivationStateType
+    CreationTime: datetime
+    CreatedBy: UserContextTypeDef
+    LastModifiedTime: datetime
+    LastModifiedBy: UserContextTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class DescribeContextResponseTypeDef(TypedDict):
     ContextName: str
     ContextArn: str
@@ -10129,9 +10714,9 @@ class WorkerAccessConfigurationTypeDef(TypedDict):
 class CreateInferenceComponentInputRequestTypeDef(TypedDict):
     InferenceComponentName: str
     EndpointName: str
-    VariantName: str
     Specification: InferenceComponentSpecificationTypeDef
-    RuntimeConfig: InferenceComponentRuntimeConfigTypeDef
+    VariantName: NotRequired[str]
+    RuntimeConfig: NotRequired[InferenceComponentRuntimeConfigTypeDef]
     Tags: NotRequired[Sequence[TagTypeDef]]
 
 
@@ -10260,6 +10845,7 @@ class ListModelMetadataRequestRequestTypeDef(TypedDict):
 class OptimizationConfigTypeDef(TypedDict):
     ModelQuantizationConfig: NotRequired[ModelQuantizationConfigUnionTypeDef]
     ModelCompilationConfig: NotRequired[ModelCompilationConfigUnionTypeDef]
+    ModelShardingConfig: NotRequired[ModelShardingConfigUnionTypeDef]
 
 
 BatchTransformInputOutputTypeDef = TypedDict(
@@ -10343,6 +10929,38 @@ class PendingDeploymentSummaryTypeDef(TypedDict):
     ShadowProductionVariants: NotRequired[List[PendingProductionVariantSummaryTypeDef]]
 
 
+class DescribeClusterSchedulerConfigResponseTypeDef(TypedDict):
+    ClusterSchedulerConfigArn: str
+    ClusterSchedulerConfigId: str
+    Name: str
+    ClusterSchedulerConfigVersion: int
+    Status: SchedulerResourceStatusType
+    FailureReason: str
+    ClusterArn: str
+    SchedulerConfig: SchedulerConfigOutputTypeDef
+    Description: str
+    CreationTime: datetime
+    CreatedBy: UserContextTypeDef
+    LastModifiedTime: datetime
+    LastModifiedBy: UserContextTypeDef
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
+class CreateClusterSchedulerConfigRequestRequestTypeDef(TypedDict):
+    Name: str
+    ClusterArn: str
+    SchedulerConfig: SchedulerConfigTypeDef
+    Description: NotRequired[str]
+    Tags: NotRequired[Sequence[TagTypeDef]]
+
+
+class UpdateClusterSchedulerConfigRequestRequestTypeDef(TypedDict):
+    ClusterSchedulerConfigId: str
+    TargetVersion: int
+    SchedulerConfig: NotRequired[SchedulerConfigTypeDef]
+    Description: NotRequired[str]
+
+
 class ProcessingOutputConfigOutputTypeDef(TypedDict):
     Outputs: List[ProcessingOutputTypeDef]
     KmsKeyId: NotRequired[str]
@@ -10473,6 +11091,11 @@ class RecommendationJobContainerConfigTypeDef(TypedDict):
     SupportedResponseMIMETypes: NotRequired[Sequence[str]]
 
 
+class SearchTrainingPlanOfferingsResponseTypeDef(TypedDict):
+    TrainingPlanOfferings: List[TrainingPlanOfferingTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+
+
 class DataSourceTypeDef(TypedDict):
     S3DataSource: NotRequired[S3DataSourceUnionTypeDef]
     FileSystemDataSource: NotRequired[FileSystemDataSourceTypeDef]
@@ -10750,21 +11373,6 @@ class DescribeClusterResponseTypeDef(TypedDict):
     ResponseMetadata: ResponseMetadataTypeDef
 
 
-class CreateClusterRequestRequestTypeDef(TypedDict):
-    ClusterName: str
-    InstanceGroups: Sequence[ClusterInstanceGroupSpecificationTypeDef]
-    VpcConfig: NotRequired[VpcConfigTypeDef]
-    Tags: NotRequired[Sequence[TagTypeDef]]
-    Orchestrator: NotRequired[ClusterOrchestratorTypeDef]
-    NodeRecovery: NotRequired[ClusterNodeRecoveryType]
-
-
-class UpdateClusterRequestRequestTypeDef(TypedDict):
-    ClusterName: str
-    InstanceGroups: Sequence[ClusterInstanceGroupSpecificationTypeDef]
-    NodeRecovery: NotRequired[ClusterNodeRecoveryType]
-
-
 class DescribeClusterNodeResponseTypeDef(TypedDict):
     NodeDetails: ClusterNodeDetailsTypeDef
     ResponseMetadata: ResponseMetadataTypeDef
@@ -10831,6 +11439,12 @@ class UpdateFeatureGroupRequestRequestTypeDef(TypedDict):
     ThroughputConfig: NotRequired[ThroughputConfigUpdateTypeDef]
 
 
+class ListComputeQuotasResponseTypeDef(TypedDict):
+    ComputeQuotaSummaries: List[ComputeQuotaSummaryTypeDef]
+    ResponseMetadata: ResponseMetadataTypeDef
+    NextToken: NotRequired[str]
+
+
 class CreateAppImageConfigRequestRequestTypeDef(TypedDict):
     AppImageConfigName: str
     Tags: NotRequired[Sequence[TagTypeDef]]
@@ -10849,6 +11463,23 @@ class UpdateAppImageConfigRequestRequestTypeDef(TypedDict):
 AutoMLSecurityConfigUnionTypeDef = Union[
     AutoMLSecurityConfigTypeDef, AutoMLSecurityConfigOutputTypeDef
 ]
+
+
+class CreateClusterRequestRequestTypeDef(TypedDict):
+    ClusterName: str
+    InstanceGroups: Sequence[ClusterInstanceGroupSpecificationTypeDef]
+    VpcConfig: NotRequired[VpcConfigTypeDef]
+    Tags: NotRequired[Sequence[TagTypeDef]]
+    Orchestrator: NotRequired[ClusterOrchestratorTypeDef]
+    NodeRecovery: NotRequired[ClusterNodeRecoveryType]
+
+
+class UpdateClusterRequestRequestTypeDef(TypedDict):
+    ClusterName: str
+    InstanceGroups: Sequence[ClusterInstanceGroupSpecificationTypeDef]
+    NodeRecovery: NotRequired[ClusterNodeRecoveryType]
+
+
 LabelingJobResourceConfigUnionTypeDef = Union[
     LabelingJobResourceConfigTypeDef, LabelingJobResourceConfigOutputTypeDef
 ]