mypy-boto3-sagemaker 1.34.122__py3-none-any.whl → 1.34.129__py3-none-any.whl

mypy_boto3_sagemaker/type_defs.pyi

@@ -128,6 +128,7 @@ from .literals import (
     InferenceExperimentStopDesiredStateType,
     InputModeType,
     InstanceTypeType,
+    IsTrackingServerActiveType,
     JobTypeType,
     JoinSourceType,
     LabelingJobStatusType,
@@ -233,6 +234,7 @@ from .literals import (
     SortOrderType,
     SortPipelineExecutionsByType,
     SortPipelinesByType,
+    SortTrackingServerByType,
     SortTrialComponentsByType,
     SortTrialsByType,
     SpaceSortKeyType,
@@ -251,6 +253,8 @@ from .literals import (
     TargetPlatformArchType,
     TargetPlatformOsType,
     ThroughputModeType,
+    TrackingServerSizeType,
+    TrackingServerStatusType,
     TrafficRoutingConfigTypeType,
     TrafficTypeType,
     TrainingInputModeType,
@@ -362,6 +366,7 @@ __all__ = (
     "ClarifyInferenceConfigTypeDef",
     "ClarifyShapBaselineConfigTypeDef",
     "ClarifyTextConfigTypeDef",
+    "ClusterEbsVolumeConfigTypeDef",
     "ClusterLifeCycleConfigTypeDef",
     "ClusterInstancePlacementTypeDef",
     "ClusterInstanceStatusDetailsTypeDef",
@@ -417,6 +422,7 @@ __all__ = (
     "ParallelismConfigurationTypeDef",
     "PipelineDefinitionS3LocationTypeDef",
     "CreatePresignedDomainUrlRequestRequestTypeDef",
+    "CreatePresignedMlflowTrackingServerUrlRequestRequestTypeDef",
     "CreatePresignedNotebookInstanceUrlInputRequestTypeDef",
     "ExperimentConfigTypeDef",
     "ProcessingStoppingConditionTypeDef",
@@ -488,6 +494,7 @@ __all__ = (
     "DeleteImageVersionRequestRequestTypeDef",
     "DeleteInferenceComponentInputRequestTypeDef",
     "DeleteInferenceExperimentRequestRequestTypeDef",
+    "DeleteMlflowTrackingServerRequestRequestTypeDef",
     "DeleteModelBiasJobDefinitionRequestRequestTypeDef",
     "DeleteModelCardRequestRequestTypeDef",
     "DeleteModelExplainabilityJobDefinitionRequestRequestTypeDef",
@@ -575,6 +582,7 @@ __all__ = (
     "LabelCountersTypeDef",
     "LabelingJobOutputTypeDef",
     "DescribeLineageGroupRequestRequestTypeDef",
+    "DescribeMlflowTrackingServerRequestRequestTypeDef",
     "DescribeModelBiasJobDefinitionRequestRequestTypeDef",
     "ModelBiasAppSpecificationOutputTypeDef",
     "DescribeModelCardExportJobRequestRequestTypeDef",
@@ -694,6 +702,7 @@ __all__ = (
     "MonitoringJobDefinitionSummaryTypeDef",
     "ListDomainsRequestRequestTypeDef",
     "ListInferenceRecommendationsJobStepsRequestRequestTypeDef",
+    "TrackingServerSummaryTypeDef",
     "ModelCardExportJobSummaryTypeDef",
     "ModelCardVersionSummaryTypeDef",
     "ModelCardSummaryTypeDef",
@@ -800,6 +809,7 @@ __all__ = (
     "StairsTypeDef",
     "StartEdgeDeploymentStageRequestRequestTypeDef",
     "StartInferenceExperimentRequestRequestTypeDef",
+    "StartMlflowTrackingServerRequestRequestTypeDef",
     "StartMonitoringScheduleRequestRequestTypeDef",
     "StartNotebookInstanceInputRequestTypeDef",
     "StopAutoMLJobRequestRequestTypeDef",
@@ -809,6 +819,7 @@ __all__ = (
     "StopHyperParameterTuningJobRequestRequestTypeDef",
     "StopInferenceRecommendationsJobRequestRequestTypeDef",
     "StopLabelingJobRequestRequestTypeDef",
+    "StopMlflowTrackingServerRequestRequestTypeDef",
     "StopMonitoringScheduleRequestRequestTypeDef",
     "StopNotebookInstanceInputRequestTypeDef",
     "StopPipelineExecutionRequestRequestTypeDef",
@@ -831,6 +842,7 @@ __all__ = (
     "UpdateHubRequestRequestTypeDef",
     "UpdateImageRequestRequestTypeDef",
     "UpdateImageVersionRequestRequestTypeDef",
+    "UpdateMlflowTrackingServerRequestRequestTypeDef",
     "UpdateModelCardRequestRequestTypeDef",
     "UpdateMonitoringAlertRequestRequestTypeDef",
     "UpdateTrialRequestRequestTypeDef",
@@ -866,6 +878,7 @@ __all__ = (
     "CreateInferenceExperimentResponseTypeDef",
     "CreateInferenceRecommendationsJobResponseTypeDef",
     "CreateLabelingJobResponseTypeDef",
+    "CreateMlflowTrackingServerResponseTypeDef",
     "CreateModelBiasJobDefinitionResponseTypeDef",
     "CreateModelCardExportJobResponseTypeDef",
     "CreateModelCardResponseTypeDef",
@@ -879,6 +892,7 @@ __all__ = (
     "CreateNotebookInstanceOutputTypeDef",
     "CreatePipelineResponseTypeDef",
     "CreatePresignedDomainUrlResponseTypeDef",
+    "CreatePresignedMlflowTrackingServerUrlResponseTypeDef",
     "CreatePresignedNotebookInstanceUrlOutputTypeDef",
     "CreateProcessingJobResponseTypeDef",
     "CreateProjectOutputTypeDef",
@@ -898,6 +912,7 @@ __all__ = (
     "DeleteContextResponseTypeDef",
     "DeleteExperimentResponseTypeDef",
     "DeleteInferenceExperimentResponseTypeDef",
+    "DeleteMlflowTrackingServerResponseTypeDef",
     "DeletePipelineResponseTypeDef",
     "DeleteTrialComponentResponseTypeDef",
     "DeleteTrialResponseTypeDef",
@@ -918,8 +933,10 @@ __all__ = (
     "SendPipelineExecutionStepFailureResponseTypeDef",
     "SendPipelineExecutionStepSuccessResponseTypeDef",
     "StartInferenceExperimentResponseTypeDef",
+    "StartMlflowTrackingServerResponseTypeDef",
     "StartPipelineExecutionResponseTypeDef",
     "StopInferenceExperimentResponseTypeDef",
+    "StopMlflowTrackingServerResponseTypeDef",
     "StopPipelineExecutionResponseTypeDef",
     "UpdateActionResponseTypeDef",
     "UpdateAppImageConfigResponseTypeDef",
@@ -939,6 +956,7 @@ __all__ = (
     "UpdateInferenceComponentOutputTypeDef",
     "UpdateInferenceComponentRuntimeConfigOutputTypeDef",
     "UpdateInferenceExperimentResponseTypeDef",
+    "UpdateMlflowTrackingServerResponseTypeDef",
     "UpdateModelCardResponseTypeDef",
     "UpdateModelPackageOutputTypeDef",
     "UpdateMonitoringAlertResponseTypeDef",
@@ -955,6 +973,7 @@ __all__ = (
     "AddTagsOutputTypeDef",
     "CreateExperimentRequestRequestTypeDef",
     "CreateImageRequestRequestTypeDef",
+    "CreateMlflowTrackingServerRequestRequestTypeDef",
     "CreateModelPackageGroupInputRequestTypeDef",
     "CreateStudioLifecycleConfigRequestRequestTypeDef",
     "ImportHubContentRequestRequestTypeDef",
@@ -1018,9 +1037,7 @@ __all__ = (
     "EnvironmentParameterRangesOutputTypeDef",
     "EnvironmentParameterRangesTypeDef",
     "ClarifyShapConfigTypeDef",
-    "ClusterInstanceGroupDetailsTypeDef",
-    "ClusterInstanceGroupSpecificationTypeDef",
-    "ClusterNodeDetailsTypeDef",
+    "ClusterInstanceStorageConfigTypeDef",
     "ClusterNodeSummaryTypeDef",
     "ListClustersResponseTypeDef",
     "CodeEditorAppImageConfigExtraOutputTypeDef",
@@ -1109,6 +1126,7 @@ __all__ = (
     "ListLabelingJobsForWorkteamRequestRequestTypeDef",
     "ListLabelingJobsRequestRequestTypeDef",
     "ListLineageGroupsRequestRequestTypeDef",
+    "ListMlflowTrackingServersRequestRequestTypeDef",
     "ListModelBiasJobDefinitionsRequestRequestTypeDef",
     "ListModelCardExportJobsRequestRequestTypeDef",
     "ListModelCardVersionsRequestRequestTypeDef",
@@ -1287,6 +1305,7 @@ __all__ = (
     "ListLabelingJobsForWorkteamRequestListLabelingJobsForWorkteamPaginateTypeDef",
     "ListLabelingJobsRequestListLabelingJobsPaginateTypeDef",
     "ListLineageGroupsRequestListLineageGroupsPaginateTypeDef",
+    "ListMlflowTrackingServersRequestListMlflowTrackingServersPaginateTypeDef",
     "ListModelBiasJobDefinitionsRequestListModelBiasJobDefinitionsPaginateTypeDef",
     "ListModelCardExportJobsRequestListModelCardExportJobsPaginateTypeDef",
     "ListModelCardVersionsRequestListModelCardVersionsPaginateTypeDef",
@@ -1325,6 +1344,7 @@ __all__ = (
     "ListModelBiasJobDefinitionsResponseTypeDef",
     "ListModelExplainabilityJobDefinitionsResponseTypeDef",
     "ListModelQualityJobDefinitionsResponseTypeDef",
+    "ListMlflowTrackingServersResponseTypeDef",
     "ListModelCardExportJobsResponseTypeDef",
     "ListModelCardVersionsResponseTypeDef",
     "ListModelCardsResponseTypeDef",
@@ -1433,10 +1453,9 @@ __all__ = (
     "EndpointInputConfigurationTypeDef",
     "ClarifyExplainerConfigOutputTypeDef",
     "ClarifyExplainerConfigTypeDef",
-    "DescribeClusterResponseTypeDef",
-    "CreateClusterRequestRequestTypeDef",
-    "UpdateClusterRequestRequestTypeDef",
-    "DescribeClusterNodeResponseTypeDef",
+    "ClusterInstanceGroupDetailsTypeDef",
+    "ClusterInstanceGroupSpecificationTypeDef",
+    "ClusterNodeDetailsTypeDef",
     "ListClusterNodesResponseTypeDef",
     "CodeEditorAppImageConfigUnionTypeDef",
     "JupyterLabAppImageConfigUnionTypeDef",
@@ -1482,6 +1501,7 @@ __all__ = (
     "DescribeContextResponseTypeDef",
     "DescribeExperimentResponseTypeDef",
     "DescribeLineageGroupResponseTypeDef",
+    "DescribeMlflowTrackingServerResponseTypeDef",
     "DescribeModelCardResponseTypeDef",
     "DescribeModelPackageGroupOutputTypeDef",
     "DescribePipelineResponseTypeDef",
@@ -1573,6 +1593,10 @@ __all__ = (
     "RecommendationJobInputConfigTypeDef",
     "ExplainerConfigOutputTypeDef",
     "ExplainerConfigTypeDef",
+    "DescribeClusterResponseTypeDef",
+    "CreateClusterRequestRequestTypeDef",
+    "UpdateClusterRequestRequestTypeDef",
+    "DescribeClusterNodeResponseTypeDef",
     "CreateFeatureGroupRequestRequestTypeDef",
     "DescribeFeatureGroupResponseTypeDef",
     "FeatureGroupTypeDef",
@@ -2329,6 +2353,12 @@ ClarifyTextConfigTypeDef = TypedDict(
         "Granularity": ClarifyTextGranularityType,
     },
 )
+ClusterEbsVolumeConfigTypeDef = TypedDict(
+    "ClusterEbsVolumeConfigTypeDef",
+    {
+        "VolumeSizeInGB": int,
+    },
+)
 ClusterLifeCycleConfigTypeDef = TypedDict(
     "ClusterLifeCycleConfigTypeDef",
     {
@@ -2752,6 +2782,14 @@ CreatePresignedDomainUrlRequestRequestTypeDef = TypedDict(
         "LandingUri": NotRequired[str],
     },
 )
+CreatePresignedMlflowTrackingServerUrlRequestRequestTypeDef = TypedDict(
+    "CreatePresignedMlflowTrackingServerUrlRequestRequestTypeDef",
+    {
+        "TrackingServerName": str,
+        "ExpiresInSeconds": NotRequired[int],
+        "SessionExpirationDurationInSeconds": NotRequired[int],
+    },
+)
 CreatePresignedNotebookInstanceUrlInputRequestTypeDef = TypedDict(
     "CreatePresignedNotebookInstanceUrlInputRequestTypeDef",
     {
@@ -2899,6 +2937,8 @@ OidcConfigTypeDef = TypedDict(
         "UserInfoEndpoint": str,
         "LogoutEndpoint": str,
         "JwksUri": str,
+        "Scope": NotRequired[str],
+        "AuthenticationRequestExtraParams": NotRequired[Mapping[str, str]],
     },
 )
 SourceIpConfigTypeDef = TypedDict(
@@ -3285,6 +3325,12 @@ DeleteInferenceExperimentRequestRequestTypeDef = TypedDict(
         "Name": str,
     },
 )
+DeleteMlflowTrackingServerRequestRequestTypeDef = TypedDict(
+    "DeleteMlflowTrackingServerRequestRequestTypeDef",
+    {
+        "TrackingServerName": str,
+    },
+)
 DeleteModelBiasJobDefinitionRequestRequestTypeDef = TypedDict(
     "DeleteModelBiasJobDefinitionRequestRequestTypeDef",
     {
@@ -3876,6 +3922,12 @@ DescribeLineageGroupRequestRequestTypeDef = TypedDict(
         "LineageGroupName": str,
     },
 )
+DescribeMlflowTrackingServerRequestRequestTypeDef = TypedDict(
+    "DescribeMlflowTrackingServerRequestRequestTypeDef",
+    {
+        "TrackingServerName": str,
+    },
+)
 DescribeModelBiasJobDefinitionRequestRequestTypeDef = TypedDict(
     "DescribeModelBiasJobDefinitionRequestRequestTypeDef",
     {
@@ -4835,6 +4887,18 @@ ListInferenceRecommendationsJobStepsRequestRequestTypeDef = TypedDict(
         "NextToken": NotRequired[str],
     },
 )
+TrackingServerSummaryTypeDef = TypedDict(
+    "TrackingServerSummaryTypeDef",
+    {
+        "TrackingServerArn": NotRequired[str],
+        "TrackingServerName": NotRequired[str],
+        "CreationTime": NotRequired[datetime],
+        "LastModifiedTime": NotRequired[datetime],
+        "TrackingServerStatus": NotRequired[TrackingServerStatusType],
+        "IsActive": NotRequired[IsTrackingServerActiveType],
+        "MlflowVersion": NotRequired[str],
+    },
+)
 ModelCardExportJobSummaryTypeDef = TypedDict(
     "ModelCardExportJobSummaryTypeDef",
     {
@@ -5333,6 +5397,8 @@ OidcConfigForResponseTypeDef = TypedDict(
         "UserInfoEndpoint": NotRequired[str],
         "LogoutEndpoint": NotRequired[str],
         "JwksUri": NotRequired[str],
+        "Scope": NotRequired[str],
+        "AuthenticationRequestExtraParams": NotRequired[Dict[str, str]],
     },
 )
 OnlineStoreSecurityConfigTypeDef = TypedDict(
@@ -5705,6 +5771,12 @@ StartInferenceExperimentRequestRequestTypeDef = TypedDict(
         "Name": str,
     },
 )
+StartMlflowTrackingServerRequestRequestTypeDef = TypedDict(
+    "StartMlflowTrackingServerRequestRequestTypeDef",
+    {
+        "TrackingServerName": str,
+    },
+)
 StartMonitoringScheduleRequestRequestTypeDef = TypedDict(
     "StartMonitoringScheduleRequestRequestTypeDef",
     {
@@ -5760,6 +5832,12 @@ StopLabelingJobRequestRequestTypeDef = TypedDict(
         "LabelingJobName": str,
     },
 )
+StopMlflowTrackingServerRequestRequestTypeDef = TypedDict(
+    "StopMlflowTrackingServerRequestRequestTypeDef",
+    {
+        "TrackingServerName": str,
+    },
+)
 StopMonitoringScheduleRequestRequestTypeDef = TypedDict(
     "StopMonitoringScheduleRequestRequestTypeDef",
     {
@@ -5934,6 +6012,16 @@ UpdateImageVersionRequestRequestTypeDef = TypedDict(
         "ReleaseNotes": NotRequired[str],
     },
 )
+UpdateMlflowTrackingServerRequestRequestTypeDef = TypedDict(
+    "UpdateMlflowTrackingServerRequestRequestTypeDef",
+    {
+        "TrackingServerName": str,
+        "ArtifactStoreUri": NotRequired[str],
+        "TrackingServerSize": NotRequired[TrackingServerSizeType],
+        "AutomaticModelRegistration": NotRequired[bool],
+        "WeeklyMaintenanceWindowStart": NotRequired[str],
+    },
+)
 UpdateModelCardRequestRequestTypeDef = TypedDict(
     "UpdateModelCardRequestRequestTypeDef",
     {
@@ -6192,6 +6280,13 @@ CreateLabelingJobResponseTypeDef = TypedDict(
         "ResponseMetadata": ResponseMetadataTypeDef,
     },
 )
+CreateMlflowTrackingServerResponseTypeDef = TypedDict(
+    "CreateMlflowTrackingServerResponseTypeDef",
+    {
+        "TrackingServerArn": str,
+        "ResponseMetadata": ResponseMetadataTypeDef,
+    },
+)
 CreateModelBiasJobDefinitionResponseTypeDef = TypedDict(
     "CreateModelBiasJobDefinitionResponseTypeDef",
     {
@@ -6283,6 +6378,13 @@ CreatePresignedDomainUrlResponseTypeDef = TypedDict(
         "ResponseMetadata": ResponseMetadataTypeDef,
     },
 )
+CreatePresignedMlflowTrackingServerUrlResponseTypeDef = TypedDict(
+    "CreatePresignedMlflowTrackingServerUrlResponseTypeDef",
+    {
+        "AuthorizedUrl": str,
+        "ResponseMetadata": ResponseMetadataTypeDef,
+    },
+)
 CreatePresignedNotebookInstanceUrlOutputTypeDef = TypedDict(
     "CreatePresignedNotebookInstanceUrlOutputTypeDef",
     {
@@ -6418,6 +6520,13 @@ DeleteInferenceExperimentResponseTypeDef = TypedDict(
         "ResponseMetadata": ResponseMetadataTypeDef,
     },
 )
+DeleteMlflowTrackingServerResponseTypeDef = TypedDict(
+    "DeleteMlflowTrackingServerResponseTypeDef",
+    {
+        "TrackingServerArn": str,
+        "ResponseMetadata": ResponseMetadataTypeDef,
+    },
+)
 DeletePipelineResponseTypeDef = TypedDict(
     "DeletePipelineResponseTypeDef",
     {
@@ -6590,6 +6699,13 @@ StartInferenceExperimentResponseTypeDef = TypedDict(
         "ResponseMetadata": ResponseMetadataTypeDef,
     },
 )
+StartMlflowTrackingServerResponseTypeDef = TypedDict(
+    "StartMlflowTrackingServerResponseTypeDef",
+    {
+        "TrackingServerArn": str,
+        "ResponseMetadata": ResponseMetadataTypeDef,
+    },
+)
 StartPipelineExecutionResponseTypeDef = TypedDict(
     "StartPipelineExecutionResponseTypeDef",
     {
@@ -6604,6 +6720,13 @@ StopInferenceExperimentResponseTypeDef = TypedDict(
         "ResponseMetadata": ResponseMetadataTypeDef,
     },
 )
+StopMlflowTrackingServerResponseTypeDef = TypedDict(
+    "StopMlflowTrackingServerResponseTypeDef",
+    {
+        "TrackingServerArn": str,
+        "ResponseMetadata": ResponseMetadataTypeDef,
+    },
+)
 StopPipelineExecutionResponseTypeDef = TypedDict(
     "StopPipelineExecutionResponseTypeDef",
     {
@@ -6737,6 +6860,13 @@ UpdateInferenceExperimentResponseTypeDef = TypedDict(
         "ResponseMetadata": ResponseMetadataTypeDef,
     },
 )
+UpdateMlflowTrackingServerResponseTypeDef = TypedDict(
+    "UpdateMlflowTrackingServerResponseTypeDef",
+    {
+        "TrackingServerArn": str,
+        "ResponseMetadata": ResponseMetadataTypeDef,
+    },
+)
 UpdateModelCardResponseTypeDef = TypedDict(
     "UpdateModelCardResponseTypeDef",
     {
@@ -6855,6 +6985,19 @@ CreateImageRequestRequestTypeDef = TypedDict(
         "Tags": NotRequired[Sequence[TagTypeDef]],
     },
 )
+CreateMlflowTrackingServerRequestRequestTypeDef = TypedDict(
+    "CreateMlflowTrackingServerRequestRequestTypeDef",
+    {
+        "TrackingServerName": str,
+        "ArtifactStoreUri": str,
+        "RoleArn": str,
+        "TrackingServerSize": NotRequired[TrackingServerSizeType],
+        "MlflowVersion": NotRequired[str],
+        "AutomaticModelRegistration": NotRequired[bool],
+        "WeeklyMaintenanceWindowStart": NotRequired[str],
+        "Tags": NotRequired[Sequence[TagTypeDef]],
+    },
+)
 CreateModelPackageGroupInputRequestTypeDef = TypedDict(
     "CreateModelPackageGroupInputRequestTypeDef",
     {
@@ -7374,42 +7517,10 @@ ClarifyShapConfigTypeDef = TypedDict(
         "TextConfig": NotRequired[ClarifyTextConfigTypeDef],
     },
 )
-ClusterInstanceGroupDetailsTypeDef = TypedDict(
-    "ClusterInstanceGroupDetailsTypeDef",
-    {
-        "CurrentCount": NotRequired[int],
-        "TargetCount": NotRequired[int],
-        "InstanceGroupName": NotRequired[str],
-        "InstanceType": NotRequired[ClusterInstanceTypeType],
-        "LifeCycleConfig": NotRequired[ClusterLifeCycleConfigTypeDef],
-        "ExecutionRole": NotRequired[str],
-        "ThreadsPerCore": NotRequired[int],
-    },
-)
-ClusterInstanceGroupSpecificationTypeDef = TypedDict(
-    "ClusterInstanceGroupSpecificationTypeDef",
-    {
-        "InstanceCount": int,
-        "InstanceGroupName": str,
-        "InstanceType": ClusterInstanceTypeType,
-        "LifeCycleConfig": ClusterLifeCycleConfigTypeDef,
-        "ExecutionRole": str,
-        "ThreadsPerCore": NotRequired[int],
-    },
-)
-ClusterNodeDetailsTypeDef = TypedDict(
-    "ClusterNodeDetailsTypeDef",
+ClusterInstanceStorageConfigTypeDef = TypedDict(
+    "ClusterInstanceStorageConfigTypeDef",
     {
-        "InstanceGroupName": NotRequired[str],
-        "InstanceId": NotRequired[str],
-        "InstanceStatus": NotRequired[ClusterInstanceStatusDetailsTypeDef],
-        "InstanceType": NotRequired[ClusterInstanceTypeType],
-        "LaunchTime": NotRequired[datetime],
-        "LifeCycleConfig": NotRequired[ClusterLifeCycleConfigTypeDef],
-        "ThreadsPerCore": NotRequired[int],
-        "PrivatePrimaryIp": NotRequired[str],
-        "PrivateDnsHostname": NotRequired[str],
-        "Placement": NotRequired[ClusterInstancePlacementTypeDef],
+        "EbsVolumeConfig": NotRequired[ClusterEbsVolumeConfigTypeDef],
     },
 )
 ClusterNodeSummaryTypeDef = TypedDict(
@@ -8415,6 +8526,19 @@ ListLineageGroupsRequestRequestTypeDef = TypedDict(
         "MaxResults": NotRequired[int],
     },
 )
+ListMlflowTrackingServersRequestRequestTypeDef = TypedDict(
+    "ListMlflowTrackingServersRequestRequestTypeDef",
+    {
+        "CreatedAfter": NotRequired[TimestampTypeDef],
+        "CreatedBefore": NotRequired[TimestampTypeDef],
+        "TrackingServerStatus": NotRequired[TrackingServerStatusType],
+        "MlflowVersion": NotRequired[str],
+        "SortBy": NotRequired[SortTrackingServerByType],
+        "SortOrder": NotRequired[SortOrderType],
+        "NextToken": NotRequired[str],
+        "MaxResults": NotRequired[int],
+    },
+)
 ListModelBiasJobDefinitionsRequestRequestTypeDef = TypedDict(
     "ListModelBiasJobDefinitionsRequestRequestTypeDef",
     {
@@ -8492,6 +8616,7 @@ ListModelPackageGroupsInputRequestTypeDef = TypedDict(
         "NextToken": NotRequired[str],
         "SortBy": NotRequired[ModelPackageGroupSortByType],
         "SortOrder": NotRequired[SortOrderType],
+        "CrossAccountFilterOption": NotRequired[CrossAccountFilterOptionType],
     },
 )
 ListModelPackagesInputRequestTypeDef = TypedDict(
@@ -10232,6 +10357,18 @@ ListLineageGroupsRequestListLineageGroupsPaginateTypeDef = TypedDict(
         "PaginationConfig": NotRequired[PaginatorConfigTypeDef],
     },
 )
+ListMlflowTrackingServersRequestListMlflowTrackingServersPaginateTypeDef = TypedDict(
+    "ListMlflowTrackingServersRequestListMlflowTrackingServersPaginateTypeDef",
+    {
+        "CreatedAfter": NotRequired[TimestampTypeDef],
+        "CreatedBefore": NotRequired[TimestampTypeDef],
+        "TrackingServerStatus": NotRequired[TrackingServerStatusType],
+        "MlflowVersion": NotRequired[str],
+        "SortBy": NotRequired[SortTrackingServerByType],
+        "SortOrder": NotRequired[SortOrderType],
+        "PaginationConfig": NotRequired[PaginatorConfigTypeDef],
+    },
+)
 ListModelBiasJobDefinitionsRequestListModelBiasJobDefinitionsPaginateTypeDef = TypedDict(
     "ListModelBiasJobDefinitionsRequestListModelBiasJobDefinitionsPaginateTypeDef",
     {
@@ -10302,6 +10439,7 @@ ListModelPackageGroupsInputListModelPackageGroupsPaginateTypeDef = TypedDict(
         "NameContains": NotRequired[str],
         "SortBy": NotRequired[ModelPackageGroupSortByType],
         "SortOrder": NotRequired[SortOrderType],
+        "CrossAccountFilterOption": NotRequired[CrossAccountFilterOptionType],
         "PaginationConfig": NotRequired[PaginatorConfigTypeDef],
     },
 )
@@ -10661,6 +10799,14 @@ ListModelQualityJobDefinitionsResponseTypeDef = TypedDict(
         "NextToken": NotRequired[str],
     },
 )
+ListMlflowTrackingServersResponseTypeDef = TypedDict(
+    "ListMlflowTrackingServersResponseTypeDef",
+    {
+        "TrackingServerSummaries": List[TrackingServerSummaryTypeDef],
+        "ResponseMetadata": ResponseMetadataTypeDef,
+        "NextToken": NotRequired[str],
+    },
+)
 ListModelCardExportJobsResponseTypeDef = TypedDict(
     "ListModelCardExportJobsResponseTypeDef",
     {
@@ -11580,40 +11726,45 @@ ClarifyExplainerConfigTypeDef = TypedDict(
         "InferenceConfig": NotRequired[ClarifyInferenceConfigTypeDef],
     },
 )
-DescribeClusterResponseTypeDef = TypedDict(
-    "DescribeClusterResponseTypeDef",
-    {
-        "ClusterArn": str,
-        "ClusterName": str,
-        "ClusterStatus": ClusterStatusType,
-        "CreationTime": datetime,
-        "FailureMessage": str,
-        "InstanceGroups": List[ClusterInstanceGroupDetailsTypeDef],
-        "VpcConfig": VpcConfigOutputTypeDef,
-        "ResponseMetadata": ResponseMetadataTypeDef,
-    },
-)
-CreateClusterRequestRequestTypeDef = TypedDict(
-    "CreateClusterRequestRequestTypeDef",
+ClusterInstanceGroupDetailsTypeDef = TypedDict(
+    "ClusterInstanceGroupDetailsTypeDef",
     {
-        "ClusterName": str,
-        "InstanceGroups": Sequence[ClusterInstanceGroupSpecificationTypeDef],
-        "VpcConfig": NotRequired[VpcConfigTypeDef],
-        "Tags": NotRequired[Sequence[TagTypeDef]],
+        "CurrentCount": NotRequired[int],
+        "TargetCount": NotRequired[int],
+        "InstanceGroupName": NotRequired[str],
+        "InstanceType": NotRequired[ClusterInstanceTypeType],
+        "LifeCycleConfig": NotRequired[ClusterLifeCycleConfigTypeDef],
+        "ExecutionRole": NotRequired[str],
+        "ThreadsPerCore": NotRequired[int],
+        "InstanceStorageConfigs": NotRequired[List[ClusterInstanceStorageConfigTypeDef]],
     },
 )
-UpdateClusterRequestRequestTypeDef = TypedDict(
-    "UpdateClusterRequestRequestTypeDef",
+ClusterInstanceGroupSpecificationTypeDef = TypedDict(
+    "ClusterInstanceGroupSpecificationTypeDef",
     {
-        "ClusterName": str,
-        "InstanceGroups": Sequence[ClusterInstanceGroupSpecificationTypeDef],
+        "InstanceCount": int,
+        "InstanceGroupName": str,
+        "InstanceType": ClusterInstanceTypeType,
+        "LifeCycleConfig": ClusterLifeCycleConfigTypeDef,
+        "ExecutionRole": str,
+        "ThreadsPerCore": NotRequired[int],
+        "InstanceStorageConfigs": NotRequired[Sequence[ClusterInstanceStorageConfigTypeDef]],
     },
 )
-DescribeClusterNodeResponseTypeDef = TypedDict(
-    "DescribeClusterNodeResponseTypeDef",
+ClusterNodeDetailsTypeDef = TypedDict(
+    "ClusterNodeDetailsTypeDef",
     {
-        "NodeDetails": ClusterNodeDetailsTypeDef,
-        "ResponseMetadata": ResponseMetadataTypeDef,
+        "InstanceGroupName": NotRequired[str],
+        "InstanceId": NotRequired[str],
+        "InstanceStatus": NotRequired[ClusterInstanceStatusDetailsTypeDef],
+        "InstanceType": NotRequired[ClusterInstanceTypeType],
+        "LaunchTime": NotRequired[datetime],
+        "LifeCycleConfig": NotRequired[ClusterLifeCycleConfigTypeDef],
+        "ThreadsPerCore": NotRequired[int],
+        "InstanceStorageConfigs": NotRequired[List[ClusterInstanceStorageConfigTypeDef]],
+        "PrivatePrimaryIp": NotRequired[str],
+        "PrivateDnsHostname": NotRequired[str],
+        "Placement": NotRequired[ClusterInstancePlacementTypeDef],
     },
 )
 ListClusterNodesResponseTypeDef = TypedDict(
@@ -12073,6 +12224,27 @@ DescribeLineageGroupResponseTypeDef = TypedDict(
         "ResponseMetadata": ResponseMetadataTypeDef,
     },
 )
+DescribeMlflowTrackingServerResponseTypeDef = TypedDict(
+    "DescribeMlflowTrackingServerResponseTypeDef",
+    {
+        "TrackingServerArn": str,
+        "TrackingServerName": str,
+        "ArtifactStoreUri": str,
+        "TrackingServerSize": TrackingServerSizeType,
+        "MlflowVersion": str,
+        "RoleArn": str,
+        "TrackingServerStatus": TrackingServerStatusType,
+        "IsActive": IsTrackingServerActiveType,
+        "TrackingServerUrl": str,
+        "WeeklyMaintenanceWindowStart": str,
+        "AutomaticModelRegistration": bool,
+        "CreationTime": datetime,
+        "CreatedBy": UserContextTypeDef,
+        "LastModifiedTime": datetime,
+        "LastModifiedBy": UserContextTypeDef,
+        "ResponseMetadata": ResponseMetadataTypeDef,
+    },
+)
 DescribeModelCardResponseTypeDef = TypedDict(
     "DescribeModelCardResponseTypeDef",
     {
@@ -13057,6 +13229,42 @@ ExplainerConfigTypeDef = TypedDict(
         "ClarifyExplainerConfig": NotRequired[ClarifyExplainerConfigTypeDef],
     },
 )
+DescribeClusterResponseTypeDef = TypedDict(
+    "DescribeClusterResponseTypeDef",
+    {
+        "ClusterArn": str,
+        "ClusterName": str,
+        "ClusterStatus": ClusterStatusType,
+        "CreationTime": datetime,
+        "FailureMessage": str,
+        "InstanceGroups": List[ClusterInstanceGroupDetailsTypeDef],
+        "VpcConfig": VpcConfigOutputTypeDef,
+        "ResponseMetadata": ResponseMetadataTypeDef,
+    },
+)
+CreateClusterRequestRequestTypeDef = TypedDict(
+    "CreateClusterRequestRequestTypeDef",
+    {
+        "ClusterName": str,
+        "InstanceGroups": Sequence[ClusterInstanceGroupSpecificationTypeDef],
+        "VpcConfig": NotRequired[VpcConfigTypeDef],
+        "Tags": NotRequired[Sequence[TagTypeDef]],
+    },
+)
+UpdateClusterRequestRequestTypeDef = TypedDict(
+    "UpdateClusterRequestRequestTypeDef",
+    {
+        "ClusterName": str,
+        "InstanceGroups": Sequence[ClusterInstanceGroupSpecificationTypeDef],
+    },
+)
+DescribeClusterNodeResponseTypeDef = TypedDict(
+    "DescribeClusterNodeResponseTypeDef",
+    {
+        "NodeDetails": ClusterNodeDetailsTypeDef,
+        "ResponseMetadata": ResponseMetadataTypeDef,
+    },
+)
 CreateFeatureGroupRequestRequestTypeDef = TypedDict(
     "CreateFeatureGroupRequestRequestTypeDef",
     {