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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of mypy-boto3-sagemaker might be problematic. Click here for more details.

@@ -50,6 +50,7 @@ Usage::
50
50
  ListLabelingJobsForWorkteamPaginator,
51
51
  ListLabelingJobsPaginator,
52
52
  ListLineageGroupsPaginator,
53
+ ListMlflowTrackingServersPaginator,
53
54
  ListModelBiasJobDefinitionsPaginator,
54
55
  ListModelCardExportJobsPaginator,
55
56
  ListModelCardVersionsPaginator,
@@ -148,6 +149,7 @@ Usage::
148
149
  list_labeling_jobs_paginator: ListLabelingJobsPaginator = client.get_paginator("list_labeling_jobs")
149
150
  list_labeling_jobs_for_workteam_paginator: ListLabelingJobsForWorkteamPaginator = client.get_paginator("list_labeling_jobs_for_workteam")
150
151
  list_lineage_groups_paginator: ListLineageGroupsPaginator = client.get_paginator("list_lineage_groups")
152
+ list_mlflow_tracking_servers_paginator: ListMlflowTrackingServersPaginator = client.get_paginator("list_mlflow_tracking_servers")
151
153
  list_model_bias_job_definitions_paginator: ListModelBiasJobDefinitionsPaginator = client.get_paginator("list_model_bias_job_definitions")
152
154
  list_model_card_export_jobs_paginator: ListModelCardExportJobsPaginator = client.get_paginator("list_model_card_export_jobs")
153
155
  list_model_card_versions_paginator: ListModelCardVersionsPaginator = client.get_paginator("list_model_card_versions")
@@ -225,6 +227,7 @@ from .paginator import (
225
227
  ListLabelingJobsForWorkteamPaginator,
226
228
  ListLabelingJobsPaginator,
227
229
  ListLineageGroupsPaginator,
230
+ ListMlflowTrackingServersPaginator,
228
231
  ListModelBiasJobDefinitionsPaginator,
229
232
  ListModelCardExportJobsPaginator,
230
233
  ListModelCardsPaginator,
@@ -325,6 +328,7 @@ __all__ = (
325
328
  "ListLabelingJobsForWorkteamPaginator",
326
329
  "ListLabelingJobsPaginator",
327
330
  "ListLineageGroupsPaginator",
331
+ "ListMlflowTrackingServersPaginator",
328
332
  "ListModelBiasJobDefinitionsPaginator",
329
333
  "ListModelCardExportJobsPaginator",
330
334
  "ListModelCardVersionsPaginator",
@@ -50,6 +50,7 @@ Usage::
50
50
  ListLabelingJobsForWorkteamPaginator,
51
51
  ListLabelingJobsPaginator,
52
52
  ListLineageGroupsPaginator,
53
+ ListMlflowTrackingServersPaginator,
53
54
  ListModelBiasJobDefinitionsPaginator,
54
55
  ListModelCardExportJobsPaginator,
55
56
  ListModelCardVersionsPaginator,
@@ -148,6 +149,7 @@ Usage::
148
149
  list_labeling_jobs_paginator: ListLabelingJobsPaginator = client.get_paginator("list_labeling_jobs")
149
150
  list_labeling_jobs_for_workteam_paginator: ListLabelingJobsForWorkteamPaginator = client.get_paginator("list_labeling_jobs_for_workteam")
150
151
  list_lineage_groups_paginator: ListLineageGroupsPaginator = client.get_paginator("list_lineage_groups")
152
+ list_mlflow_tracking_servers_paginator: ListMlflowTrackingServersPaginator = client.get_paginator("list_mlflow_tracking_servers")
151
153
  list_model_bias_job_definitions_paginator: ListModelBiasJobDefinitionsPaginator = client.get_paginator("list_model_bias_job_definitions")
152
154
  list_model_card_export_jobs_paginator: ListModelCardExportJobsPaginator = client.get_paginator("list_model_card_export_jobs")
153
155
  list_model_card_versions_paginator: ListModelCardVersionsPaginator = client.get_paginator("list_model_card_versions")
@@ -225,6 +227,7 @@ from .paginator import (
225
227
  ListLabelingJobsForWorkteamPaginator,
226
228
  ListLabelingJobsPaginator,
227
229
  ListLineageGroupsPaginator,
230
+ ListMlflowTrackingServersPaginator,
228
231
  ListModelBiasJobDefinitionsPaginator,
229
232
  ListModelCardExportJobsPaginator,
230
233
  ListModelCardsPaginator,
@@ -325,6 +328,7 @@ __all__ = (
325
328
  "ListLabelingJobsForWorkteamPaginator",
326
329
  "ListLabelingJobsPaginator",
327
330
  "ListLineageGroupsPaginator",
331
+ "ListMlflowTrackingServersPaginator",
328
332
  "ListModelBiasJobDefinitionsPaginator",
329
333
  "ListModelCardExportJobsPaginator",
330
334
  "ListModelCardVersionsPaginator",
@@ -10,8 +10,8 @@ def print_info() -> None:
10
10
  Print package info to stdout.
11
11
  """
12
12
  print(
13
- "Type annotations for boto3.SageMaker 1.34.124\n"
14
- "Version: 1.34.124\n"
13
+ "Type annotations for boto3.SageMaker 1.34.129\n"
14
+ "Version: 1.34.129\n"
15
15
  "Builder version: 7.24.0\n"
16
16
  "Docs: https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker//\n"
17
17
  "Boto3 docs: https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker\n"
@@ -24,7 +24,7 @@ def print_version() -> None:
24
24
  """
25
25
  Print package version to stdout.
26
26
  """
27
- print("1.34.124")
27
+ print("1.34.129")
28
28
 
29
29
 
30
30
  def main() -> None:
@@ -121,11 +121,14 @@ from .literals import (
121
121
  SortOrderType,
122
122
  SortPipelineExecutionsByType,
123
123
  SortPipelinesByType,
124
+ SortTrackingServerByType,
124
125
  SortTrialComponentsByType,
125
126
  SortTrialsByType,
126
127
  SpaceSortKeyType,
127
128
  StudioLifecycleConfigAppTypeType,
128
129
  StudioLifecycleConfigSortKeyType,
130
+ TrackingServerSizeType,
131
+ TrackingServerStatusType,
129
132
  TrainingJobSortByOptionsType,
130
133
  TrainingJobStatusType,
131
134
  TransformJobStatusType,
@@ -170,6 +173,7 @@ from .paginator import (
170
173
  ListLabelingJobsForWorkteamPaginator,
171
174
  ListLabelingJobsPaginator,
172
175
  ListLineageGroupsPaginator,
176
+ ListMlflowTrackingServersPaginator,
173
177
  ListModelBiasJobDefinitionsPaginator,
174
178
  ListModelCardExportJobsPaginator,
175
179
  ListModelCardsPaginator,
@@ -264,6 +268,7 @@ from .type_defs import (
264
268
  CreateInferenceExperimentResponseTypeDef,
265
269
  CreateInferenceRecommendationsJobResponseTypeDef,
266
270
  CreateLabelingJobResponseTypeDef,
271
+ CreateMlflowTrackingServerResponseTypeDef,
267
272
  CreateModelBiasJobDefinitionResponseTypeDef,
268
273
  CreateModelCardExportJobResponseTypeDef,
269
274
  CreateModelCardResponseTypeDef,
@@ -277,6 +282,7 @@ from .type_defs import (
277
282
  CreateNotebookInstanceOutputTypeDef,
278
283
  CreatePipelineResponseTypeDef,
279
284
  CreatePresignedDomainUrlResponseTypeDef,
285
+ CreatePresignedMlflowTrackingServerUrlResponseTypeDef,
280
286
  CreatePresignedNotebookInstanceUrlOutputTypeDef,
281
287
  CreateProcessingJobResponseTypeDef,
282
288
  CreateProjectOutputTypeDef,
@@ -304,6 +310,7 @@ from .type_defs import (
304
310
  DeleteContextResponseTypeDef,
305
311
  DeleteExperimentResponseTypeDef,
306
312
  DeleteInferenceExperimentResponseTypeDef,
313
+ DeleteMlflowTrackingServerResponseTypeDef,
307
314
  DeletePipelineResponseTypeDef,
308
315
  DeleteTrialComponentResponseTypeDef,
309
316
  DeleteTrialResponseTypeDef,
@@ -345,6 +352,7 @@ from .type_defs import (
345
352
  DescribeInferenceRecommendationsJobResponseTypeDef,
346
353
  DescribeLabelingJobResponseTypeDef,
347
354
  DescribeLineageGroupResponseTypeDef,
355
+ DescribeMlflowTrackingServerResponseTypeDef,
348
356
  DescribeModelBiasJobDefinitionResponseTypeDef,
349
357
  DescribeModelCardExportJobResponseTypeDef,
350
358
  DescribeModelCardResponseTypeDef,
@@ -456,6 +464,7 @@ from .type_defs import (
456
464
  ListLabelingJobsForWorkteamResponseTypeDef,
457
465
  ListLabelingJobsResponseTypeDef,
458
466
  ListLineageGroupsResponseTypeDef,
467
+ ListMlflowTrackingServersResponseTypeDef,
459
468
  ListModelBiasJobDefinitionsResponseTypeDef,
460
469
  ListModelCardExportJobsResponseTypeDef,
461
470
  ListModelCardsResponseTypeDef,
@@ -572,8 +581,10 @@ from .type_defs import (
572
581
  SpaceSettingsUnionTypeDef,
573
582
  SpaceSharingSettingsTypeDef,
574
583
  StartInferenceExperimentResponseTypeDef,
584
+ StartMlflowTrackingServerResponseTypeDef,
575
585
  StartPipelineExecutionResponseTypeDef,
576
586
  StopInferenceExperimentResponseTypeDef,
587
+ StopMlflowTrackingServerResponseTypeDef,
577
588
  StoppingConditionTypeDef,
578
589
  StopPipelineExecutionResponseTypeDef,
579
590
  SuggestionQueryTypeDef,
@@ -608,6 +619,7 @@ from .type_defs import (
608
619
  UpdateInferenceComponentOutputTypeDef,
609
620
  UpdateInferenceComponentRuntimeConfigOutputTypeDef,
610
621
  UpdateInferenceExperimentResponseTypeDef,
622
+ UpdateMlflowTrackingServerResponseTypeDef,
611
623
  UpdateModelCardResponseTypeDef,
612
624
  UpdateModelPackageOutputTypeDef,
613
625
  UpdateMonitoringAlertResponseTypeDef,
@@ -1308,6 +1320,27 @@ class SageMakerClient(BaseClient):
1308
1320
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_labeling_job)
1309
1321
  """
1310
1322
 
1323
+ def create_mlflow_tracking_server(
1324
+ self,
1325
+ *,
1326
+ TrackingServerName: str,
1327
+ ArtifactStoreUri: str,
1328
+ RoleArn: str,
1329
+ TrackingServerSize: TrackingServerSizeType = ...,
1330
+ MlflowVersion: str = ...,
1331
+ AutomaticModelRegistration: bool = ...,
1332
+ WeeklyMaintenanceWindowStart: str = ...,
1333
+ Tags: Sequence[TagTypeDef] = ...,
1334
+ ) -> CreateMlflowTrackingServerResponseTypeDef:
1335
+ """
1336
+ Creates an MLflow Tracking Server using a general purpose Amazon S3 bucket as
1337
+ the artifact
1338
+ store.
1339
+
1340
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.create_mlflow_tracking_server)
1341
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_mlflow_tracking_server)
1342
+ """
1343
+
1311
1344
  def create_model(
1312
1345
  self,
1313
1346
  *,
@@ -1567,6 +1600,22 @@ class SageMakerClient(BaseClient):
1567
1600
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_presigned_domain_url)
1568
1601
  """
1569
1602
 
1603
+ def create_presigned_mlflow_tracking_server_url(
1604
+ self,
1605
+ *,
1606
+ TrackingServerName: str,
1607
+ ExpiresInSeconds: int = ...,
1608
+ SessionExpirationDurationInSeconds: int = ...,
1609
+ ) -> CreatePresignedMlflowTrackingServerUrlResponseTypeDef:
1610
+ """
1611
+ Returns a presigned URL that you can use to connect to the MLflow UI attached
1612
+ to your tracking
1613
+ server.
1614
+
1615
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.create_presigned_mlflow_tracking_server_url)
1616
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_presigned_mlflow_tracking_server_url)
1617
+ """
1618
+
1570
1619
  def create_presigned_notebook_instance_url(
1571
1620
  self, *, NotebookInstanceName: str, SessionExpirationDurationInSeconds: int = ...
1572
1621
  ) -> CreatePresignedNotebookInstanceUrlOutputTypeDef:
@@ -2062,6 +2111,16 @@ class SageMakerClient(BaseClient):
2062
2111
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#delete_inference_experiment)
2063
2112
  """
2064
2113
 
2114
+ def delete_mlflow_tracking_server(
2115
+ self, *, TrackingServerName: str
2116
+ ) -> DeleteMlflowTrackingServerResponseTypeDef:
2117
+ """
2118
+ Deletes an MLflow Tracking Server.
2119
+
2120
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.delete_mlflow_tracking_server)
2121
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#delete_mlflow_tracking_server)
2122
+ """
2123
+
2065
2124
  def delete_model(self, *, ModelName: str) -> EmptyResponseMetadataTypeDef:
2066
2125
  """
2067
2126
  Deletes a model.
@@ -2605,6 +2664,16 @@ class SageMakerClient(BaseClient):
2605
2664
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#describe_lineage_group)
2606
2665
  """
2607
2666
 
2667
+ def describe_mlflow_tracking_server(
2668
+ self, *, TrackingServerName: str
2669
+ ) -> DescribeMlflowTrackingServerResponseTypeDef:
2670
+ """
2671
+ Returns information about an MLflow Tracking Server.
2672
+
2673
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.describe_mlflow_tracking_server)
2674
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#describe_mlflow_tracking_server)
2675
+ """
2676
+
2608
2677
  def describe_model(self, *, ModelName: str) -> DescribeModelOutputTypeDef:
2609
2678
  """
2610
2679
  Describes a model that you created using the `CreateModel` API.
@@ -3735,6 +3804,25 @@ class SageMakerClient(BaseClient):
3735
3804
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#list_lineage_groups)
3736
3805
  """
3737
3806
 
3807
+ def list_mlflow_tracking_servers(
3808
+ self,
3809
+ *,
3810
+ CreatedAfter: TimestampTypeDef = ...,
3811
+ CreatedBefore: TimestampTypeDef = ...,
3812
+ TrackingServerStatus: TrackingServerStatusType = ...,
3813
+ MlflowVersion: str = ...,
3814
+ SortBy: SortTrackingServerByType = ...,
3815
+ SortOrder: SortOrderType = ...,
3816
+ NextToken: str = ...,
3817
+ MaxResults: int = ...,
3818
+ ) -> ListMlflowTrackingServersResponseTypeDef:
3819
+ """
3820
+ Lists all MLflow Tracking Servers.
3821
+
3822
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.list_mlflow_tracking_servers)
3823
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#list_mlflow_tracking_servers)
3824
+ """
3825
+
3738
3826
  def list_model_bias_job_definitions(
3739
3827
  self,
3740
3828
  *,
@@ -4541,6 +4629,16 @@ class SageMakerClient(BaseClient):
4541
4629
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#start_inference_experiment)
4542
4630
  """
4543
4631
 
4632
+ def start_mlflow_tracking_server(
4633
+ self, *, TrackingServerName: str
4634
+ ) -> StartMlflowTrackingServerResponseTypeDef:
4635
+ """
4636
+ Programmatically start an MLflow Tracking Server.
4637
+
4638
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.start_mlflow_tracking_server)
4639
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#start_mlflow_tracking_server)
4640
+ """
4641
+
4544
4642
  def start_monitoring_schedule(
4545
4643
  self, *, MonitoringScheduleName: str
4546
4644
  ) -> EmptyResponseMetadataTypeDef:
@@ -4657,6 +4755,16 @@ class SageMakerClient(BaseClient):
4657
4755
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#stop_labeling_job)
4658
4756
  """
4659
4757
 
4758
+ def stop_mlflow_tracking_server(
4759
+ self, *, TrackingServerName: str
4760
+ ) -> StopMlflowTrackingServerResponseTypeDef:
4761
+ """
4762
+ Programmatically stop an MLflow Tracking Server.
4763
+
4764
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.stop_mlflow_tracking_server)
4765
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#stop_mlflow_tracking_server)
4766
+ """
4767
+
4660
4768
  def stop_monitoring_schedule(
4661
4769
  self, *, MonitoringScheduleName: str
4662
4770
  ) -> EmptyResponseMetadataTypeDef:
@@ -5020,6 +5128,22 @@ class SageMakerClient(BaseClient):
5020
5128
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#update_inference_experiment)
5021
5129
  """
5022
5130
 
5131
+ def update_mlflow_tracking_server(
5132
+ self,
5133
+ *,
5134
+ TrackingServerName: str,
5135
+ ArtifactStoreUri: str = ...,
5136
+ TrackingServerSize: TrackingServerSizeType = ...,
5137
+ AutomaticModelRegistration: bool = ...,
5138
+ WeeklyMaintenanceWindowStart: str = ...,
5139
+ ) -> UpdateMlflowTrackingServerResponseTypeDef:
5140
+ """
5141
+ Updates properties of an existing MLflow Tracking Server.
5142
+
5143
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.update_mlflow_tracking_server)
5144
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#update_mlflow_tracking_server)
5145
+ """
5146
+
5023
5147
  def update_model_card(
5024
5148
  self, *, ModelCardName: str, Content: str = ..., ModelCardStatus: ModelCardStatusType = ...
5025
5149
  ) -> UpdateModelCardResponseTypeDef:
@@ -5577,6 +5701,15 @@ class SageMakerClient(BaseClient):
5577
5701
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#get_paginator)
5578
5702
  """
5579
5703
 
5704
+ @overload
5705
+ def get_paginator(
5706
+ self, operation_name: Literal["list_mlflow_tracking_servers"]
5707
+ ) -> ListMlflowTrackingServersPaginator:
5708
+ """
5709
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.get_paginator)
5710
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#get_paginator)
5711
+ """
5712
+
5580
5713
  @overload
5581
5714
  def get_paginator(
5582
5715
  self, operation_name: Literal["list_model_bias_job_definitions"]
@@ -121,11 +121,14 @@ from .literals import (
121
121
  SortOrderType,
122
122
  SortPipelineExecutionsByType,
123
123
  SortPipelinesByType,
124
+ SortTrackingServerByType,
124
125
  SortTrialComponentsByType,
125
126
  SortTrialsByType,
126
127
  SpaceSortKeyType,
127
128
  StudioLifecycleConfigAppTypeType,
128
129
  StudioLifecycleConfigSortKeyType,
130
+ TrackingServerSizeType,
131
+ TrackingServerStatusType,
129
132
  TrainingJobSortByOptionsType,
130
133
  TrainingJobStatusType,
131
134
  TransformJobStatusType,
@@ -170,6 +173,7 @@ from .paginator import (
170
173
  ListLabelingJobsForWorkteamPaginator,
171
174
  ListLabelingJobsPaginator,
172
175
  ListLineageGroupsPaginator,
176
+ ListMlflowTrackingServersPaginator,
173
177
  ListModelBiasJobDefinitionsPaginator,
174
178
  ListModelCardExportJobsPaginator,
175
179
  ListModelCardsPaginator,
@@ -264,6 +268,7 @@ from .type_defs import (
264
268
  CreateInferenceExperimentResponseTypeDef,
265
269
  CreateInferenceRecommendationsJobResponseTypeDef,
266
270
  CreateLabelingJobResponseTypeDef,
271
+ CreateMlflowTrackingServerResponseTypeDef,
267
272
  CreateModelBiasJobDefinitionResponseTypeDef,
268
273
  CreateModelCardExportJobResponseTypeDef,
269
274
  CreateModelCardResponseTypeDef,
@@ -277,6 +282,7 @@ from .type_defs import (
277
282
  CreateNotebookInstanceOutputTypeDef,
278
283
  CreatePipelineResponseTypeDef,
279
284
  CreatePresignedDomainUrlResponseTypeDef,
285
+ CreatePresignedMlflowTrackingServerUrlResponseTypeDef,
280
286
  CreatePresignedNotebookInstanceUrlOutputTypeDef,
281
287
  CreateProcessingJobResponseTypeDef,
282
288
  CreateProjectOutputTypeDef,
@@ -304,6 +310,7 @@ from .type_defs import (
304
310
  DeleteContextResponseTypeDef,
305
311
  DeleteExperimentResponseTypeDef,
306
312
  DeleteInferenceExperimentResponseTypeDef,
313
+ DeleteMlflowTrackingServerResponseTypeDef,
307
314
  DeletePipelineResponseTypeDef,
308
315
  DeleteTrialComponentResponseTypeDef,
309
316
  DeleteTrialResponseTypeDef,
@@ -345,6 +352,7 @@ from .type_defs import (
345
352
  DescribeInferenceRecommendationsJobResponseTypeDef,
346
353
  DescribeLabelingJobResponseTypeDef,
347
354
  DescribeLineageGroupResponseTypeDef,
355
+ DescribeMlflowTrackingServerResponseTypeDef,
348
356
  DescribeModelBiasJobDefinitionResponseTypeDef,
349
357
  DescribeModelCardExportJobResponseTypeDef,
350
358
  DescribeModelCardResponseTypeDef,
@@ -456,6 +464,7 @@ from .type_defs import (
456
464
  ListLabelingJobsForWorkteamResponseTypeDef,
457
465
  ListLabelingJobsResponseTypeDef,
458
466
  ListLineageGroupsResponseTypeDef,
467
+ ListMlflowTrackingServersResponseTypeDef,
459
468
  ListModelBiasJobDefinitionsResponseTypeDef,
460
469
  ListModelCardExportJobsResponseTypeDef,
461
470
  ListModelCardsResponseTypeDef,
@@ -572,8 +581,10 @@ from .type_defs import (
572
581
  SpaceSettingsUnionTypeDef,
573
582
  SpaceSharingSettingsTypeDef,
574
583
  StartInferenceExperimentResponseTypeDef,
584
+ StartMlflowTrackingServerResponseTypeDef,
575
585
  StartPipelineExecutionResponseTypeDef,
576
586
  StopInferenceExperimentResponseTypeDef,
587
+ StopMlflowTrackingServerResponseTypeDef,
577
588
  StoppingConditionTypeDef,
578
589
  StopPipelineExecutionResponseTypeDef,
579
590
  SuggestionQueryTypeDef,
@@ -608,6 +619,7 @@ from .type_defs import (
608
619
  UpdateInferenceComponentOutputTypeDef,
609
620
  UpdateInferenceComponentRuntimeConfigOutputTypeDef,
610
621
  UpdateInferenceExperimentResponseTypeDef,
622
+ UpdateMlflowTrackingServerResponseTypeDef,
611
623
  UpdateModelCardResponseTypeDef,
612
624
  UpdateModelPackageOutputTypeDef,
613
625
  UpdateMonitoringAlertResponseTypeDef,
@@ -1305,6 +1317,27 @@ class SageMakerClient(BaseClient):
1305
1317
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_labeling_job)
1306
1318
  """
1307
1319
 
1320
+ def create_mlflow_tracking_server(
1321
+ self,
1322
+ *,
1323
+ TrackingServerName: str,
1324
+ ArtifactStoreUri: str,
1325
+ RoleArn: str,
1326
+ TrackingServerSize: TrackingServerSizeType = ...,
1327
+ MlflowVersion: str = ...,
1328
+ AutomaticModelRegistration: bool = ...,
1329
+ WeeklyMaintenanceWindowStart: str = ...,
1330
+ Tags: Sequence[TagTypeDef] = ...,
1331
+ ) -> CreateMlflowTrackingServerResponseTypeDef:
1332
+ """
1333
+ Creates an MLflow Tracking Server using a general purpose Amazon S3 bucket as
1334
+ the artifact
1335
+ store.
1336
+
1337
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.create_mlflow_tracking_server)
1338
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_mlflow_tracking_server)
1339
+ """
1340
+
1308
1341
  def create_model(
1309
1342
  self,
1310
1343
  *,
@@ -1564,6 +1597,22 @@ class SageMakerClient(BaseClient):
1564
1597
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_presigned_domain_url)
1565
1598
  """
1566
1599
 
1600
+ def create_presigned_mlflow_tracking_server_url(
1601
+ self,
1602
+ *,
1603
+ TrackingServerName: str,
1604
+ ExpiresInSeconds: int = ...,
1605
+ SessionExpirationDurationInSeconds: int = ...,
1606
+ ) -> CreatePresignedMlflowTrackingServerUrlResponseTypeDef:
1607
+ """
1608
+ Returns a presigned URL that you can use to connect to the MLflow UI attached
1609
+ to your tracking
1610
+ server.
1611
+
1612
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.create_presigned_mlflow_tracking_server_url)
1613
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#create_presigned_mlflow_tracking_server_url)
1614
+ """
1615
+
1567
1616
  def create_presigned_notebook_instance_url(
1568
1617
  self, *, NotebookInstanceName: str, SessionExpirationDurationInSeconds: int = ...
1569
1618
  ) -> CreatePresignedNotebookInstanceUrlOutputTypeDef:
@@ -2059,6 +2108,16 @@ class SageMakerClient(BaseClient):
2059
2108
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#delete_inference_experiment)
2060
2109
  """
2061
2110
 
2111
+ def delete_mlflow_tracking_server(
2112
+ self, *, TrackingServerName: str
2113
+ ) -> DeleteMlflowTrackingServerResponseTypeDef:
2114
+ """
2115
+ Deletes an MLflow Tracking Server.
2116
+
2117
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.delete_mlflow_tracking_server)
2118
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#delete_mlflow_tracking_server)
2119
+ """
2120
+
2062
2121
  def delete_model(self, *, ModelName: str) -> EmptyResponseMetadataTypeDef:
2063
2122
  """
2064
2123
  Deletes a model.
@@ -2602,6 +2661,16 @@ class SageMakerClient(BaseClient):
2602
2661
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#describe_lineage_group)
2603
2662
  """
2604
2663
 
2664
+ def describe_mlflow_tracking_server(
2665
+ self, *, TrackingServerName: str
2666
+ ) -> DescribeMlflowTrackingServerResponseTypeDef:
2667
+ """
2668
+ Returns information about an MLflow Tracking Server.
2669
+
2670
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.describe_mlflow_tracking_server)
2671
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#describe_mlflow_tracking_server)
2672
+ """
2673
+
2605
2674
  def describe_model(self, *, ModelName: str) -> DescribeModelOutputTypeDef:
2606
2675
  """
2607
2676
  Describes a model that you created using the `CreateModel` API.
@@ -3732,6 +3801,25 @@ class SageMakerClient(BaseClient):
3732
3801
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#list_lineage_groups)
3733
3802
  """
3734
3803
 
3804
+ def list_mlflow_tracking_servers(
3805
+ self,
3806
+ *,
3807
+ CreatedAfter: TimestampTypeDef = ...,
3808
+ CreatedBefore: TimestampTypeDef = ...,
3809
+ TrackingServerStatus: TrackingServerStatusType = ...,
3810
+ MlflowVersion: str = ...,
3811
+ SortBy: SortTrackingServerByType = ...,
3812
+ SortOrder: SortOrderType = ...,
3813
+ NextToken: str = ...,
3814
+ MaxResults: int = ...,
3815
+ ) -> ListMlflowTrackingServersResponseTypeDef:
3816
+ """
3817
+ Lists all MLflow Tracking Servers.
3818
+
3819
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.list_mlflow_tracking_servers)
3820
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#list_mlflow_tracking_servers)
3821
+ """
3822
+
3735
3823
  def list_model_bias_job_definitions(
3736
3824
  self,
3737
3825
  *,
@@ -4538,6 +4626,16 @@ class SageMakerClient(BaseClient):
4538
4626
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#start_inference_experiment)
4539
4627
  """
4540
4628
 
4629
+ def start_mlflow_tracking_server(
4630
+ self, *, TrackingServerName: str
4631
+ ) -> StartMlflowTrackingServerResponseTypeDef:
4632
+ """
4633
+ Programmatically start an MLflow Tracking Server.
4634
+
4635
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.start_mlflow_tracking_server)
4636
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#start_mlflow_tracking_server)
4637
+ """
4638
+
4541
4639
  def start_monitoring_schedule(
4542
4640
  self, *, MonitoringScheduleName: str
4543
4641
  ) -> EmptyResponseMetadataTypeDef:
@@ -4654,6 +4752,16 @@ class SageMakerClient(BaseClient):
4654
4752
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#stop_labeling_job)
4655
4753
  """
4656
4754
 
4755
+ def stop_mlflow_tracking_server(
4756
+ self, *, TrackingServerName: str
4757
+ ) -> StopMlflowTrackingServerResponseTypeDef:
4758
+ """
4759
+ Programmatically stop an MLflow Tracking Server.
4760
+
4761
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.stop_mlflow_tracking_server)
4762
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#stop_mlflow_tracking_server)
4763
+ """
4764
+
4657
4765
  def stop_monitoring_schedule(
4658
4766
  self, *, MonitoringScheduleName: str
4659
4767
  ) -> EmptyResponseMetadataTypeDef:
@@ -5017,6 +5125,22 @@ class SageMakerClient(BaseClient):
5017
5125
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#update_inference_experiment)
5018
5126
  """
5019
5127
 
5128
+ def update_mlflow_tracking_server(
5129
+ self,
5130
+ *,
5131
+ TrackingServerName: str,
5132
+ ArtifactStoreUri: str = ...,
5133
+ TrackingServerSize: TrackingServerSizeType = ...,
5134
+ AutomaticModelRegistration: bool = ...,
5135
+ WeeklyMaintenanceWindowStart: str = ...,
5136
+ ) -> UpdateMlflowTrackingServerResponseTypeDef:
5137
+ """
5138
+ Updates properties of an existing MLflow Tracking Server.
5139
+
5140
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.update_mlflow_tracking_server)
5141
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#update_mlflow_tracking_server)
5142
+ """
5143
+
5020
5144
  def update_model_card(
5021
5145
  self, *, ModelCardName: str, Content: str = ..., ModelCardStatus: ModelCardStatusType = ...
5022
5146
  ) -> UpdateModelCardResponseTypeDef:
@@ -5574,6 +5698,15 @@ class SageMakerClient(BaseClient):
5574
5698
  [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#get_paginator)
5575
5699
  """
5576
5700
 
5701
+ @overload
5702
+ def get_paginator(
5703
+ self, operation_name: Literal["list_mlflow_tracking_servers"]
5704
+ ) -> ListMlflowTrackingServersPaginator:
5705
+ """
5706
+ [Show boto3 documentation](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/sagemaker.html#SageMaker.Client.get_paginator)
5707
+ [Show boto3-stubs documentation](https://youtype.github.io/boto3_stubs_docs/mypy_boto3_sagemaker/client/#get_paginator)
5708
+ """
5709
+
5577
5710
  @overload
5578
5711
  def get_paginator(
5579
5712
  self, operation_name: Literal["list_model_bias_job_definitions"]