mypy-boto3-sagemaker 1.34.124__tar.gz → 1.34.129__tar.gz

{mypy_boto3_sagemaker-1.34.124 → mypy_boto3_sagemaker-1.34.129}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: mypy-boto3-sagemaker
-Version: 1.34.124
-Summary: Type annotations for boto3.SageMaker 1.34.124 service generated with mypy-boto3-builder 7.24.0
+Version: 1.34.129
+Summary: Type annotations for boto3.SageMaker 1.34.129 service generated with mypy-boto3-builder 7.24.0
 Home-page: https://github.com/youtype/mypy_boto3_builder
 Author: Vlad Emelianov
 Author-email: vlad.emelianov.nz@gmail.com
@@ -43,7 +43,7 @@ Requires-Dist: typing-extensions>=4.1.0; python_version < "3.12"
 ![boto3.typed](https://github.com/youtype/mypy_boto3_builder/raw/main/logo.png)
 
 Type annotations for
-[boto3.SageMaker 1.34.124](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker)
+[boto3.SageMaker 1.34.129](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker)
 service compatible with [VSCode](https://code.visualstudio.com/),
 [PyCharm](https://www.jetbrains.com/pycharm/),
 [Emacs](https://www.gnu.org/software/emacs/),
@@ -321,6 +321,7 @@ from mypy_boto3_sagemaker.paginator import (
     ListLabelingJobsPaginator,
     ListLabelingJobsForWorkteamPaginator,
     ListLineageGroupsPaginator,
+    ListMlflowTrackingServersPaginator,
     ListModelBiasJobDefinitionsPaginator,
     ListModelCardExportJobsPaginator,
     ListModelCardVersionsPaginator,
@@ -437,6 +438,9 @@ list_labeling_jobs_for_workteam_paginator: ListLabelingJobsForWorkteamPaginator
 list_lineage_groups_paginator: ListLineageGroupsPaginator = client.get_paginator(
     "list_lineage_groups"
 )
+list_mlflow_tracking_servers_paginator: ListMlflowTrackingServersPaginator = client.get_paginator(
+    "list_mlflow_tracking_servers"
+)
 list_model_bias_job_definitions_paginator: ListModelBiasJobDefinitionsPaginator = (
     client.get_paginator("list_model_bias_job_definitions")
 )

{mypy_boto3_sagemaker-1.34.124 → mypy_boto3_sagemaker-1.34.129}/README.md

@@ -10,7 +10,7 @@
 ![boto3.typed](https://github.com/youtype/mypy_boto3_builder/raw/main/logo.png)
 
 Type annotations for
-[boto3.SageMaker 1.34.124](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker)
+[boto3.SageMaker 1.34.129](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker)
 service compatible with [VSCode](https://code.visualstudio.com/),
 [PyCharm](https://www.jetbrains.com/pycharm/),
 [Emacs](https://www.gnu.org/software/emacs/),
@@ -288,6 +288,7 @@ from mypy_boto3_sagemaker.paginator import (
     ListLabelingJobsPaginator,
     ListLabelingJobsForWorkteamPaginator,
     ListLineageGroupsPaginator,
+    ListMlflowTrackingServersPaginator,
     ListModelBiasJobDefinitionsPaginator,
     ListModelCardExportJobsPaginator,
     ListModelCardVersionsPaginator,
@@ -404,6 +405,9 @@ list_labeling_jobs_for_workteam_paginator: ListLabelingJobsForWorkteamPaginator
 list_lineage_groups_paginator: ListLineageGroupsPaginator = client.get_paginator(
     "list_lineage_groups"
 )
+list_mlflow_tracking_servers_paginator: ListMlflowTrackingServersPaginator = client.get_paginator(
+    "list_mlflow_tracking_servers"
+)
 list_model_bias_job_definitions_paginator: ListModelBiasJobDefinitionsPaginator = (
     client.get_paginator("list_model_bias_job_definitions")
 )

{mypy_boto3_sagemaker-1.34.124 → mypy_boto3_sagemaker-1.34.129}/mypy_boto3_sagemaker/__init__.py

@@ -50,6 +50,7 @@ Usage::
         ListLabelingJobsForWorkteamPaginator,
         ListLabelingJobsPaginator,
         ListLineageGroupsPaginator,
+        ListMlflowTrackingServersPaginator,
         ListModelBiasJobDefinitionsPaginator,
         ListModelCardExportJobsPaginator,
         ListModelCardVersionsPaginator,
@@ -148,6 +149,7 @@ Usage::
     list_labeling_jobs_paginator: ListLabelingJobsPaginator = client.get_paginator("list_labeling_jobs")
     list_labeling_jobs_for_workteam_paginator: ListLabelingJobsForWorkteamPaginator = client.get_paginator("list_labeling_jobs_for_workteam")
     list_lineage_groups_paginator: ListLineageGroupsPaginator = client.get_paginator("list_lineage_groups")
+    list_mlflow_tracking_servers_paginator: ListMlflowTrackingServersPaginator = client.get_paginator("list_mlflow_tracking_servers")
     list_model_bias_job_definitions_paginator: ListModelBiasJobDefinitionsPaginator = client.get_paginator("list_model_bias_job_definitions")
     list_model_card_export_jobs_paginator: ListModelCardExportJobsPaginator = client.get_paginator("list_model_card_export_jobs")
     list_model_card_versions_paginator: ListModelCardVersionsPaginator = client.get_paginator("list_model_card_versions")
@@ -225,6 +227,7 @@ from .paginator import (
     ListLabelingJobsForWorkteamPaginator,
     ListLabelingJobsPaginator,
     ListLineageGroupsPaginator,
+    ListMlflowTrackingServersPaginator,
     ListModelBiasJobDefinitionsPaginator,
     ListModelCardExportJobsPaginator,
     ListModelCardsPaginator,
@@ -325,6 +328,7 @@ __all__ = (
     "ListLabelingJobsForWorkteamPaginator",
     "ListLabelingJobsPaginator",
     "ListLineageGroupsPaginator",
+    "ListMlflowTrackingServersPaginator",
     "ListModelBiasJobDefinitionsPaginator",
     "ListModelCardExportJobsPaginator",
     "ListModelCardVersionsPaginator",

{mypy_boto3_sagemaker-1.34.124 → mypy_boto3_sagemaker-1.34.129}/mypy_boto3_sagemaker/__init__.pyi

@@ -50,6 +50,7 @@ Usage::
         ListLabelingJobsForWorkteamPaginator,
         ListLabelingJobsPaginator,
         ListLineageGroupsPaginator,
+        ListMlflowTrackingServersPaginator,
         ListModelBiasJobDefinitionsPaginator,
         ListModelCardExportJobsPaginator,
         ListModelCardVersionsPaginator,
@@ -148,6 +149,7 @@ Usage::
     list_labeling_jobs_paginator: ListLabelingJobsPaginator = client.get_paginator("list_labeling_jobs")
     list_labeling_jobs_for_workteam_paginator: ListLabelingJobsForWorkteamPaginator = client.get_paginator("list_labeling_jobs_for_workteam")
     list_lineage_groups_paginator: ListLineageGroupsPaginator = client.get_paginator("list_lineage_groups")
+    list_mlflow_tracking_servers_paginator: ListMlflowTrackingServersPaginator = client.get_paginator("list_mlflow_tracking_servers")
     list_model_bias_job_definitions_paginator: ListModelBiasJobDefinitionsPaginator = client.get_paginator("list_model_bias_job_definitions")
     list_model_card_export_jobs_paginator: ListModelCardExportJobsPaginator = client.get_paginator("list_model_card_export_jobs")
     list_model_card_versions_paginator: ListModelCardVersionsPaginator = client.get_paginator("list_model_card_versions")
@@ -225,6 +227,7 @@ from .paginator import (
     ListLabelingJobsForWorkteamPaginator,
     ListLabelingJobsPaginator,
     ListLineageGroupsPaginator,
+    ListMlflowTrackingServersPaginator,
     ListModelBiasJobDefinitionsPaginator,
     ListModelCardExportJobsPaginator,
     ListModelCardsPaginator,
@@ -325,6 +328,7 @@ __all__ = (
     "ListLabelingJobsForWorkteamPaginator",
     "ListLabelingJobsPaginator",
     "ListLineageGroupsPaginator",
+    "ListMlflowTrackingServersPaginator",
     "ListModelBiasJobDefinitionsPaginator",
     "ListModelCardExportJobsPaginator",
     "ListModelCardVersionsPaginator",

{mypy_boto3_sagemaker-1.34.124 → mypy_boto3_sagemaker-1.34.129}/mypy_boto3_sagemaker/__main__.py

@@ -10,8 +10,8 @@ def print_info() -> None:
     Print package info to stdout.
     """
     print(
-        "Type annotations for boto3.SageMaker 1.34.124\n"
-        "Version:         1.34.124\n"
+        "Type annotations for boto3.SageMaker 1.34.129\n"
+        "Version:         1.34.129\n"
         "Builder version: 7.24.0\n"
         "Docs:            https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker//\n"
         "Boto3 docs:      https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker\n"
@@ -24,7 +24,7 @@ def print_version() -> None:
     """
     Print package version to stdout.
     """
-    print("1.34.124")
+    print("1.34.129")
 
 
 def main() -> None:

{mypy_boto3_sagemaker-1.34.124 → mypy_boto3_sagemaker-1.34.129}/mypy_boto3_sagemaker/client.py

@@ -121,11 +121,14 @@ from .literals import (
     SortOrderType,
     SortPipelineExecutionsByType,
     SortPipelinesByType,
+    SortTrackingServerByType,
     SortTrialComponentsByType,
     SortTrialsByType,
     SpaceSortKeyType,
     StudioLifecycleConfigAppTypeType,
     StudioLifecycleConfigSortKeyType,
+    TrackingServerSizeType,
+    TrackingServerStatusType,
     TrainingJobSortByOptionsType,
     TrainingJobStatusType,
     TransformJobStatusType,
@@ -170,6 +173,7 @@ from .paginator import (
     ListLabelingJobsForWorkteamPaginator,
     ListLabelingJobsPaginator,
     ListLineageGroupsPaginator,
+    ListMlflowTrackingServersPaginator,
     ListModelBiasJobDefinitionsPaginator,
     ListModelCardExportJobsPaginator,
     ListModelCardsPaginator,
@@ -264,6 +268,7 @@ from .type_defs import (
     CreateInferenceExperimentResponseTypeDef,
     CreateInferenceRecommendationsJobResponseTypeDef,
     CreateLabelingJobResponseTypeDef,
+    CreateMlflowTrackingServerResponseTypeDef,
     CreateModelBiasJobDefinitionResponseTypeDef,
     CreateModelCardExportJobResponseTypeDef,
     CreateModelCardResponseTypeDef,
@@ -277,6 +282,7 @@ from .type_defs import (
     CreateNotebookInstanceOutputTypeDef,
     CreatePipelineResponseTypeDef,
     CreatePresignedDomainUrlResponseTypeDef,
+    CreatePresignedMlflowTrackingServerUrlResponseTypeDef,
     CreatePresignedNotebookInstanceUrlOutputTypeDef,
     CreateProcessingJobResponseTypeDef,
     CreateProjectOutputTypeDef,
@@ -304,6 +310,7 @@ from .type_defs import (
     DeleteContextResponseTypeDef,
     DeleteExperimentResponseTypeDef,
     DeleteInferenceExperimentResponseTypeDef,
+    DeleteMlflowTrackingServerResponseTypeDef,
     DeletePipelineResponseTypeDef,
     DeleteTrialComponentResponseTypeDef,
     DeleteTrialResponseTypeDef,
@@ -345,6 +352,7 @@ from .type_defs import (
     DescribeInferenceRecommendationsJobResponseTypeDef,
     DescribeLabelingJobResponseTypeDef,
     DescribeLineageGroupResponseTypeDef,
+    DescribeMlflowTrackingServerResponseTypeDef,
     DescribeModelBiasJobDefinitionResponseTypeDef,
     DescribeModelCardExportJobResponseTypeDef,
     DescribeModelCardResponseTypeDef,
@@ -456,6 +464,7 @@ from .type_defs import (
     ListLabelingJobsForWorkteamResponseTypeDef,
     ListLabelingJobsResponseTypeDef,
     ListLineageGroupsResponseTypeDef,
+    ListMlflowTrackingServersResponseTypeDef,
     ListModelBiasJobDefinitionsResponseTypeDef,
     ListModelCardExportJobsResponseTypeDef,
     ListModelCardsResponseTypeDef,
@@ -572,8 +581,10 @@ from .type_defs import (
     SpaceSettingsUnionTypeDef,
     SpaceSharingSettingsTypeDef,
     StartInferenceExperimentResponseTypeDef,
+    StartMlflowTrackingServerResponseTypeDef,
     StartPipelineExecutionResponseTypeDef,
     StopInferenceExperimentResponseTypeDef,
+    StopMlflowTrackingServerResponseTypeDef,
     StoppingConditionTypeDef,
     StopPipelineExecutionResponseTypeDef,
     SuggestionQueryTypeDef,
@@ -608,6 +619,7 @@ from .type_defs import (
     UpdateInferenceComponentOutputTypeDef,
     UpdateInferenceComponentRuntimeConfigOutputTypeDef,
     UpdateInferenceExperimentResponseTypeDef,
+    UpdateMlflowTrackingServerResponseTypeDef,
     UpdateModelCardResponseTypeDef,
     UpdateModelPackageOutputTypeDef,
     UpdateMonitoringAlertResponseTypeDef,
@@ -1308,6 +1320,27 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_labeling_job)
         """
 
+    def create_mlflow_tracking_server(
+        self,
+        *,
+        TrackingServerName: str,
+        ArtifactStoreUri: str,
+        RoleArn: str,
+        TrackingServerSize: TrackingServerSizeType = ...,
+        MlflowVersion: str = ...,
+        AutomaticModelRegistration: bool = ...,
+        WeeklyMaintenanceWindowStart: str = ...,
+        Tags: Sequence[TagTypeDef] = ...,
+    ) -> CreateMlflowTrackingServerResponseTypeDef:
+        """
+        Creates an MLflow Tracking Server using a general purpose Amazon S3 bucket as
+        the artifact
+        store.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.create_mlflow_tracking_server)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_mlflow_tracking_server)
+        """
+
     def create_model(
         self,
         *,
@@ -1567,6 +1600,22 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_presigned_domain_url)
         """
 
+    def create_presigned_mlflow_tracking_server_url(
+        self,
+        *,
+        TrackingServerName: str,
+        ExpiresInSeconds: int = ...,
+        SessionExpirationDurationInSeconds: int = ...,
+    ) -> CreatePresignedMlflowTrackingServerUrlResponseTypeDef:
+        """
+        Returns a presigned URL that you can use to connect to the MLflow UI attached
+        to your tracking
+        server.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.create_presigned_mlflow_tracking_server_url)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_presigned_mlflow_tracking_server_url)
+        """
+
     def create_presigned_notebook_instance_url(
         self, *, NotebookInstanceName: str, SessionExpirationDurationInSeconds: int = ...
     ) -> CreatePresignedNotebookInstanceUrlOutputTypeDef:
@@ -2062,6 +2111,16 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#delete_inference_experiment)
         """
 
+    def delete_mlflow_tracking_server(
+        self, *, TrackingServerName: str
+    ) -> DeleteMlflowTrackingServerResponseTypeDef:
+        """
+        Deletes an MLflow Tracking Server.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.delete_mlflow_tracking_server)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#delete_mlflow_tracking_server)
+        """
+
     def delete_model(self, *, ModelName: str) -> EmptyResponseMetadataTypeDef:
         """
         Deletes a model.
@@ -2605,6 +2664,16 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#describe_lineage_group)
         """
 
+    def describe_mlflow_tracking_server(
+        self, *, TrackingServerName: str
+    ) -> DescribeMlflowTrackingServerResponseTypeDef:
+        """
+        Returns information about an MLflow Tracking Server.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.describe_mlflow_tracking_server)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#describe_mlflow_tracking_server)
+        """
+
     def describe_model(self, *, ModelName: str) -> DescribeModelOutputTypeDef:
         """
         Describes a model that you created using the `CreateModel` API.
@@ -3735,6 +3804,25 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#list_lineage_groups)
         """
 
+    def list_mlflow_tracking_servers(
+        self,
+        *,
+        CreatedAfter: TimestampTypeDef = ...,
+        CreatedBefore: TimestampTypeDef = ...,
+        TrackingServerStatus: TrackingServerStatusType = ...,
+        MlflowVersion: str = ...,
+        SortBy: SortTrackingServerByType = ...,
+        SortOrder: SortOrderType = ...,
+        NextToken: str = ...,
+        MaxResults: int = ...,
+    ) -> ListMlflowTrackingServersResponseTypeDef:
+        """
+        Lists all MLflow Tracking Servers.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.list_mlflow_tracking_servers)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#list_mlflow_tracking_servers)
+        """
+
     def list_model_bias_job_definitions(
         self,
         *,
@@ -4541,6 +4629,16 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#start_inference_experiment)
         """
 
+    def start_mlflow_tracking_server(
+        self, *, TrackingServerName: str
+    ) -> StartMlflowTrackingServerResponseTypeDef:
+        """
+        Programmatically start an MLflow Tracking Server.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.start_mlflow_tracking_server)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#start_mlflow_tracking_server)
+        """
+
     def start_monitoring_schedule(
         self, *, MonitoringScheduleName: str
     ) -> EmptyResponseMetadataTypeDef:
@@ -4657,6 +4755,16 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#stop_labeling_job)
         """
 
+    def stop_mlflow_tracking_server(
+        self, *, TrackingServerName: str
+    ) -> StopMlflowTrackingServerResponseTypeDef:
+        """
+        Programmatically stop an MLflow Tracking Server.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.stop_mlflow_tracking_server)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#stop_mlflow_tracking_server)
+        """
+
     def stop_monitoring_schedule(
         self, *, MonitoringScheduleName: str
     ) -> EmptyResponseMetadataTypeDef:
@@ -5020,6 +5128,22 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#update_inference_experiment)
         """
 
+    def update_mlflow_tracking_server(
+        self,
+        *,
+        TrackingServerName: str,
+        ArtifactStoreUri: str = ...,
+        TrackingServerSize: TrackingServerSizeType = ...,
+        AutomaticModelRegistration: bool = ...,
+        WeeklyMaintenanceWindowStart: str = ...,
+    ) -> UpdateMlflowTrackingServerResponseTypeDef:
+        """
+        Updates properties of an existing MLflow Tracking Server.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.update_mlflow_tracking_server)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#update_mlflow_tracking_server)
+        """
+
     def update_model_card(
         self, *, ModelCardName: str, Content: str = ..., ModelCardStatus: ModelCardStatusType = ...
     ) -> UpdateModelCardResponseTypeDef:
@@ -5577,6 +5701,15 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#get_paginator)
         """
 
+    @overload
+    def get_paginator(
+        self, operation_name: Literal["list_mlflow_tracking_servers"]
+    ) -> ListMlflowTrackingServersPaginator:
+        """
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.get_paginator)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#get_paginator)
+        """
+
     @overload
     def get_paginator(
         self, operation_name: Literal["list_model_bias_job_definitions"]

{mypy_boto3_sagemaker-1.34.124 → mypy_boto3_sagemaker-1.34.129}/mypy_boto3_sagemaker/client.pyi

@@ -121,11 +121,14 @@ from .literals import (
     SortOrderType,
     SortPipelineExecutionsByType,
     SortPipelinesByType,
+    SortTrackingServerByType,
     SortTrialComponentsByType,
     SortTrialsByType,
     SpaceSortKeyType,
     StudioLifecycleConfigAppTypeType,
     StudioLifecycleConfigSortKeyType,
+    TrackingServerSizeType,
+    TrackingServerStatusType,
     TrainingJobSortByOptionsType,
     TrainingJobStatusType,
     TransformJobStatusType,
@@ -170,6 +173,7 @@ from .paginator import (
     ListLabelingJobsForWorkteamPaginator,
     ListLabelingJobsPaginator,
     ListLineageGroupsPaginator,
+    ListMlflowTrackingServersPaginator,
     ListModelBiasJobDefinitionsPaginator,
     ListModelCardExportJobsPaginator,
     ListModelCardsPaginator,
@@ -264,6 +268,7 @@ from .type_defs import (
     CreateInferenceExperimentResponseTypeDef,
     CreateInferenceRecommendationsJobResponseTypeDef,
     CreateLabelingJobResponseTypeDef,
+    CreateMlflowTrackingServerResponseTypeDef,
     CreateModelBiasJobDefinitionResponseTypeDef,
     CreateModelCardExportJobResponseTypeDef,
     CreateModelCardResponseTypeDef,
@@ -277,6 +282,7 @@ from .type_defs import (
     CreateNotebookInstanceOutputTypeDef,
     CreatePipelineResponseTypeDef,
     CreatePresignedDomainUrlResponseTypeDef,
+    CreatePresignedMlflowTrackingServerUrlResponseTypeDef,
     CreatePresignedNotebookInstanceUrlOutputTypeDef,
     CreateProcessingJobResponseTypeDef,
     CreateProjectOutputTypeDef,
@@ -304,6 +310,7 @@ from .type_defs import (
     DeleteContextResponseTypeDef,
     DeleteExperimentResponseTypeDef,
     DeleteInferenceExperimentResponseTypeDef,
+    DeleteMlflowTrackingServerResponseTypeDef,
     DeletePipelineResponseTypeDef,
     DeleteTrialComponentResponseTypeDef,
     DeleteTrialResponseTypeDef,
@@ -345,6 +352,7 @@ from .type_defs import (
     DescribeInferenceRecommendationsJobResponseTypeDef,
     DescribeLabelingJobResponseTypeDef,
     DescribeLineageGroupResponseTypeDef,
+    DescribeMlflowTrackingServerResponseTypeDef,
     DescribeModelBiasJobDefinitionResponseTypeDef,
     DescribeModelCardExportJobResponseTypeDef,
     DescribeModelCardResponseTypeDef,
@@ -456,6 +464,7 @@ from .type_defs import (
     ListLabelingJobsForWorkteamResponseTypeDef,
     ListLabelingJobsResponseTypeDef,
     ListLineageGroupsResponseTypeDef,
+    ListMlflowTrackingServersResponseTypeDef,
     ListModelBiasJobDefinitionsResponseTypeDef,
     ListModelCardExportJobsResponseTypeDef,
     ListModelCardsResponseTypeDef,
@@ -572,8 +581,10 @@ from .type_defs import (
     SpaceSettingsUnionTypeDef,
     SpaceSharingSettingsTypeDef,
     StartInferenceExperimentResponseTypeDef,
+    StartMlflowTrackingServerResponseTypeDef,
     StartPipelineExecutionResponseTypeDef,
     StopInferenceExperimentResponseTypeDef,
+    StopMlflowTrackingServerResponseTypeDef,
     StoppingConditionTypeDef,
     StopPipelineExecutionResponseTypeDef,
     SuggestionQueryTypeDef,
@@ -608,6 +619,7 @@ from .type_defs import (
     UpdateInferenceComponentOutputTypeDef,
     UpdateInferenceComponentRuntimeConfigOutputTypeDef,
     UpdateInferenceExperimentResponseTypeDef,
+    UpdateMlflowTrackingServerResponseTypeDef,
     UpdateModelCardResponseTypeDef,
     UpdateModelPackageOutputTypeDef,
     UpdateMonitoringAlertResponseTypeDef,
@@ -1305,6 +1317,27 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_labeling_job)
         """
 
+    def create_mlflow_tracking_server(
+        self,
+        *,
+        TrackingServerName: str,
+        ArtifactStoreUri: str,
+        RoleArn: str,
+        TrackingServerSize: TrackingServerSizeType = ...,
+        MlflowVersion: str = ...,
+        AutomaticModelRegistration: bool = ...,
+        WeeklyMaintenanceWindowStart: str = ...,
+        Tags: Sequence[TagTypeDef] = ...,
+    ) -> CreateMlflowTrackingServerResponseTypeDef:
+        """
+        Creates an MLflow Tracking Server using a general purpose Amazon S3 bucket as
+        the artifact
+        store.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.create_mlflow_tracking_server)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_mlflow_tracking_server)
+        """
+
     def create_model(
         self,
         *,
@@ -1564,6 +1597,22 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_presigned_domain_url)
         """
 
+    def create_presigned_mlflow_tracking_server_url(
+        self,
+        *,
+        TrackingServerName: str,
+        ExpiresInSeconds: int = ...,
+        SessionExpirationDurationInSeconds: int = ...,
+    ) -> CreatePresignedMlflowTrackingServerUrlResponseTypeDef:
+        """
+        Returns a presigned URL that you can use to connect to the MLflow UI attached
+        to your tracking
+        server.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.create_presigned_mlflow_tracking_server_url)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_presigned_mlflow_tracking_server_url)
+        """
+
     def create_presigned_notebook_instance_url(
         self, *, NotebookInstanceName: str, SessionExpirationDurationInSeconds: int = ...
     ) -> CreatePresignedNotebookInstanceUrlOutputTypeDef:
@@ -2059,6 +2108,16 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#delete_inference_experiment)
         """
 
+    def delete_mlflow_tracking_server(
+        self, *, TrackingServerName: str
+    ) -> DeleteMlflowTrackingServerResponseTypeDef:
+        """
+        Deletes an MLflow Tracking Server.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.delete_mlflow_tracking_server)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#delete_mlflow_tracking_server)
+        """
+
     def delete_model(self, *, ModelName: str) -> EmptyResponseMetadataTypeDef:
         """
         Deletes a model.
@@ -2602,6 +2661,16 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#describe_lineage_group)
         """
 
+    def describe_mlflow_tracking_server(
+        self, *, TrackingServerName: str
+    ) -> DescribeMlflowTrackingServerResponseTypeDef:
+        """
+        Returns information about an MLflow Tracking Server.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.describe_mlflow_tracking_server)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#describe_mlflow_tracking_server)
+        """
+
     def describe_model(self, *, ModelName: str) -> DescribeModelOutputTypeDef:
         """
         Describes a model that you created using the `CreateModel` API.
@@ -3732,6 +3801,25 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#list_lineage_groups)
         """
 
+    def list_mlflow_tracking_servers(
+        self,
+        *,
+        CreatedAfter: TimestampTypeDef = ...,
+        CreatedBefore: TimestampTypeDef = ...,
+        TrackingServerStatus: TrackingServerStatusType = ...,
+        MlflowVersion: str = ...,
+        SortBy: SortTrackingServerByType = ...,
+        SortOrder: SortOrderType = ...,
+        NextToken: str = ...,
+        MaxResults: int = ...,
+    ) -> ListMlflowTrackingServersResponseTypeDef:
+        """
+        Lists all MLflow Tracking Servers.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.list_mlflow_tracking_servers)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#list_mlflow_tracking_servers)
+        """
+
     def list_model_bias_job_definitions(
         self,
         *,
@@ -4538,6 +4626,16 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#start_inference_experiment)
         """
 
+    def start_mlflow_tracking_server(
+        self, *, TrackingServerName: str
+    ) -> StartMlflowTrackingServerResponseTypeDef:
+        """
+        Programmatically start an MLflow Tracking Server.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.start_mlflow_tracking_server)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#start_mlflow_tracking_server)
+        """
+
     def start_monitoring_schedule(
         self, *, MonitoringScheduleName: str
     ) -> EmptyResponseMetadataTypeDef:
@@ -4654,6 +4752,16 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#stop_labeling_job)
         """
 
+    def stop_mlflow_tracking_server(
+        self, *, TrackingServerName: str
+    ) -> StopMlflowTrackingServerResponseTypeDef:
+        """
+        Programmatically stop an MLflow Tracking Server.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.stop_mlflow_tracking_server)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#stop_mlflow_tracking_server)
+        """
+
     def stop_monitoring_schedule(
         self, *, MonitoringScheduleName: str
     ) -> EmptyResponseMetadataTypeDef:
@@ -5017,6 +5125,22 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#update_inference_experiment)
         """
 
+    def update_mlflow_tracking_server(
+        self,
+        *,
+        TrackingServerName: str,
+        ArtifactStoreUri: str = ...,
+        TrackingServerSize: TrackingServerSizeType = ...,
+        AutomaticModelRegistration: bool = ...,
+        WeeklyMaintenanceWindowStart: str = ...,
+    ) -> UpdateMlflowTrackingServerResponseTypeDef:
+        """
+        Updates properties of an existing MLflow Tracking Server.
+
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.update_mlflow_tracking_server)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#update_mlflow_tracking_server)
+        """
+
     def update_model_card(
         self, *, ModelCardName: str, Content: str = ..., ModelCardStatus: ModelCardStatusType = ...
     ) -> UpdateModelCardResponseTypeDef:
@@ -5574,6 +5698,15 @@ class SageMakerClient(BaseClient):
         [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#get_paginator)
         """
 
+    @overload
+    def get_paginator(
+        self, operation_name: Literal["list_mlflow_tracking_servers"]
+    ) -> ListMlflowTrackingServersPaginator:
+        """
+        [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.get_paginator)
+        [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#get_paginator)
+        """
+
     @overload
     def get_paginator(
         self, operation_name: Literal["list_model_bias_job_definitions"]