sagemaker-core 1.0.29__py3-none-any.whl → 1.0.30__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 sagemaker-core might be problematic. Click here for more details.

@@ -14,7 +14,7 @@ import botocore
14
14
  import datetime
15
15
  import time
16
16
  import functools
17
- from pydantic import validate_call
17
+ from pydantic import validate_call, ConfigDict, BaseModel
18
18
  from typing import Dict, List, Literal, Optional, Union, Any
19
19
  from boto3.session import Session
20
20
  from rich.console import Group
@@ -42,8 +42,8 @@ from sagemaker_core.main.intelligent_defaults_helper import (
42
42
  get_config_value,
43
43
  )
44
44
  from sagemaker_core.main.logs import MultiLogStreamHandler
45
- from sagemaker_core.main.shapes import *
46
45
  from sagemaker_core.main.exceptions import *
46
+ import sagemaker_core.main.shapes as shapes
47
47
 
48
48
 
49
49
  logger = get_textual_rich_logger(__name__)
@@ -75,7 +75,9 @@ class Base(BaseModel):
75
75
  configurable_attribute, resource_defaults, global_defaults
76
76
  ):
77
77
  resource_name = snake_to_pascal(configurable_attribute)
78
- class_object = globals()[resource_name]
78
+ class_object = getattr(shapes, resource_name, None) or globals().get(
79
+ resource_name
80
+ )
79
81
  kwargs[configurable_attribute] = class_object(**config_value)
80
82
  except BaseException as e:
81
83
  logger.debug("Could not load Default Configs. Continuing.", exc_info=True)
@@ -121,7 +123,9 @@ class Base(BaseModel):
121
123
  @staticmethod
122
124
  def _get_chained_attribute(item_value: Any):
123
125
  resource_name = type(item_value).__name__
124
- class_object = globals()[resource_name]
126
+ class_object = globals().get(resource_name) or getattr(shapes, resource_name, None)
127
+ if class_object is None:
128
+ return item_value
125
129
  return class_object(
126
130
  **Base.populate_chained_attributes(
127
131
  resource_name=resource_name, operation_input_args=vars(item_value)
@@ -161,16 +165,16 @@ class Action(Base):
161
165
 
162
166
  action_name: str
163
167
  action_arn: Optional[str] = Unassigned()
164
- source: Optional[ActionSource] = Unassigned()
168
+ source: Optional[shapes.ActionSource] = Unassigned()
165
169
  action_type: Optional[str] = Unassigned()
166
170
  description: Optional[str] = Unassigned()
167
171
  status: Optional[str] = Unassigned()
168
172
  properties: Optional[Dict[str, str]] = Unassigned()
169
173
  creation_time: Optional[datetime.datetime] = Unassigned()
170
- created_by: Optional[UserContext] = Unassigned()
174
+ created_by: Optional[shapes.UserContext] = Unassigned()
171
175
  last_modified_time: Optional[datetime.datetime] = Unassigned()
172
- last_modified_by: Optional[UserContext] = Unassigned()
173
- metadata_properties: Optional[MetadataProperties] = Unassigned()
176
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
177
+ metadata_properties: Optional[shapes.MetadataProperties] = Unassigned()
174
178
  lineage_group_arn: Optional[str] = Unassigned()
175
179
 
176
180
  def get_name(self) -> str:
@@ -194,13 +198,13 @@ class Action(Base):
194
198
  def create(
195
199
  cls,
196
200
  action_name: str,
197
- source: ActionSource,
201
+ source: shapes.ActionSource,
198
202
  action_type: str,
199
203
  description: Optional[str] = Unassigned(),
200
204
  status: Optional[str] = Unassigned(),
201
205
  properties: Optional[Dict[str, str]] = Unassigned(),
202
- metadata_properties: Optional[MetadataProperties] = Unassigned(),
203
- tags: Optional[List[Tag]] = Unassigned(),
206
+ metadata_properties: Optional[shapes.MetadataProperties] = Unassigned(),
207
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
204
208
  session: Optional[Session] = None,
205
209
  region: Optional[str] = None,
206
210
  ) -> Optional["Action"]:
@@ -537,11 +541,11 @@ class Algorithm(Base):
537
541
  algorithm_arn: Optional[str] = Unassigned()
538
542
  algorithm_description: Optional[str] = Unassigned()
539
543
  creation_time: Optional[datetime.datetime] = Unassigned()
540
- training_specification: Optional[TrainingSpecification] = Unassigned()
541
- inference_specification: Optional[InferenceSpecification] = Unassigned()
542
- validation_specification: Optional[AlgorithmValidationSpecification] = Unassigned()
544
+ training_specification: Optional[shapes.TrainingSpecification] = Unassigned()
545
+ inference_specification: Optional[shapes.InferenceSpecification] = Unassigned()
546
+ validation_specification: Optional[shapes.AlgorithmValidationSpecification] = Unassigned()
543
547
  algorithm_status: Optional[str] = Unassigned()
544
- algorithm_status_details: Optional[AlgorithmStatusDetails] = Unassigned()
548
+ algorithm_status_details: Optional[shapes.AlgorithmStatusDetails] = Unassigned()
545
549
  product_id: Optional[str] = Unassigned()
546
550
  certify_for_marketplace: Optional[bool] = Unassigned()
547
551
 
@@ -588,12 +592,12 @@ class Algorithm(Base):
588
592
  def create(
589
593
  cls,
590
594
  algorithm_name: str,
591
- training_specification: TrainingSpecification,
595
+ training_specification: shapes.TrainingSpecification,
592
596
  algorithm_description: Optional[str] = Unassigned(),
593
- inference_specification: Optional[InferenceSpecification] = Unassigned(),
594
- validation_specification: Optional[AlgorithmValidationSpecification] = Unassigned(),
597
+ inference_specification: Optional[shapes.InferenceSpecification] = Unassigned(),
598
+ validation_specification: Optional[shapes.AlgorithmValidationSpecification] = Unassigned(),
595
599
  certify_for_marketplace: Optional[bool] = Unassigned(),
596
- tags: Optional[List[Tag]] = Unassigned(),
600
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
597
601
  session: Optional[Session] = None,
598
602
  region: Optional[str] = None,
599
603
  ) -> Optional["Algorithm"]:
@@ -996,7 +1000,7 @@ class App(Base):
996
1000
  last_user_activity_timestamp: Optional[datetime.datetime] = Unassigned()
997
1001
  creation_time: Optional[datetime.datetime] = Unassigned()
998
1002
  failure_reason: Optional[str] = Unassigned()
999
- resource_spec: Optional[ResourceSpec] = Unassigned()
1003
+ resource_spec: Optional[shapes.ResourceSpec] = Unassigned()
1000
1004
  built_in_lifecycle_config_arn: Optional[str] = Unassigned()
1001
1005
 
1002
1006
  def get_name(self) -> str:
@@ -1024,8 +1028,8 @@ class App(Base):
1024
1028
  app_name: str,
1025
1029
  user_profile_name: Optional[Union[str, object]] = Unassigned(),
1026
1030
  space_name: Optional[Union[str, object]] = Unassigned(),
1027
- tags: Optional[List[Tag]] = Unassigned(),
1028
- resource_spec: Optional[ResourceSpec] = Unassigned(),
1031
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
1032
+ resource_spec: Optional[shapes.ResourceSpec] = Unassigned(),
1029
1033
  recovery_mode: Optional[bool] = Unassigned(),
1030
1034
  session: Optional[Session] = None,
1031
1035
  region: Optional[str] = None,
@@ -1451,9 +1455,9 @@ class AppImageConfig(Base):
1451
1455
  app_image_config_arn: Optional[str] = Unassigned()
1452
1456
  creation_time: Optional[datetime.datetime] = Unassigned()
1453
1457
  last_modified_time: Optional[datetime.datetime] = Unassigned()
1454
- kernel_gateway_image_config: Optional[KernelGatewayImageConfig] = Unassigned()
1455
- jupyter_lab_app_image_config: Optional[JupyterLabAppImageConfig] = Unassigned()
1456
- code_editor_app_image_config: Optional[CodeEditorAppImageConfig] = Unassigned()
1458
+ kernel_gateway_image_config: Optional[shapes.KernelGatewayImageConfig] = Unassigned()
1459
+ jupyter_lab_app_image_config: Optional[shapes.JupyterLabAppImageConfig] = Unassigned()
1460
+ code_editor_app_image_config: Optional[shapes.CodeEditorAppImageConfig] = Unassigned()
1457
1461
 
1458
1462
  def get_name(self) -> str:
1459
1463
  attributes = vars(self)
@@ -1476,10 +1480,10 @@ class AppImageConfig(Base):
1476
1480
  def create(
1477
1481
  cls,
1478
1482
  app_image_config_name: str,
1479
- tags: Optional[List[Tag]] = Unassigned(),
1480
- kernel_gateway_image_config: Optional[KernelGatewayImageConfig] = Unassigned(),
1481
- jupyter_lab_app_image_config: Optional[JupyterLabAppImageConfig] = Unassigned(),
1482
- code_editor_app_image_config: Optional[CodeEditorAppImageConfig] = Unassigned(),
1483
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
1484
+ kernel_gateway_image_config: Optional[shapes.KernelGatewayImageConfig] = Unassigned(),
1485
+ jupyter_lab_app_image_config: Optional[shapes.JupyterLabAppImageConfig] = Unassigned(),
1486
+ code_editor_app_image_config: Optional[shapes.CodeEditorAppImageConfig] = Unassigned(),
1483
1487
  session: Optional[Session] = None,
1484
1488
  region: Optional[str] = None,
1485
1489
  ) -> Optional["AppImageConfig"]:
@@ -1633,9 +1637,9 @@ class AppImageConfig(Base):
1633
1637
  @Base.add_validate_call
1634
1638
  def update(
1635
1639
  self,
1636
- kernel_gateway_image_config: Optional[KernelGatewayImageConfig] = Unassigned(),
1637
- jupyter_lab_app_image_config: Optional[JupyterLabAppImageConfig] = Unassigned(),
1638
- code_editor_app_image_config: Optional[CodeEditorAppImageConfig] = Unassigned(),
1640
+ kernel_gateway_image_config: Optional[shapes.KernelGatewayImageConfig] = Unassigned(),
1641
+ jupyter_lab_app_image_config: Optional[shapes.JupyterLabAppImageConfig] = Unassigned(),
1642
+ code_editor_app_image_config: Optional[shapes.CodeEditorAppImageConfig] = Unassigned(),
1639
1643
  ) -> Optional["AppImageConfig"]:
1640
1644
  """
1641
1645
  Update a AppImageConfig resource
@@ -1804,14 +1808,14 @@ class Artifact(Base):
1804
1808
 
1805
1809
  artifact_arn: str
1806
1810
  artifact_name: Optional[str] = Unassigned()
1807
- source: Optional[ArtifactSource] = Unassigned()
1811
+ source: Optional[shapes.ArtifactSource] = Unassigned()
1808
1812
  artifact_type: Optional[str] = Unassigned()
1809
1813
  properties: Optional[Dict[str, str]] = Unassigned()
1810
1814
  creation_time: Optional[datetime.datetime] = Unassigned()
1811
- created_by: Optional[UserContext] = Unassigned()
1815
+ created_by: Optional[shapes.UserContext] = Unassigned()
1812
1816
  last_modified_time: Optional[datetime.datetime] = Unassigned()
1813
- last_modified_by: Optional[UserContext] = Unassigned()
1814
- metadata_properties: Optional[MetadataProperties] = Unassigned()
1817
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
1818
+ metadata_properties: Optional[shapes.MetadataProperties] = Unassigned()
1815
1819
  lineage_group_arn: Optional[str] = Unassigned()
1816
1820
 
1817
1821
  def get_name(self) -> str:
@@ -1834,12 +1838,12 @@ class Artifact(Base):
1834
1838
  @Base.add_validate_call
1835
1839
  def create(
1836
1840
  cls,
1837
- source: ArtifactSource,
1841
+ source: shapes.ArtifactSource,
1838
1842
  artifact_type: str,
1839
1843
  artifact_name: Optional[str] = Unassigned(),
1840
1844
  properties: Optional[Dict[str, str]] = Unassigned(),
1841
- metadata_properties: Optional[MetadataProperties] = Unassigned(),
1842
- tags: Optional[List[Tag]] = Unassigned(),
1845
+ metadata_properties: Optional[shapes.MetadataProperties] = Unassigned(),
1846
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
1843
1847
  session: Optional[Session] = None,
1844
1848
  region: Optional[str] = None,
1845
1849
  ) -> Optional["Artifact"]:
@@ -2173,7 +2177,7 @@ class Association(Base):
2173
2177
  source_name: Optional[str] = Unassigned()
2174
2178
  destination_name: Optional[str] = Unassigned()
2175
2179
  creation_time: Optional[datetime.datetime] = Unassigned()
2176
- created_by: Optional[UserContext] = Unassigned()
2180
+ created_by: Optional[shapes.UserContext] = Unassigned()
2177
2181
 
2178
2182
  def get_name(self) -> str:
2179
2183
  attributes = vars(self)
@@ -2387,25 +2391,25 @@ class AutoMLJob(Base):
2387
2391
 
2388
2392
  auto_ml_job_name: str
2389
2393
  auto_ml_job_arn: Optional[str] = Unassigned()
2390
- input_data_config: Optional[List[AutoMLChannel]] = Unassigned()
2391
- output_data_config: Optional[AutoMLOutputDataConfig] = Unassigned()
2394
+ input_data_config: Optional[List[shapes.AutoMLChannel]] = Unassigned()
2395
+ output_data_config: Optional[shapes.AutoMLOutputDataConfig] = Unassigned()
2392
2396
  role_arn: Optional[str] = Unassigned()
2393
- auto_ml_job_objective: Optional[AutoMLJobObjective] = Unassigned()
2397
+ auto_ml_job_objective: Optional[shapes.AutoMLJobObjective] = Unassigned()
2394
2398
  problem_type: Optional[str] = Unassigned()
2395
- auto_ml_job_config: Optional[AutoMLJobConfig] = Unassigned()
2399
+ auto_ml_job_config: Optional[shapes.AutoMLJobConfig] = Unassigned()
2396
2400
  creation_time: Optional[datetime.datetime] = Unassigned()
2397
2401
  end_time: Optional[datetime.datetime] = Unassigned()
2398
2402
  last_modified_time: Optional[datetime.datetime] = Unassigned()
2399
2403
  failure_reason: Optional[str] = Unassigned()
2400
- partial_failure_reasons: Optional[List[AutoMLPartialFailureReason]] = Unassigned()
2401
- best_candidate: Optional[AutoMLCandidate] = Unassigned()
2404
+ partial_failure_reasons: Optional[List[shapes.AutoMLPartialFailureReason]] = Unassigned()
2405
+ best_candidate: Optional[shapes.AutoMLCandidate] = Unassigned()
2402
2406
  auto_ml_job_status: Optional[str] = Unassigned()
2403
2407
  auto_ml_job_secondary_status: Optional[str] = Unassigned()
2404
2408
  generate_candidate_definitions_only: Optional[bool] = Unassigned()
2405
- auto_ml_job_artifacts: Optional[AutoMLJobArtifacts] = Unassigned()
2406
- resolved_attributes: Optional[ResolvedAttributes] = Unassigned()
2407
- model_deploy_config: Optional[ModelDeployConfig] = Unassigned()
2408
- model_deploy_result: Optional[ModelDeployResult] = Unassigned()
2409
+ auto_ml_job_artifacts: Optional[shapes.AutoMLJobArtifacts] = Unassigned()
2410
+ resolved_attributes: Optional[shapes.ResolvedAttributes] = Unassigned()
2411
+ model_deploy_config: Optional[shapes.ModelDeployConfig] = Unassigned()
2412
+ model_deploy_result: Optional[shapes.ModelDeployResult] = Unassigned()
2409
2413
 
2410
2414
  def get_name(self) -> str:
2411
2415
  attributes = vars(self)
@@ -2460,15 +2464,15 @@ class AutoMLJob(Base):
2460
2464
  def create(
2461
2465
  cls,
2462
2466
  auto_ml_job_name: str,
2463
- input_data_config: List[AutoMLChannel],
2464
- output_data_config: AutoMLOutputDataConfig,
2467
+ input_data_config: List[shapes.AutoMLChannel],
2468
+ output_data_config: shapes.AutoMLOutputDataConfig,
2465
2469
  role_arn: str,
2466
2470
  problem_type: Optional[str] = Unassigned(),
2467
- auto_ml_job_objective: Optional[AutoMLJobObjective] = Unassigned(),
2468
- auto_ml_job_config: Optional[AutoMLJobConfig] = Unassigned(),
2471
+ auto_ml_job_objective: Optional[shapes.AutoMLJobObjective] = Unassigned(),
2472
+ auto_ml_job_config: Optional[shapes.AutoMLJobConfig] = Unassigned(),
2469
2473
  generate_candidate_definitions_only: Optional[bool] = Unassigned(),
2470
- tags: Optional[List[Tag]] = Unassigned(),
2471
- model_deploy_config: Optional[ModelDeployConfig] = Unassigned(),
2474
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
2475
+ model_deploy_config: Optional[shapes.ModelDeployConfig] = Unassigned(),
2472
2476
  session: Optional[Session] = None,
2473
2477
  region: Optional[str] = None,
2474
2478
  ) -> Optional["AutoMLJob"]:
@@ -2804,7 +2808,7 @@ class AutoMLJob(Base):
2804
2808
  sort_by: Optional[str] = Unassigned(),
2805
2809
  session: Optional[Session] = None,
2806
2810
  region: Optional[str] = None,
2807
- ) -> ResourceIterator[AutoMLCandidate]:
2811
+ ) -> ResourceIterator[shapes.AutoMLCandidate]:
2808
2812
  """
2809
2813
  List the candidates created for the job.
2810
2814
 
@@ -2854,7 +2858,7 @@ class AutoMLJob(Base):
2854
2858
  list_method="list_candidates_for_auto_ml_job",
2855
2859
  summaries_key="Candidates",
2856
2860
  summary_name="AutoMLCandidate",
2857
- resource_cls=AutoMLCandidate,
2861
+ resource_cls=shapes.AutoMLCandidate,
2858
2862
  list_method_kwargs=operation_input_args,
2859
2863
  )
2860
2864
 
@@ -2892,27 +2896,27 @@ class AutoMLJobV2(Base):
2892
2896
 
2893
2897
  auto_ml_job_name: str
2894
2898
  auto_ml_job_arn: Optional[str] = Unassigned()
2895
- auto_ml_job_input_data_config: Optional[List[AutoMLJobChannel]] = Unassigned()
2896
- output_data_config: Optional[AutoMLOutputDataConfig] = Unassigned()
2899
+ auto_ml_job_input_data_config: Optional[List[shapes.AutoMLJobChannel]] = Unassigned()
2900
+ output_data_config: Optional[shapes.AutoMLOutputDataConfig] = Unassigned()
2897
2901
  role_arn: Optional[str] = Unassigned()
2898
- auto_ml_job_objective: Optional[AutoMLJobObjective] = Unassigned()
2899
- auto_ml_problem_type_config: Optional[AutoMLProblemTypeConfig] = Unassigned()
2902
+ auto_ml_job_objective: Optional[shapes.AutoMLJobObjective] = Unassigned()
2903
+ auto_ml_problem_type_config: Optional[shapes.AutoMLProblemTypeConfig] = Unassigned()
2900
2904
  auto_ml_problem_type_config_name: Optional[str] = Unassigned()
2901
2905
  creation_time: Optional[datetime.datetime] = Unassigned()
2902
2906
  end_time: Optional[datetime.datetime] = Unassigned()
2903
2907
  last_modified_time: Optional[datetime.datetime] = Unassigned()
2904
2908
  failure_reason: Optional[str] = Unassigned()
2905
- partial_failure_reasons: Optional[List[AutoMLPartialFailureReason]] = Unassigned()
2906
- best_candidate: Optional[AutoMLCandidate] = Unassigned()
2909
+ partial_failure_reasons: Optional[List[shapes.AutoMLPartialFailureReason]] = Unassigned()
2910
+ best_candidate: Optional[shapes.AutoMLCandidate] = Unassigned()
2907
2911
  auto_ml_job_status: Optional[str] = Unassigned()
2908
2912
  auto_ml_job_secondary_status: Optional[str] = Unassigned()
2909
- auto_ml_job_artifacts: Optional[AutoMLJobArtifacts] = Unassigned()
2910
- resolved_attributes: Optional[AutoMLResolvedAttributes] = Unassigned()
2911
- model_deploy_config: Optional[ModelDeployConfig] = Unassigned()
2912
- model_deploy_result: Optional[ModelDeployResult] = Unassigned()
2913
- data_split_config: Optional[AutoMLDataSplitConfig] = Unassigned()
2914
- security_config: Optional[AutoMLSecurityConfig] = Unassigned()
2915
- auto_ml_compute_config: Optional[AutoMLComputeConfig] = Unassigned()
2913
+ auto_ml_job_artifacts: Optional[shapes.AutoMLJobArtifacts] = Unassigned()
2914
+ resolved_attributes: Optional[shapes.AutoMLResolvedAttributes] = Unassigned()
2915
+ model_deploy_config: Optional[shapes.ModelDeployConfig] = Unassigned()
2916
+ model_deploy_result: Optional[shapes.ModelDeployResult] = Unassigned()
2917
+ data_split_config: Optional[shapes.AutoMLDataSplitConfig] = Unassigned()
2918
+ security_config: Optional[shapes.AutoMLSecurityConfig] = Unassigned()
2919
+ auto_ml_compute_config: Optional[shapes.AutoMLComputeConfig] = Unassigned()
2916
2920
 
2917
2921
  def get_name(self) -> str:
2918
2922
  attributes = vars(self)
@@ -2971,16 +2975,16 @@ class AutoMLJobV2(Base):
2971
2975
  def create(
2972
2976
  cls,
2973
2977
  auto_ml_job_name: str,
2974
- auto_ml_job_input_data_config: List[AutoMLJobChannel],
2975
- output_data_config: AutoMLOutputDataConfig,
2976
- auto_ml_problem_type_config: AutoMLProblemTypeConfig,
2978
+ auto_ml_job_input_data_config: List[shapes.AutoMLJobChannel],
2979
+ output_data_config: shapes.AutoMLOutputDataConfig,
2980
+ auto_ml_problem_type_config: shapes.AutoMLProblemTypeConfig,
2977
2981
  role_arn: str,
2978
- tags: Optional[List[Tag]] = Unassigned(),
2979
- security_config: Optional[AutoMLSecurityConfig] = Unassigned(),
2980
- auto_ml_job_objective: Optional[AutoMLJobObjective] = Unassigned(),
2981
- model_deploy_config: Optional[ModelDeployConfig] = Unassigned(),
2982
- data_split_config: Optional[AutoMLDataSplitConfig] = Unassigned(),
2983
- auto_ml_compute_config: Optional[AutoMLComputeConfig] = Unassigned(),
2982
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
2983
+ security_config: Optional[shapes.AutoMLSecurityConfig] = Unassigned(),
2984
+ auto_ml_job_objective: Optional[shapes.AutoMLJobObjective] = Unassigned(),
2985
+ model_deploy_config: Optional[shapes.ModelDeployConfig] = Unassigned(),
2986
+ data_split_config: Optional[shapes.AutoMLDataSplitConfig] = Unassigned(),
2987
+ auto_ml_compute_config: Optional[shapes.AutoMLComputeConfig] = Unassigned(),
2984
2988
  session: Optional[Session] = None,
2985
2989
  region: Optional[str] = None,
2986
2990
  ) -> Optional["AutoMLJobV2"]:
@@ -3226,9 +3230,9 @@ class Cluster(Base):
3226
3230
  cluster_status: Optional[str] = Unassigned()
3227
3231
  creation_time: Optional[datetime.datetime] = Unassigned()
3228
3232
  failure_message: Optional[str] = Unassigned()
3229
- instance_groups: Optional[List[ClusterInstanceGroupDetails]] = Unassigned()
3230
- vpc_config: Optional[VpcConfig] = Unassigned()
3231
- orchestrator: Optional[ClusterOrchestrator] = Unassigned()
3233
+ instance_groups: Optional[List[shapes.ClusterInstanceGroupDetails]] = Unassigned()
3234
+ vpc_config: Optional[shapes.VpcConfig] = Unassigned()
3235
+ orchestrator: Optional[shapes.ClusterOrchestrator] = Unassigned()
3232
3236
  node_recovery: Optional[str] = Unassigned()
3233
3237
 
3234
3238
  def get_name(self) -> str:
@@ -3271,10 +3275,10 @@ class Cluster(Base):
3271
3275
  def create(
3272
3276
  cls,
3273
3277
  cluster_name: str,
3274
- instance_groups: List[ClusterInstanceGroupSpecification],
3275
- vpc_config: Optional[VpcConfig] = Unassigned(),
3276
- tags: Optional[List[Tag]] = Unassigned(),
3277
- orchestrator: Optional[ClusterOrchestrator] = Unassigned(),
3278
+ instance_groups: List[shapes.ClusterInstanceGroupSpecification],
3279
+ vpc_config: Optional[shapes.VpcConfig] = Unassigned(),
3280
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
3281
+ orchestrator: Optional[shapes.ClusterOrchestrator] = Unassigned(),
3278
3282
  node_recovery: Optional[str] = Unassigned(),
3279
3283
  session: Optional[Session] = None,
3280
3284
  region: Optional[str] = None,
@@ -3433,7 +3437,7 @@ class Cluster(Base):
3433
3437
  @Base.add_validate_call
3434
3438
  def update(
3435
3439
  self,
3436
- instance_groups: List[ClusterInstanceGroupSpecification],
3440
+ instance_groups: List[shapes.ClusterInstanceGroupSpecification],
3437
3441
  node_recovery: Optional[str] = Unassigned(),
3438
3442
  instance_groups_to_delete: Optional[List[str]] = Unassigned(),
3439
3443
  ) -> Optional["Cluster"]:
@@ -3716,7 +3720,7 @@ class Cluster(Base):
3716
3720
  node_id: str,
3717
3721
  session: Optional[Session] = None,
3718
3722
  region: Optional[str] = None,
3719
- ) -> Optional[ClusterNodeDetails]:
3723
+ ) -> Optional[shapes.ClusterNodeDetails]:
3720
3724
  """
3721
3725
  Retrieves information of a node (also called a instance interchangeably) of a SageMaker HyperPod cluster.
3722
3726
 
@@ -3726,7 +3730,7 @@ class Cluster(Base):
3726
3730
  region: Region name.
3727
3731
 
3728
3732
  Returns:
3729
- ClusterNodeDetails
3733
+ shapes.ClusterNodeDetails
3730
3734
 
3731
3735
  Raises:
3732
3736
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -3758,7 +3762,7 @@ class Cluster(Base):
3758
3762
  logger.debug(f"Response: {response}")
3759
3763
 
3760
3764
  transformed_response = transform(response, "DescribeClusterNodeResponse")
3761
- return ClusterNodeDetails(**transformed_response)
3765
+ return shapes.ClusterNodeDetails(**transformed_response)
3762
3766
 
3763
3767
  @Base.add_validate_call
3764
3768
  def get_all_nodes(
@@ -3770,7 +3774,7 @@ class Cluster(Base):
3770
3774
  sort_order: Optional[str] = Unassigned(),
3771
3775
  session: Optional[Session] = None,
3772
3776
  region: Optional[str] = None,
3773
- ) -> ResourceIterator[ClusterNodeDetails]:
3777
+ ) -> ResourceIterator[shapes.ClusterNodeDetails]:
3774
3778
  """
3775
3779
  Retrieves the list of instances (also called nodes interchangeably) in a SageMaker HyperPod cluster.
3776
3780
 
@@ -3822,7 +3826,7 @@ class Cluster(Base):
3822
3826
  list_method="list_cluster_nodes",
3823
3827
  summaries_key="ClusterNodeSummaries",
3824
3828
  summary_name="ClusterNodeSummary",
3825
- resource_cls=ClusterNodeDetails,
3829
+ resource_cls=shapes.ClusterNodeDetails,
3826
3830
  list_method_kwargs=operation_input_args,
3827
3831
  )
3828
3832
 
@@ -3874,7 +3878,7 @@ class Cluster(Base):
3874
3878
  node_ids: List[str],
3875
3879
  session: Optional[Session] = None,
3876
3880
  region: Optional[str] = None,
3877
- ) -> Optional[BatchDeleteClusterNodesResponse]:
3881
+ ) -> Optional[shapes.BatchDeleteClusterNodesResponse]:
3878
3882
  """
3879
3883
  Deletes specific nodes within a SageMaker HyperPod cluster.
3880
3884
 
@@ -3884,7 +3888,7 @@ class Cluster(Base):
3884
3888
  region: Region name.
3885
3889
 
3886
3890
  Returns:
3887
- BatchDeleteClusterNodesResponse
3891
+ shapes.BatchDeleteClusterNodesResponse
3888
3892
 
3889
3893
  Raises:
3890
3894
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -3916,7 +3920,7 @@ class Cluster(Base):
3916
3920
  logger.debug(f"Response: {response}")
3917
3921
 
3918
3922
  transformed_response = transform(response, "BatchDeleteClusterNodesResponse")
3919
- return BatchDeleteClusterNodesResponse(**transformed_response)
3923
+ return shapes.BatchDeleteClusterNodesResponse(**transformed_response)
3920
3924
 
3921
3925
 
3922
3926
  class ClusterSchedulerConfig(Base):
@@ -3947,12 +3951,12 @@ class ClusterSchedulerConfig(Base):
3947
3951
  status: Optional[str] = Unassigned()
3948
3952
  failure_reason: Optional[str] = Unassigned()
3949
3953
  cluster_arn: Optional[str] = Unassigned()
3950
- scheduler_config: Optional[SchedulerConfig] = Unassigned()
3954
+ scheduler_config: Optional[shapes.SchedulerConfig] = Unassigned()
3951
3955
  description: Optional[str] = Unassigned()
3952
3956
  creation_time: Optional[datetime.datetime] = Unassigned()
3953
- created_by: Optional[UserContext] = Unassigned()
3957
+ created_by: Optional[shapes.UserContext] = Unassigned()
3954
3958
  last_modified_time: Optional[datetime.datetime] = Unassigned()
3955
- last_modified_by: Optional[UserContext] = Unassigned()
3959
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
3956
3960
 
3957
3961
  def get_name(self) -> str:
3958
3962
  attributes = vars(self)
@@ -3976,9 +3980,9 @@ class ClusterSchedulerConfig(Base):
3976
3980
  cls,
3977
3981
  name: str,
3978
3982
  cluster_arn: str,
3979
- scheduler_config: SchedulerConfig,
3983
+ scheduler_config: shapes.SchedulerConfig,
3980
3984
  description: Optional[str] = Unassigned(),
3981
- tags: Optional[List[Tag]] = Unassigned(),
3985
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
3982
3986
  session: Optional[Session] = None,
3983
3987
  region: Optional[str] = None,
3984
3988
  ) -> Optional["ClusterSchedulerConfig"]:
@@ -4142,7 +4146,7 @@ class ClusterSchedulerConfig(Base):
4142
4146
  def update(
4143
4147
  self,
4144
4148
  target_version: int,
4145
- scheduler_config: Optional[SchedulerConfig] = Unassigned(),
4149
+ scheduler_config: Optional[shapes.SchedulerConfig] = Unassigned(),
4146
4150
  description: Optional[str] = Unassigned(),
4147
4151
  ) -> Optional["ClusterSchedulerConfig"]:
4148
4152
  """
@@ -4455,7 +4459,7 @@ class CodeRepository(Base):
4455
4459
  code_repository_arn: Optional[str] = Unassigned()
4456
4460
  creation_time: Optional[datetime.datetime] = Unassigned()
4457
4461
  last_modified_time: Optional[datetime.datetime] = Unassigned()
4458
- git_config: Optional[GitConfig] = Unassigned()
4462
+ git_config: Optional[shapes.GitConfig] = Unassigned()
4459
4463
 
4460
4464
  def get_name(self) -> str:
4461
4465
  attributes = vars(self)
@@ -4478,8 +4482,8 @@ class CodeRepository(Base):
4478
4482
  def create(
4479
4483
  cls,
4480
4484
  code_repository_name: str,
4481
- git_config: GitConfig,
4482
- tags: Optional[List[Tag]] = Unassigned(),
4485
+ git_config: shapes.GitConfig,
4486
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
4483
4487
  session: Optional[Session] = None,
4484
4488
  region: Optional[str] = None,
4485
4489
  ) -> Optional["CodeRepository"]:
@@ -4626,7 +4630,7 @@ class CodeRepository(Base):
4626
4630
  @Base.add_validate_call
4627
4631
  def update(
4628
4632
  self,
4629
- git_config: Optional[GitConfigForUpdate] = Unassigned(),
4633
+ git_config: Optional[shapes.GitConfigForUpdate] = Unassigned(),
4630
4634
  ) -> Optional["CodeRepository"]:
4631
4635
  """
4632
4636
  Update a CodeRepository resource
@@ -4801,19 +4805,19 @@ class CompilationJob(Base):
4801
4805
  compilation_job_status: Optional[str] = Unassigned()
4802
4806
  compilation_start_time: Optional[datetime.datetime] = Unassigned()
4803
4807
  compilation_end_time: Optional[datetime.datetime] = Unassigned()
4804
- stopping_condition: Optional[StoppingCondition] = Unassigned()
4808
+ stopping_condition: Optional[shapes.StoppingCondition] = Unassigned()
4805
4809
  inference_image: Optional[str] = Unassigned()
4806
4810
  model_package_version_arn: Optional[str] = Unassigned()
4807
4811
  creation_time: Optional[datetime.datetime] = Unassigned()
4808
4812
  last_modified_time: Optional[datetime.datetime] = Unassigned()
4809
4813
  failure_reason: Optional[str] = Unassigned()
4810
- model_artifacts: Optional[ModelArtifacts] = Unassigned()
4811
- model_digests: Optional[ModelDigests] = Unassigned()
4814
+ model_artifacts: Optional[shapes.ModelArtifacts] = Unassigned()
4815
+ model_digests: Optional[shapes.ModelDigests] = Unassigned()
4812
4816
  role_arn: Optional[str] = Unassigned()
4813
- input_config: Optional[InputConfig] = Unassigned()
4814
- output_config: Optional[OutputConfig] = Unassigned()
4815
- vpc_config: Optional[NeoVpcConfig] = Unassigned()
4816
- derived_information: Optional[DerivedInformation] = Unassigned()
4817
+ input_config: Optional[shapes.InputConfig] = Unassigned()
4818
+ output_config: Optional[shapes.OutputConfig] = Unassigned()
4819
+ vpc_config: Optional[shapes.NeoVpcConfig] = Unassigned()
4820
+ derived_information: Optional[shapes.DerivedInformation] = Unassigned()
4817
4821
 
4818
4822
  def get_name(self) -> str:
4819
4823
  attributes = vars(self)
@@ -4863,12 +4867,12 @@ class CompilationJob(Base):
4863
4867
  cls,
4864
4868
  compilation_job_name: str,
4865
4869
  role_arn: str,
4866
- output_config: OutputConfig,
4867
- stopping_condition: StoppingCondition,
4870
+ output_config: shapes.OutputConfig,
4871
+ stopping_condition: shapes.StoppingCondition,
4868
4872
  model_package_version_arn: Optional[str] = Unassigned(),
4869
- input_config: Optional[InputConfig] = Unassigned(),
4870
- vpc_config: Optional[NeoVpcConfig] = Unassigned(),
4871
- tags: Optional[List[Tag]] = Unassigned(),
4873
+ input_config: Optional[shapes.InputConfig] = Unassigned(),
4874
+ vpc_config: Optional[shapes.NeoVpcConfig] = Unassigned(),
4875
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
4872
4876
  session: Optional[Session] = None,
4873
4877
  region: Optional[str] = None,
4874
4878
  ) -> Optional["CompilationJob"]:
@@ -5256,13 +5260,13 @@ class ComputeQuota(Base):
5256
5260
  status: Optional[str] = Unassigned()
5257
5261
  failure_reason: Optional[str] = Unassigned()
5258
5262
  cluster_arn: Optional[str] = Unassigned()
5259
- compute_quota_config: Optional[ComputeQuotaConfig] = Unassigned()
5260
- compute_quota_target: Optional[ComputeQuotaTarget] = Unassigned()
5263
+ compute_quota_config: Optional[shapes.ComputeQuotaConfig] = Unassigned()
5264
+ compute_quota_target: Optional[shapes.ComputeQuotaTarget] = Unassigned()
5261
5265
  activation_state: Optional[str] = Unassigned()
5262
5266
  creation_time: Optional[datetime.datetime] = Unassigned()
5263
- created_by: Optional[UserContext] = Unassigned()
5267
+ created_by: Optional[shapes.UserContext] = Unassigned()
5264
5268
  last_modified_time: Optional[datetime.datetime] = Unassigned()
5265
- last_modified_by: Optional[UserContext] = Unassigned()
5269
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
5266
5270
 
5267
5271
  def get_name(self) -> str:
5268
5272
  attributes = vars(self)
@@ -5286,11 +5290,11 @@ class ComputeQuota(Base):
5286
5290
  cls,
5287
5291
  name: str,
5288
5292
  cluster_arn: str,
5289
- compute_quota_config: ComputeQuotaConfig,
5290
- compute_quota_target: ComputeQuotaTarget,
5293
+ compute_quota_config: shapes.ComputeQuotaConfig,
5294
+ compute_quota_target: shapes.ComputeQuotaTarget,
5291
5295
  description: Optional[str] = Unassigned(),
5292
5296
  activation_state: Optional[str] = Unassigned(),
5293
- tags: Optional[List[Tag]] = Unassigned(),
5297
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
5294
5298
  session: Optional[Session] = None,
5295
5299
  region: Optional[str] = None,
5296
5300
  ) -> Optional["ComputeQuota"]:
@@ -5454,8 +5458,8 @@ class ComputeQuota(Base):
5454
5458
  def update(
5455
5459
  self,
5456
5460
  target_version: int,
5457
- compute_quota_config: Optional[ComputeQuotaConfig] = Unassigned(),
5458
- compute_quota_target: Optional[ComputeQuotaTarget] = Unassigned(),
5461
+ compute_quota_config: Optional[shapes.ComputeQuotaConfig] = Unassigned(),
5462
+ compute_quota_target: Optional[shapes.ComputeQuotaTarget] = Unassigned(),
5459
5463
  activation_state: Optional[str] = Unassigned(),
5460
5464
  description: Optional[str] = Unassigned(),
5461
5465
  ) -> Optional["ComputeQuota"]:
@@ -5771,14 +5775,14 @@ class Context(Base):
5771
5775
 
5772
5776
  context_name: str
5773
5777
  context_arn: Optional[str] = Unassigned()
5774
- source: Optional[ContextSource] = Unassigned()
5778
+ source: Optional[shapes.ContextSource] = Unassigned()
5775
5779
  context_type: Optional[str] = Unassigned()
5776
5780
  description: Optional[str] = Unassigned()
5777
5781
  properties: Optional[Dict[str, str]] = Unassigned()
5778
5782
  creation_time: Optional[datetime.datetime] = Unassigned()
5779
- created_by: Optional[UserContext] = Unassigned()
5783
+ created_by: Optional[shapes.UserContext] = Unassigned()
5780
5784
  last_modified_time: Optional[datetime.datetime] = Unassigned()
5781
- last_modified_by: Optional[UserContext] = Unassigned()
5785
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
5782
5786
  lineage_group_arn: Optional[str] = Unassigned()
5783
5787
 
5784
5788
  def get_name(self) -> str:
@@ -5802,11 +5806,11 @@ class Context(Base):
5802
5806
  def create(
5803
5807
  cls,
5804
5808
  context_name: str,
5805
- source: ContextSource,
5809
+ source: shapes.ContextSource,
5806
5810
  context_type: str,
5807
5811
  description: Optional[str] = Unassigned(),
5808
5812
  properties: Optional[Dict[str, str]] = Unassigned(),
5809
- tags: Optional[List[Tag]] = Unassigned(),
5813
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
5810
5814
  session: Optional[Session] = None,
5811
5815
  region: Optional[str] = None,
5812
5816
  ) -> Optional["Context"]:
@@ -6136,14 +6140,14 @@ class DataQualityJobDefinition(Base):
6136
6140
  job_definition_name: str
6137
6141
  job_definition_arn: Optional[str] = Unassigned()
6138
6142
  creation_time: Optional[datetime.datetime] = Unassigned()
6139
- data_quality_baseline_config: Optional[DataQualityBaselineConfig] = Unassigned()
6140
- data_quality_app_specification: Optional[DataQualityAppSpecification] = Unassigned()
6141
- data_quality_job_input: Optional[DataQualityJobInput] = Unassigned()
6142
- data_quality_job_output_config: Optional[MonitoringOutputConfig] = Unassigned()
6143
- job_resources: Optional[MonitoringResources] = Unassigned()
6144
- network_config: Optional[MonitoringNetworkConfig] = Unassigned()
6143
+ data_quality_baseline_config: Optional[shapes.DataQualityBaselineConfig] = Unassigned()
6144
+ data_quality_app_specification: Optional[shapes.DataQualityAppSpecification] = Unassigned()
6145
+ data_quality_job_input: Optional[shapes.DataQualityJobInput] = Unassigned()
6146
+ data_quality_job_output_config: Optional[shapes.MonitoringOutputConfig] = Unassigned()
6147
+ job_resources: Optional[shapes.MonitoringResources] = Unassigned()
6148
+ network_config: Optional[shapes.MonitoringNetworkConfig] = Unassigned()
6145
6149
  role_arn: Optional[str] = Unassigned()
6146
- stopping_condition: Optional[MonitoringStoppingCondition] = Unassigned()
6150
+ stopping_condition: Optional[shapes.MonitoringStoppingCondition] = Unassigned()
6147
6151
 
6148
6152
  def get_name(self) -> str:
6149
6153
  attributes = vars(self)
@@ -6205,15 +6209,15 @@ class DataQualityJobDefinition(Base):
6205
6209
  def create(
6206
6210
  cls,
6207
6211
  job_definition_name: str,
6208
- data_quality_app_specification: DataQualityAppSpecification,
6209
- data_quality_job_input: DataQualityJobInput,
6210
- data_quality_job_output_config: MonitoringOutputConfig,
6211
- job_resources: MonitoringResources,
6212
+ data_quality_app_specification: shapes.DataQualityAppSpecification,
6213
+ data_quality_job_input: shapes.DataQualityJobInput,
6214
+ data_quality_job_output_config: shapes.MonitoringOutputConfig,
6215
+ job_resources: shapes.MonitoringResources,
6212
6216
  role_arn: str,
6213
- data_quality_baseline_config: Optional[DataQualityBaselineConfig] = Unassigned(),
6214
- network_config: Optional[MonitoringNetworkConfig] = Unassigned(),
6215
- stopping_condition: Optional[MonitoringStoppingCondition] = Unassigned(),
6216
- tags: Optional[List[Tag]] = Unassigned(),
6217
+ data_quality_baseline_config: Optional[shapes.DataQualityBaselineConfig] = Unassigned(),
6218
+ network_config: Optional[shapes.MonitoringNetworkConfig] = Unassigned(),
6219
+ stopping_condition: Optional[shapes.MonitoringStoppingCondition] = Unassigned(),
6220
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
6217
6221
  session: Optional[Session] = None,
6218
6222
  region: Optional[str] = None,
6219
6223
  ) -> Optional["DataQualityJobDefinition"]:
@@ -6508,7 +6512,7 @@ class Device(Base):
6508
6512
  iot_thing_name: Optional[str] = Unassigned()
6509
6513
  registration_time: Optional[datetime.datetime] = Unassigned()
6510
6514
  latest_heartbeat: Optional[datetime.datetime] = Unassigned()
6511
- models: Optional[List[EdgeModel]] = Unassigned()
6515
+ models: Optional[List[shapes.EdgeModel]] = Unassigned()
6512
6516
  max_models: Optional[int] = Unassigned()
6513
6517
  next_token: Optional[str] = Unassigned()
6514
6518
  agent_version: Optional[str] = Unassigned()
@@ -6704,7 +6708,7 @@ class DeviceFleet(Base):
6704
6708
 
6705
6709
  device_fleet_name: str
6706
6710
  device_fleet_arn: Optional[str] = Unassigned()
6707
- output_config: Optional[EdgeOutputConfig] = Unassigned()
6711
+ output_config: Optional[shapes.EdgeOutputConfig] = Unassigned()
6708
6712
  description: Optional[str] = Unassigned()
6709
6713
  creation_time: Optional[datetime.datetime] = Unassigned()
6710
6714
  last_modified_time: Optional[datetime.datetime] = Unassigned()
@@ -6753,10 +6757,10 @@ class DeviceFleet(Base):
6753
6757
  def create(
6754
6758
  cls,
6755
6759
  device_fleet_name: str,
6756
- output_config: EdgeOutputConfig,
6760
+ output_config: shapes.EdgeOutputConfig,
6757
6761
  role_arn: Optional[str] = Unassigned(),
6758
6762
  description: Optional[str] = Unassigned(),
6759
- tags: Optional[List[Tag]] = Unassigned(),
6763
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
6760
6764
  enable_iot_role_alias: Optional[bool] = Unassigned(),
6761
6765
  session: Optional[Session] = None,
6762
6766
  region: Optional[str] = None,
@@ -6915,7 +6919,7 @@ class DeviceFleet(Base):
6915
6919
  @Base.add_validate_call
6916
6920
  def update(
6917
6921
  self,
6918
- output_config: EdgeOutputConfig,
6922
+ output_config: shapes.EdgeOutputConfig,
6919
6923
  role_arn: Optional[str] = Unassigned(),
6920
6924
  description: Optional[str] = Unassigned(),
6921
6925
  enable_iot_role_alias: Optional[bool] = Unassigned(),
@@ -7117,7 +7121,7 @@ class DeviceFleet(Base):
7117
7121
  self,
7118
7122
  session: Optional[Session] = None,
7119
7123
  region: Optional[str] = None,
7120
- ) -> Optional[GetDeviceFleetReportResponse]:
7124
+ ) -> Optional[shapes.GetDeviceFleetReportResponse]:
7121
7125
  """
7122
7126
  Describes a fleet.
7123
7127
 
@@ -7126,7 +7130,7 @@ class DeviceFleet(Base):
7126
7130
  region: Region name.
7127
7131
 
7128
7132
  Returns:
7129
- GetDeviceFleetReportResponse
7133
+ shapes.GetDeviceFleetReportResponse
7130
7134
 
7131
7135
  Raises:
7132
7136
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -7156,13 +7160,13 @@ class DeviceFleet(Base):
7156
7160
  logger.debug(f"Response: {response}")
7157
7161
 
7158
7162
  transformed_response = transform(response, "GetDeviceFleetReportResponse")
7159
- return GetDeviceFleetReportResponse(**transformed_response)
7163
+ return shapes.GetDeviceFleetReportResponse(**transformed_response)
7160
7164
 
7161
7165
  @Base.add_validate_call
7162
7166
  def register_devices(
7163
7167
  self,
7164
- devices: List[Device],
7165
- tags: Optional[List[Tag]] = Unassigned(),
7168
+ devices: List[shapes.Device],
7169
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
7166
7170
  session: Optional[Session] = None,
7167
7171
  region: Optional[str] = None,
7168
7172
  ) -> None:
@@ -7208,7 +7212,7 @@ class DeviceFleet(Base):
7208
7212
  @Base.add_validate_call
7209
7213
  def update_devices(
7210
7214
  self,
7211
- devices: List[Device],
7215
+ devices: List[shapes.Device],
7212
7216
  session: Optional[Session] = None,
7213
7217
  region: Optional[str] = None,
7214
7218
  ) -> None:
@@ -7292,8 +7296,8 @@ class Domain(Base):
7292
7296
  failure_reason: Optional[str] = Unassigned()
7293
7297
  security_group_id_for_domain_boundary: Optional[str] = Unassigned()
7294
7298
  auth_mode: Optional[str] = Unassigned()
7295
- default_user_settings: Optional[UserSettings] = Unassigned()
7296
- domain_settings: Optional[DomainSettings] = Unassigned()
7299
+ default_user_settings: Optional[shapes.UserSettings] = Unassigned()
7300
+ domain_settings: Optional[shapes.DomainSettings] = Unassigned()
7297
7301
  app_network_access_type: Optional[str] = Unassigned()
7298
7302
  home_efs_file_system_kms_key_id: Optional[str] = Unassigned()
7299
7303
  subnet_ids: Optional[List[str]] = Unassigned()
@@ -7302,7 +7306,7 @@ class Domain(Base):
7302
7306
  kms_key_id: Optional[str] = Unassigned()
7303
7307
  app_security_group_management: Optional[str] = Unassigned()
7304
7308
  tag_propagation: Optional[str] = Unassigned()
7305
- default_space_settings: Optional[DefaultSpaceSettings] = Unassigned()
7309
+ default_space_settings: Optional[shapes.DefaultSpaceSettings] = Unassigned()
7306
7310
 
7307
7311
  def get_name(self) -> str:
7308
7312
  attributes = vars(self)
@@ -7391,17 +7395,17 @@ class Domain(Base):
7391
7395
  cls,
7392
7396
  domain_name: str,
7393
7397
  auth_mode: str,
7394
- default_user_settings: UserSettings,
7398
+ default_user_settings: shapes.UserSettings,
7395
7399
  subnet_ids: List[str],
7396
7400
  vpc_id: str,
7397
- domain_settings: Optional[DomainSettings] = Unassigned(),
7398
- tags: Optional[List[Tag]] = Unassigned(),
7401
+ domain_settings: Optional[shapes.DomainSettings] = Unassigned(),
7402
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
7399
7403
  app_network_access_type: Optional[str] = Unassigned(),
7400
7404
  home_efs_file_system_kms_key_id: Optional[str] = Unassigned(),
7401
7405
  kms_key_id: Optional[str] = Unassigned(),
7402
7406
  app_security_group_management: Optional[str] = Unassigned(),
7403
7407
  tag_propagation: Optional[str] = Unassigned(),
7404
- default_space_settings: Optional[DefaultSpaceSettings] = Unassigned(),
7408
+ default_space_settings: Optional[shapes.DefaultSpaceSettings] = Unassigned(),
7405
7409
  session: Optional[Session] = None,
7406
7410
  region: Optional[str] = None,
7407
7411
  ) -> Optional["Domain"]:
@@ -7573,10 +7577,10 @@ class Domain(Base):
7573
7577
  @Base.add_validate_call
7574
7578
  def update(
7575
7579
  self,
7576
- default_user_settings: Optional[UserSettings] = Unassigned(),
7577
- domain_settings_for_update: Optional[DomainSettingsForUpdate] = Unassigned(),
7580
+ default_user_settings: Optional[shapes.UserSettings] = Unassigned(),
7581
+ domain_settings_for_update: Optional[shapes.DomainSettingsForUpdate] = Unassigned(),
7578
7582
  app_security_group_management: Optional[str] = Unassigned(),
7579
- default_space_settings: Optional[DefaultSpaceSettings] = Unassigned(),
7583
+ default_space_settings: Optional[shapes.DefaultSpaceSettings] = Unassigned(),
7580
7584
  subnet_ids: Optional[List[str]] = Unassigned(),
7581
7585
  app_network_access_type: Optional[str] = Unassigned(),
7582
7586
  tag_propagation: Optional[str] = Unassigned(),
@@ -7633,7 +7637,7 @@ class Domain(Base):
7633
7637
  @Base.add_validate_call
7634
7638
  def delete(
7635
7639
  self,
7636
- retention_policy: Optional[RetentionPolicy] = Unassigned(),
7640
+ retention_policy: Optional[shapes.RetentionPolicy] = Unassigned(),
7637
7641
  ) -> None:
7638
7642
  """
7639
7643
  Delete a Domain resource
@@ -7851,12 +7855,12 @@ class EdgeDeploymentPlan(Base):
7851
7855
 
7852
7856
  edge_deployment_plan_name: str
7853
7857
  edge_deployment_plan_arn: Optional[str] = Unassigned()
7854
- model_configs: Optional[List[EdgeDeploymentModelConfig]] = Unassigned()
7858
+ model_configs: Optional[List[shapes.EdgeDeploymentModelConfig]] = Unassigned()
7855
7859
  device_fleet_name: Optional[str] = Unassigned()
7856
7860
  edge_deployment_success: Optional[int] = Unassigned()
7857
7861
  edge_deployment_pending: Optional[int] = Unassigned()
7858
7862
  edge_deployment_failed: Optional[int] = Unassigned()
7859
- stages: Optional[List[DeploymentStageStatusSummary]] = Unassigned()
7863
+ stages: Optional[List[shapes.DeploymentStageStatusSummary]] = Unassigned()
7860
7864
  next_token: Optional[str] = Unassigned()
7861
7865
  creation_time: Optional[datetime.datetime] = Unassigned()
7862
7866
  last_modified_time: Optional[datetime.datetime] = Unassigned()
@@ -7882,10 +7886,10 @@ class EdgeDeploymentPlan(Base):
7882
7886
  def create(
7883
7887
  cls,
7884
7888
  edge_deployment_plan_name: str,
7885
- model_configs: List[EdgeDeploymentModelConfig],
7889
+ model_configs: List[shapes.EdgeDeploymentModelConfig],
7886
7890
  device_fleet_name: Union[str, object],
7887
- stages: Optional[List[DeploymentStage]] = Unassigned(),
7888
- tags: Optional[List[Tag]] = Unassigned(),
7891
+ stages: Optional[List[shapes.DeploymentStage]] = Unassigned(),
7892
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
7889
7893
  session: Optional[Session] = None,
7890
7894
  region: Optional[str] = None,
7891
7895
  ) -> Optional["EdgeDeploymentPlan"]:
@@ -8334,7 +8338,7 @@ class EdgeDeploymentPlan(Base):
8334
8338
  exclude_devices_deployed_in_other_stage: Optional[bool] = Unassigned(),
8335
8339
  session: Optional[Session] = None,
8336
8340
  region: Optional[str] = None,
8337
- ) -> ResourceIterator[DeviceDeploymentSummary]:
8341
+ ) -> ResourceIterator[shapes.DeviceDeploymentSummary]:
8338
8342
  """
8339
8343
  Lists devices allocated to the stage, containing detailed device information and deployment status.
8340
8344
 
@@ -8378,7 +8382,7 @@ class EdgeDeploymentPlan(Base):
8378
8382
  list_method="list_stage_devices",
8379
8383
  summaries_key="DeviceDeploymentSummaries",
8380
8384
  summary_name="DeviceDeploymentSummary",
8381
- resource_cls=DeviceDeploymentSummary,
8385
+ resource_cls=shapes.DeviceDeploymentSummary,
8382
8386
  list_method_kwargs=operation_input_args,
8383
8387
  )
8384
8388
 
@@ -8412,7 +8416,7 @@ class EdgePackagingJob(Base):
8412
8416
  model_name: Optional[str] = Unassigned()
8413
8417
  model_version: Optional[str] = Unassigned()
8414
8418
  role_arn: Optional[str] = Unassigned()
8415
- output_config: Optional[EdgeOutputConfig] = Unassigned()
8419
+ output_config: Optional[shapes.EdgeOutputConfig] = Unassigned()
8416
8420
  resource_key: Optional[str] = Unassigned()
8417
8421
  edge_packaging_job_status: Optional[str] = Unassigned()
8418
8422
  edge_packaging_job_status_message: Optional[str] = Unassigned()
@@ -8420,7 +8424,7 @@ class EdgePackagingJob(Base):
8420
8424
  last_modified_time: Optional[datetime.datetime] = Unassigned()
8421
8425
  model_artifact: Optional[str] = Unassigned()
8422
8426
  model_signature: Optional[str] = Unassigned()
8423
- preset_deployment_output: Optional[EdgePresetDeploymentOutput] = Unassigned()
8427
+ preset_deployment_output: Optional[shapes.EdgePresetDeploymentOutput] = Unassigned()
8424
8428
 
8425
8429
  def get_name(self) -> str:
8426
8430
  attributes = vars(self)
@@ -8467,9 +8471,9 @@ class EdgePackagingJob(Base):
8467
8471
  model_name: Union[str, object],
8468
8472
  model_version: str,
8469
8473
  role_arn: str,
8470
- output_config: EdgeOutputConfig,
8474
+ output_config: shapes.EdgeOutputConfig,
8471
8475
  resource_key: Optional[str] = Unassigned(),
8472
- tags: Optional[List[Tag]] = Unassigned(),
8476
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
8473
8477
  session: Optional[Session] = None,
8474
8478
  region: Optional[str] = None,
8475
8479
  ) -> Optional["EdgePackagingJob"]:
@@ -8823,17 +8827,17 @@ class Endpoint(Base):
8823
8827
  endpoint_name: str
8824
8828
  endpoint_arn: Optional[str] = Unassigned()
8825
8829
  endpoint_config_name: Optional[str] = Unassigned()
8826
- production_variants: Optional[List[ProductionVariantSummary]] = Unassigned()
8827
- data_capture_config: Optional[DataCaptureConfigSummary] = Unassigned()
8830
+ production_variants: Optional[List[shapes.ProductionVariantSummary]] = Unassigned()
8831
+ data_capture_config: Optional[shapes.DataCaptureConfigSummary] = Unassigned()
8828
8832
  endpoint_status: Optional[str] = Unassigned()
8829
8833
  failure_reason: Optional[str] = Unassigned()
8830
8834
  creation_time: Optional[datetime.datetime] = Unassigned()
8831
8835
  last_modified_time: Optional[datetime.datetime] = Unassigned()
8832
- last_deployment_config: Optional[DeploymentConfig] = Unassigned()
8833
- async_inference_config: Optional[AsyncInferenceConfig] = Unassigned()
8834
- pending_deployment_summary: Optional[PendingDeploymentSummary] = Unassigned()
8835
- explainer_config: Optional[ExplainerConfig] = Unassigned()
8836
- shadow_production_variants: Optional[List[ProductionVariantSummary]] = Unassigned()
8836
+ last_deployment_config: Optional[shapes.DeploymentConfig] = Unassigned()
8837
+ async_inference_config: Optional[shapes.AsyncInferenceConfig] = Unassigned()
8838
+ pending_deployment_summary: Optional[shapes.PendingDeploymentSummary] = Unassigned()
8839
+ explainer_config: Optional[shapes.ExplainerConfig] = Unassigned()
8840
+ shadow_production_variants: Optional[List[shapes.ProductionVariantSummary]] = Unassigned()
8837
8841
 
8838
8842
  def get_name(self) -> str:
8839
8843
  attributes = vars(self)
@@ -8883,8 +8887,8 @@ class Endpoint(Base):
8883
8887
  cls,
8884
8888
  endpoint_name: str,
8885
8889
  endpoint_config_name: Union[str, object],
8886
- deployment_config: Optional[DeploymentConfig] = Unassigned(),
8887
- tags: Optional[List[Tag]] = Unassigned(),
8890
+ deployment_config: Optional[shapes.DeploymentConfig] = Unassigned(),
8891
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
8888
8892
  session: Optional[Session] = None,
8889
8893
  region: Optional[str] = None,
8890
8894
  ) -> Optional["Endpoint"]:
@@ -9036,8 +9040,8 @@ class Endpoint(Base):
9036
9040
  def update(
9037
9041
  self,
9038
9042
  retain_all_variant_properties: Optional[bool] = Unassigned(),
9039
- exclude_retained_variant_properties: Optional[List[VariantProperty]] = Unassigned(),
9040
- deployment_config: Optional[DeploymentConfig] = Unassigned(),
9043
+ exclude_retained_variant_properties: Optional[List[shapes.VariantProperty]] = Unassigned(),
9044
+ deployment_config: Optional[shapes.DeploymentConfig] = Unassigned(),
9041
9045
  retain_deployment_config: Optional[bool] = Unassigned(),
9042
9046
  ) -> Optional["Endpoint"]:
9043
9047
  """
@@ -9325,7 +9329,7 @@ class Endpoint(Base):
9325
9329
  @Base.add_validate_call
9326
9330
  def update_weights_and_capacities(
9327
9331
  self,
9328
- desired_weights_and_capacities: List[DesiredWeightAndCapacity],
9332
+ desired_weights_and_capacities: List[shapes.DesiredWeightAndCapacity],
9329
9333
  session: Optional[Session] = None,
9330
9334
  region: Optional[str] = None,
9331
9335
  ) -> None:
@@ -9382,7 +9386,7 @@ class Endpoint(Base):
9382
9386
  session_id: Optional[str] = Unassigned(),
9383
9387
  session: Optional[Session] = None,
9384
9388
  region: Optional[str] = None,
9385
- ) -> Optional[InvokeEndpointOutput]:
9389
+ ) -> Optional[shapes.InvokeEndpointOutput]:
9386
9390
  """
9387
9391
  After you deploy a model into production using Amazon SageMaker hosting services, your client applications use this API to get inferences from the model hosted at the specified endpoint.
9388
9392
 
@@ -9402,7 +9406,7 @@ class Endpoint(Base):
9402
9406
  region: Region name.
9403
9407
 
9404
9408
  Returns:
9405
- InvokeEndpointOutput
9409
+ shapes.InvokeEndpointOutput
9406
9410
 
9407
9411
  Raises:
9408
9412
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -9449,7 +9453,7 @@ class Endpoint(Base):
9449
9453
  logger.debug(f"Response: {response}")
9450
9454
 
9451
9455
  transformed_response = transform(response, "InvokeEndpointOutput")
9452
- return InvokeEndpointOutput(**transformed_response)
9456
+ return shapes.InvokeEndpointOutput(**transformed_response)
9453
9457
 
9454
9458
  @Base.add_validate_call
9455
9459
  def invoke_async(
@@ -9463,7 +9467,7 @@ class Endpoint(Base):
9463
9467
  invocation_timeout_seconds: Optional[int] = Unassigned(),
9464
9468
  session: Optional[Session] = None,
9465
9469
  region: Optional[str] = None,
9466
- ) -> Optional[InvokeEndpointAsyncOutput]:
9470
+ ) -> Optional[shapes.InvokeEndpointAsyncOutput]:
9467
9471
  """
9468
9472
  After you deploy a model into production using Amazon SageMaker hosting services, your client applications use this API to get inferences from the model hosted at the specified endpoint in an asynchronous manner.
9469
9473
 
@@ -9479,7 +9483,7 @@ class Endpoint(Base):
9479
9483
  region: Region name.
9480
9484
 
9481
9485
  Returns:
9482
- InvokeEndpointAsyncOutput
9486
+ shapes.InvokeEndpointAsyncOutput
9483
9487
 
9484
9488
  Raises:
9485
9489
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -9519,7 +9523,7 @@ class Endpoint(Base):
9519
9523
  logger.debug(f"Response: {response}")
9520
9524
 
9521
9525
  transformed_response = transform(response, "InvokeEndpointAsyncOutput")
9522
- return InvokeEndpointAsyncOutput(**transformed_response)
9526
+ return shapes.InvokeEndpointAsyncOutput(**transformed_response)
9523
9527
 
9524
9528
  @Base.add_validate_call
9525
9529
  def invoke_with_response_stream(
@@ -9535,7 +9539,7 @@ class Endpoint(Base):
9535
9539
  session_id: Optional[str] = Unassigned(),
9536
9540
  session: Optional[Session] = None,
9537
9541
  region: Optional[str] = None,
9538
- ) -> Optional[InvokeEndpointWithResponseStreamOutput]:
9542
+ ) -> Optional[shapes.InvokeEndpointWithResponseStreamOutput]:
9539
9543
  """
9540
9544
  Invokes a model at the specified endpoint to return the inference response as a stream.
9541
9545
 
@@ -9553,7 +9557,7 @@ class Endpoint(Base):
9553
9557
  region: Region name.
9554
9558
 
9555
9559
  Returns:
9556
- InvokeEndpointWithResponseStreamOutput
9560
+ shapes.InvokeEndpointWithResponseStreamOutput
9557
9561
 
9558
9562
  Raises:
9559
9563
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -9598,7 +9602,7 @@ class Endpoint(Base):
9598
9602
  logger.debug(f"Response: {response}")
9599
9603
 
9600
9604
  transformed_response = transform(response, "InvokeEndpointWithResponseStreamOutput")
9601
- return InvokeEndpointWithResponseStreamOutput(**transformed_response)
9605
+ return shapes.InvokeEndpointWithResponseStreamOutput(**transformed_response)
9602
9606
 
9603
9607
 
9604
9608
  class EndpointConfig(Base):
@@ -9623,15 +9627,15 @@ class EndpointConfig(Base):
9623
9627
 
9624
9628
  endpoint_config_name: str
9625
9629
  endpoint_config_arn: Optional[str] = Unassigned()
9626
- production_variants: Optional[List[ProductionVariant]] = Unassigned()
9627
- data_capture_config: Optional[DataCaptureConfig] = Unassigned()
9630
+ production_variants: Optional[List[shapes.ProductionVariant]] = Unassigned()
9631
+ data_capture_config: Optional[shapes.DataCaptureConfig] = Unassigned()
9628
9632
  kms_key_id: Optional[str] = Unassigned()
9629
9633
  creation_time: Optional[datetime.datetime] = Unassigned()
9630
- async_inference_config: Optional[AsyncInferenceConfig] = Unassigned()
9631
- explainer_config: Optional[ExplainerConfig] = Unassigned()
9632
- shadow_production_variants: Optional[List[ProductionVariant]] = Unassigned()
9634
+ async_inference_config: Optional[shapes.AsyncInferenceConfig] = Unassigned()
9635
+ explainer_config: Optional[shapes.ExplainerConfig] = Unassigned()
9636
+ shadow_production_variants: Optional[List[shapes.ProductionVariant]] = Unassigned()
9633
9637
  execution_role_arn: Optional[str] = Unassigned()
9634
- vpc_config: Optional[VpcConfig] = Unassigned()
9638
+ vpc_config: Optional[shapes.VpcConfig] = Unassigned()
9635
9639
  enable_network_isolation: Optional[bool] = Unassigned()
9636
9640
 
9637
9641
  def get_name(self) -> str:
@@ -9687,15 +9691,15 @@ class EndpointConfig(Base):
9687
9691
  def create(
9688
9692
  cls,
9689
9693
  endpoint_config_name: str,
9690
- production_variants: List[ProductionVariant],
9691
- data_capture_config: Optional[DataCaptureConfig] = Unassigned(),
9692
- tags: Optional[List[Tag]] = Unassigned(),
9694
+ production_variants: List[shapes.ProductionVariant],
9695
+ data_capture_config: Optional[shapes.DataCaptureConfig] = Unassigned(),
9696
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
9693
9697
  kms_key_id: Optional[str] = Unassigned(),
9694
- async_inference_config: Optional[AsyncInferenceConfig] = Unassigned(),
9695
- explainer_config: Optional[ExplainerConfig] = Unassigned(),
9696
- shadow_production_variants: Optional[List[ProductionVariant]] = Unassigned(),
9698
+ async_inference_config: Optional[shapes.AsyncInferenceConfig] = Unassigned(),
9699
+ explainer_config: Optional[shapes.ExplainerConfig] = Unassigned(),
9700
+ shadow_production_variants: Optional[List[shapes.ProductionVariant]] = Unassigned(),
9697
9701
  execution_role_arn: Optional[str] = Unassigned(),
9698
- vpc_config: Optional[VpcConfig] = Unassigned(),
9702
+ vpc_config: Optional[shapes.VpcConfig] = Unassigned(),
9699
9703
  enable_network_isolation: Optional[bool] = Unassigned(),
9700
9704
  session: Optional[Session] = None,
9701
9705
  region: Optional[str] = None,
@@ -9976,12 +9980,12 @@ class Experiment(Base):
9976
9980
  experiment_name: str
9977
9981
  experiment_arn: Optional[str] = Unassigned()
9978
9982
  display_name: Optional[str] = Unassigned()
9979
- source: Optional[ExperimentSource] = Unassigned()
9983
+ source: Optional[shapes.ExperimentSource] = Unassigned()
9980
9984
  description: Optional[str] = Unassigned()
9981
9985
  creation_time: Optional[datetime.datetime] = Unassigned()
9982
- created_by: Optional[UserContext] = Unassigned()
9986
+ created_by: Optional[shapes.UserContext] = Unassigned()
9983
9987
  last_modified_time: Optional[datetime.datetime] = Unassigned()
9984
- last_modified_by: Optional[UserContext] = Unassigned()
9988
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
9985
9989
 
9986
9990
  def get_name(self) -> str:
9987
9991
  attributes = vars(self)
@@ -10006,7 +10010,7 @@ class Experiment(Base):
10006
10010
  experiment_name: str,
10007
10011
  display_name: Optional[str] = Unassigned(),
10008
10012
  description: Optional[str] = Unassigned(),
10009
- tags: Optional[List[Tag]] = Unassigned(),
10013
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
10010
10014
  session: Optional[Session] = None,
10011
10015
  region: Optional[str] = None,
10012
10016
  ) -> Optional["Experiment"]:
@@ -10328,16 +10332,16 @@ class FeatureGroup(Base):
10328
10332
  feature_group_arn: Optional[str] = Unassigned()
10329
10333
  record_identifier_feature_name: Optional[str] = Unassigned()
10330
10334
  event_time_feature_name: Optional[str] = Unassigned()
10331
- feature_definitions: Optional[List[FeatureDefinition]] = Unassigned()
10335
+ feature_definitions: Optional[List[shapes.FeatureDefinition]] = Unassigned()
10332
10336
  creation_time: Optional[datetime.datetime] = Unassigned()
10333
10337
  last_modified_time: Optional[datetime.datetime] = Unassigned()
10334
- online_store_config: Optional[OnlineStoreConfig] = Unassigned()
10335
- offline_store_config: Optional[OfflineStoreConfig] = Unassigned()
10336
- throughput_config: Optional[ThroughputConfigDescription] = Unassigned()
10338
+ online_store_config: Optional[shapes.OnlineStoreConfig] = Unassigned()
10339
+ offline_store_config: Optional[shapes.OfflineStoreConfig] = Unassigned()
10340
+ throughput_config: Optional[shapes.ThroughputConfigDescription] = Unassigned()
10337
10341
  role_arn: Optional[str] = Unassigned()
10338
10342
  feature_group_status: Optional[str] = Unassigned()
10339
- offline_store_status: Optional[OfflineStoreStatus] = Unassigned()
10340
- last_update_status: Optional[LastUpdateStatus] = Unassigned()
10343
+ offline_store_status: Optional[shapes.OfflineStoreStatus] = Unassigned()
10344
+ last_update_status: Optional[shapes.LastUpdateStatus] = Unassigned()
10341
10345
  failure_reason: Optional[str] = Unassigned()
10342
10346
  description: Optional[str] = Unassigned()
10343
10347
  next_token: Optional[str] = Unassigned()
@@ -10390,13 +10394,13 @@ class FeatureGroup(Base):
10390
10394
  feature_group_name: str,
10391
10395
  record_identifier_feature_name: str,
10392
10396
  event_time_feature_name: str,
10393
- feature_definitions: List[FeatureDefinition],
10394
- online_store_config: Optional[OnlineStoreConfig] = Unassigned(),
10395
- offline_store_config: Optional[OfflineStoreConfig] = Unassigned(),
10396
- throughput_config: Optional[ThroughputConfig] = Unassigned(),
10397
+ feature_definitions: List[shapes.FeatureDefinition],
10398
+ online_store_config: Optional[shapes.OnlineStoreConfig] = Unassigned(),
10399
+ offline_store_config: Optional[shapes.OfflineStoreConfig] = Unassigned(),
10400
+ throughput_config: Optional[shapes.ThroughputConfig] = Unassigned(),
10397
10401
  role_arn: Optional[str] = Unassigned(),
10398
10402
  description: Optional[str] = Unassigned(),
10399
- tags: Optional[List[Tag]] = Unassigned(),
10403
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
10400
10404
  session: Optional[Session] = None,
10401
10405
  region: Optional[str] = None,
10402
10406
  ) -> Optional["FeatureGroup"]:
@@ -10566,9 +10570,9 @@ class FeatureGroup(Base):
10566
10570
  @Base.add_validate_call
10567
10571
  def update(
10568
10572
  self,
10569
- feature_additions: Optional[List[FeatureDefinition]] = Unassigned(),
10570
- online_store_config: Optional[OnlineStoreConfigUpdate] = Unassigned(),
10571
- throughput_config: Optional[ThroughputConfigUpdate] = Unassigned(),
10573
+ feature_additions: Optional[List[shapes.FeatureDefinition]] = Unassigned(),
10574
+ online_store_config: Optional[shapes.OnlineStoreConfigUpdate] = Unassigned(),
10575
+ throughput_config: Optional[shapes.ThroughputConfigUpdate] = Unassigned(),
10572
10576
  ) -> Optional["FeatureGroup"]:
10573
10577
  """
10574
10578
  Update a FeatureGroup resource
@@ -10848,7 +10852,7 @@ class FeatureGroup(Base):
10848
10852
  expiration_time_response: Optional[str] = Unassigned(),
10849
10853
  session: Optional[Session] = None,
10850
10854
  region: Optional[str] = None,
10851
- ) -> Optional[GetRecordResponse]:
10855
+ ) -> Optional[shapes.GetRecordResponse]:
10852
10856
  """
10853
10857
  Use for OnlineStore serving from a FeatureStore.
10854
10858
 
@@ -10860,7 +10864,7 @@ class FeatureGroup(Base):
10860
10864
  region: Region name.
10861
10865
 
10862
10866
  Returns:
10863
- GetRecordResponse
10867
+ shapes.GetRecordResponse
10864
10868
 
10865
10869
  Raises:
10866
10870
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -10898,14 +10902,14 @@ class FeatureGroup(Base):
10898
10902
  logger.debug(f"Response: {response}")
10899
10903
 
10900
10904
  transformed_response = transform(response, "GetRecordResponse")
10901
- return GetRecordResponse(**transformed_response)
10905
+ return shapes.GetRecordResponse(**transformed_response)
10902
10906
 
10903
10907
  @Base.add_validate_call
10904
10908
  def put_record(
10905
10909
  self,
10906
- record: List[FeatureValue],
10910
+ record: List[shapes.FeatureValue],
10907
10911
  target_stores: Optional[List[str]] = Unassigned(),
10908
- ttl_duration: Optional[TtlDuration] = Unassigned(),
10912
+ ttl_duration: Optional[shapes.TtlDuration] = Unassigned(),
10909
10913
  session: Optional[Session] = None,
10910
10914
  region: Optional[str] = None,
10911
10915
  ) -> None:
@@ -11012,11 +11016,11 @@ class FeatureGroup(Base):
11012
11016
  @Base.add_validate_call
11013
11017
  def batch_get_record(
11014
11018
  self,
11015
- identifiers: List[BatchGetRecordIdentifier],
11019
+ identifiers: List[shapes.BatchGetRecordIdentifier],
11016
11020
  expiration_time_response: Optional[str] = Unassigned(),
11017
11021
  session: Optional[Session] = None,
11018
11022
  region: Optional[str] = None,
11019
- ) -> Optional[BatchGetRecordResponse]:
11023
+ ) -> Optional[shapes.BatchGetRecordResponse]:
11020
11024
  """
11021
11025
  Retrieves a batch of Records from a FeatureGroup.
11022
11026
 
@@ -11027,7 +11031,7 @@ class FeatureGroup(Base):
11027
11031
  region: Region name.
11028
11032
 
11029
11033
  Returns:
11030
- BatchGetRecordResponse
11034
+ shapes.BatchGetRecordResponse
11031
11035
 
11032
11036
  Raises:
11033
11037
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -11062,7 +11066,7 @@ class FeatureGroup(Base):
11062
11066
  logger.debug(f"Response: {response}")
11063
11067
 
11064
11068
  transformed_response = transform(response, "BatchGetRecordResponse")
11065
- return BatchGetRecordResponse(**transformed_response)
11069
+ return shapes.BatchGetRecordResponse(**transformed_response)
11066
11070
 
11067
11071
 
11068
11072
  class FeatureMetadata(Base):
@@ -11088,7 +11092,7 @@ class FeatureMetadata(Base):
11088
11092
  creation_time: Optional[datetime.datetime] = Unassigned()
11089
11093
  last_modified_time: Optional[datetime.datetime] = Unassigned()
11090
11094
  description: Optional[str] = Unassigned()
11091
- parameters: Optional[List[FeatureParameter]] = Unassigned()
11095
+ parameters: Optional[List[shapes.FeatureParameter]] = Unassigned()
11092
11096
 
11093
11097
  def get_name(self) -> str:
11094
11098
  attributes = vars(self)
@@ -11202,7 +11206,7 @@ class FeatureMetadata(Base):
11202
11206
  def update(
11203
11207
  self,
11204
11208
  description: Optional[str] = Unassigned(),
11205
- parameter_additions: Optional[List[FeatureParameter]] = Unassigned(),
11209
+ parameter_additions: Optional[List[shapes.FeatureParameter]] = Unassigned(),
11206
11210
  parameter_removals: Optional[List[str]] = Unassigned(),
11207
11211
  ) -> Optional["FeatureMetadata"]:
11208
11212
  """
@@ -11273,10 +11277,10 @@ class FlowDefinition(Base):
11273
11277
  flow_definition_arn: Optional[str] = Unassigned()
11274
11278
  flow_definition_status: Optional[str] = Unassigned()
11275
11279
  creation_time: Optional[datetime.datetime] = Unassigned()
11276
- human_loop_request_source: Optional[HumanLoopRequestSource] = Unassigned()
11277
- human_loop_activation_config: Optional[HumanLoopActivationConfig] = Unassigned()
11278
- human_loop_config: Optional[HumanLoopConfig] = Unassigned()
11279
- output_config: Optional[FlowDefinitionOutputConfig] = Unassigned()
11280
+ human_loop_request_source: Optional[shapes.HumanLoopRequestSource] = Unassigned()
11281
+ human_loop_activation_config: Optional[shapes.HumanLoopActivationConfig] = Unassigned()
11282
+ human_loop_config: Optional[shapes.HumanLoopConfig] = Unassigned()
11283
+ output_config: Optional[shapes.FlowDefinitionOutputConfig] = Unassigned()
11280
11284
  role_arn: Optional[str] = Unassigned()
11281
11285
  failure_reason: Optional[str] = Unassigned()
11282
11286
 
@@ -11321,12 +11325,12 @@ class FlowDefinition(Base):
11321
11325
  def create(
11322
11326
  cls,
11323
11327
  flow_definition_name: str,
11324
- output_config: FlowDefinitionOutputConfig,
11328
+ output_config: shapes.FlowDefinitionOutputConfig,
11325
11329
  role_arn: str,
11326
- human_loop_request_source: Optional[HumanLoopRequestSource] = Unassigned(),
11327
- human_loop_activation_config: Optional[HumanLoopActivationConfig] = Unassigned(),
11328
- human_loop_config: Optional[HumanLoopConfig] = Unassigned(),
11329
- tags: Optional[List[Tag]] = Unassigned(),
11330
+ human_loop_request_source: Optional[shapes.HumanLoopRequestSource] = Unassigned(),
11331
+ human_loop_activation_config: Optional[shapes.HumanLoopActivationConfig] = Unassigned(),
11332
+ human_loop_config: Optional[shapes.HumanLoopConfig] = Unassigned(),
11333
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
11330
11334
  session: Optional[Session] = None,
11331
11335
  region: Optional[str] = None,
11332
11336
  ) -> Optional["FlowDefinition"]:
@@ -11721,7 +11725,7 @@ class Hub(Base):
11721
11725
  hub_display_name: Optional[str] = Unassigned()
11722
11726
  hub_description: Optional[str] = Unassigned()
11723
11727
  hub_search_keywords: Optional[List[str]] = Unassigned()
11724
- s3_storage_config: Optional[HubS3StorageConfig] = Unassigned()
11728
+ s3_storage_config: Optional[shapes.HubS3StorageConfig] = Unassigned()
11725
11729
  hub_status: Optional[str] = Unassigned()
11726
11730
  failure_reason: Optional[str] = Unassigned()
11727
11731
  creation_time: Optional[datetime.datetime] = Unassigned()
@@ -11767,8 +11771,8 @@ class Hub(Base):
11767
11771
  hub_description: str,
11768
11772
  hub_display_name: Optional[str] = Unassigned(),
11769
11773
  hub_search_keywords: Optional[List[str]] = Unassigned(),
11770
- s3_storage_config: Optional[HubS3StorageConfig] = Unassigned(),
11771
- tags: Optional[List[Tag]] = Unassigned(),
11774
+ s3_storage_config: Optional[shapes.HubS3StorageConfig] = Unassigned(),
11775
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
11772
11776
  session: Optional[Session] = None,
11773
11777
  region: Optional[str] = None,
11774
11778
  ) -> Optional["Hub"]:
@@ -12244,7 +12248,7 @@ class HubContent(Base):
12244
12248
  reference_min_version: Optional[str] = Unassigned()
12245
12249
  support_status: Optional[str] = Unassigned()
12246
12250
  hub_content_search_keywords: Optional[List[str]] = Unassigned()
12247
- hub_content_dependencies: Optional[List[HubContentDependency]] = Unassigned()
12251
+ hub_content_dependencies: Optional[List[shapes.HubContentDependency]] = Unassigned()
12248
12252
  hub_content_status: Optional[str] = Unassigned()
12249
12253
  failure_reason: Optional[str] = Unassigned()
12250
12254
  creation_time: Optional[datetime.datetime] = Unassigned()
@@ -12527,7 +12531,7 @@ class HubContent(Base):
12527
12531
  hub_content_markdown: Optional[str] = Unassigned(),
12528
12532
  support_status: Optional[str] = Unassigned(),
12529
12533
  hub_content_search_keywords: Optional[List[str]] = Unassigned(),
12530
- tags: Optional[List[Tag]] = Unassigned(),
12534
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
12531
12535
  session: Optional[Session] = None,
12532
12536
  region: Optional[str] = None,
12533
12537
  ) -> Optional["HubContent"]:
@@ -12698,7 +12702,7 @@ class HubContentReference(Base):
12698
12702
  hub_content_arn: str
12699
12703
  hub_content_name: Optional[Union[str, object]] = Unassigned()
12700
12704
  min_version: Optional[str] = Unassigned()
12701
- tags: Optional[List[Tag]] = Unassigned()
12705
+ tags: Optional[List[shapes.Tag]] = Unassigned()
12702
12706
 
12703
12707
  def get_name(self) -> str:
12704
12708
  attributes = vars(self)
@@ -12724,7 +12728,7 @@ class HubContentReference(Base):
12724
12728
  sage_maker_public_hub_content_arn: str,
12725
12729
  hub_content_name: Optional[Union[str, object]] = Unassigned(),
12726
12730
  min_version: Optional[str] = Unassigned(),
12727
- tags: Optional[List[Tag]] = Unassigned(),
12731
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
12728
12732
  ) -> Optional["HubContentReference"]:
12729
12733
  """
12730
12734
  Create a HubContentReference resource
@@ -12885,7 +12889,7 @@ class HumanTaskUi(Base):
12885
12889
  human_task_ui_arn: Optional[str] = Unassigned()
12886
12890
  human_task_ui_status: Optional[str] = Unassigned()
12887
12891
  creation_time: Optional[datetime.datetime] = Unassigned()
12888
- ui_template: Optional[UiTemplateInfo] = Unassigned()
12892
+ ui_template: Optional[shapes.UiTemplateInfo] = Unassigned()
12889
12893
 
12890
12894
  def get_name(self) -> str:
12891
12895
  attributes = vars(self)
@@ -12908,8 +12912,8 @@ class HumanTaskUi(Base):
12908
12912
  def create(
12909
12913
  cls,
12910
12914
  human_task_ui_name: str,
12911
- ui_template: UiTemplate,
12912
- tags: Optional[List[Tag]] = Unassigned(),
12915
+ ui_template: shapes.UiTemplate,
12916
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
12913
12917
  session: Optional[Session] = None,
12914
12918
  region: Optional[str] = None,
12915
12919
  ) -> Optional["HumanTaskUi"]:
@@ -13293,22 +13297,26 @@ class HyperParameterTuningJob(Base):
13293
13297
 
13294
13298
  hyper_parameter_tuning_job_name: str
13295
13299
  hyper_parameter_tuning_job_arn: Optional[str] = Unassigned()
13296
- hyper_parameter_tuning_job_config: Optional[HyperParameterTuningJobConfig] = Unassigned()
13297
- training_job_definition: Optional[HyperParameterTrainingJobDefinition] = Unassigned()
13298
- training_job_definitions: Optional[List[HyperParameterTrainingJobDefinition]] = Unassigned()
13300
+ hyper_parameter_tuning_job_config: Optional[shapes.HyperParameterTuningJobConfig] = Unassigned()
13301
+ training_job_definition: Optional[shapes.HyperParameterTrainingJobDefinition] = Unassigned()
13302
+ training_job_definitions: Optional[List[shapes.HyperParameterTrainingJobDefinition]] = (
13303
+ Unassigned()
13304
+ )
13299
13305
  hyper_parameter_tuning_job_status: Optional[str] = Unassigned()
13300
13306
  creation_time: Optional[datetime.datetime] = Unassigned()
13301
13307
  hyper_parameter_tuning_end_time: Optional[datetime.datetime] = Unassigned()
13302
13308
  last_modified_time: Optional[datetime.datetime] = Unassigned()
13303
- training_job_status_counters: Optional[TrainingJobStatusCounters] = Unassigned()
13304
- objective_status_counters: Optional[ObjectiveStatusCounters] = Unassigned()
13305
- best_training_job: Optional[HyperParameterTrainingJobSummary] = Unassigned()
13306
- overall_best_training_job: Optional[HyperParameterTrainingJobSummary] = Unassigned()
13307
- warm_start_config: Optional[HyperParameterTuningJobWarmStartConfig] = Unassigned()
13308
- autotune: Optional[Autotune] = Unassigned()
13309
+ training_job_status_counters: Optional[shapes.TrainingJobStatusCounters] = Unassigned()
13310
+ objective_status_counters: Optional[shapes.ObjectiveStatusCounters] = Unassigned()
13311
+ best_training_job: Optional[shapes.HyperParameterTrainingJobSummary] = Unassigned()
13312
+ overall_best_training_job: Optional[shapes.HyperParameterTrainingJobSummary] = Unassigned()
13313
+ warm_start_config: Optional[shapes.HyperParameterTuningJobWarmStartConfig] = Unassigned()
13314
+ autotune: Optional[shapes.Autotune] = Unassigned()
13309
13315
  failure_reason: Optional[str] = Unassigned()
13310
- tuning_job_completion_details: Optional[HyperParameterTuningJobCompletionDetails] = Unassigned()
13311
- consumed_resources: Optional[HyperParameterTuningJobConsumedResources] = Unassigned()
13316
+ tuning_job_completion_details: Optional[shapes.HyperParameterTuningJobCompletionDetails] = (
13317
+ Unassigned()
13318
+ )
13319
+ consumed_resources: Optional[shapes.HyperParameterTuningJobConsumedResources] = Unassigned()
13312
13320
 
13313
13321
  def get_name(self) -> str:
13314
13322
  attributes = vars(self)
@@ -13362,14 +13370,16 @@ class HyperParameterTuningJob(Base):
13362
13370
  def create(
13363
13371
  cls,
13364
13372
  hyper_parameter_tuning_job_name: str,
13365
- hyper_parameter_tuning_job_config: HyperParameterTuningJobConfig,
13366
- training_job_definition: Optional[HyperParameterTrainingJobDefinition] = Unassigned(),
13373
+ hyper_parameter_tuning_job_config: shapes.HyperParameterTuningJobConfig,
13374
+ training_job_definition: Optional[
13375
+ shapes.HyperParameterTrainingJobDefinition
13376
+ ] = Unassigned(),
13367
13377
  training_job_definitions: Optional[
13368
- List[HyperParameterTrainingJobDefinition]
13378
+ List[shapes.HyperParameterTrainingJobDefinition]
13369
13379
  ] = Unassigned(),
13370
- warm_start_config: Optional[HyperParameterTuningJobWarmStartConfig] = Unassigned(),
13371
- tags: Optional[List[Tag]] = Unassigned(),
13372
- autotune: Optional[Autotune] = Unassigned(),
13380
+ warm_start_config: Optional[shapes.HyperParameterTuningJobWarmStartConfig] = Unassigned(),
13381
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
13382
+ autotune: Optional[shapes.Autotune] = Unassigned(),
13373
13383
  session: Optional[Session] = None,
13374
13384
  region: Optional[str] = None,
13375
13385
  ) -> Optional["HyperParameterTuningJob"]:
@@ -13799,7 +13809,7 @@ class HyperParameterTuningJob(Base):
13799
13809
  sort_order: Optional[str] = Unassigned(),
13800
13810
  session: Optional[Session] = None,
13801
13811
  region: Optional[str] = None,
13802
- ) -> ResourceIterator[HyperParameterTrainingJobSummary]:
13812
+ ) -> ResourceIterator[shapes.HyperParameterTrainingJobSummary]:
13803
13813
  """
13804
13814
  Gets a list of TrainingJobSummary objects that describe the training jobs that a hyperparameter tuning job launched.
13805
13815
 
@@ -13847,7 +13857,7 @@ class HyperParameterTuningJob(Base):
13847
13857
  list_method="list_training_jobs_for_hyper_parameter_tuning_job",
13848
13858
  summaries_key="TrainingJobSummaries",
13849
13859
  summary_name="HyperParameterTrainingJobSummary",
13850
- resource_cls=HyperParameterTrainingJobSummary,
13860
+ resource_cls=shapes.HyperParameterTrainingJobSummary,
13851
13861
  list_method_kwargs=operation_input_args,
13852
13862
  )
13853
13863
 
@@ -13917,7 +13927,7 @@ class Image(Base):
13917
13927
  role_arn: str,
13918
13928
  description: Optional[str] = Unassigned(),
13919
13929
  display_name: Optional[str] = Unassigned(),
13920
- tags: Optional[List[Tag]] = Unassigned(),
13930
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
13921
13931
  session: Optional[Session] = None,
13922
13932
  region: Optional[str] = None,
13923
13933
  ) -> Optional["Image"]:
@@ -14928,12 +14938,12 @@ class InferenceComponent(Base):
14928
14938
  endpoint_arn: Optional[str] = Unassigned()
14929
14939
  variant_name: Optional[str] = Unassigned()
14930
14940
  failure_reason: Optional[str] = Unassigned()
14931
- specification: Optional[InferenceComponentSpecificationSummary] = Unassigned()
14932
- runtime_config: Optional[InferenceComponentRuntimeConfigSummary] = Unassigned()
14941
+ specification: Optional[shapes.InferenceComponentSpecificationSummary] = Unassigned()
14942
+ runtime_config: Optional[shapes.InferenceComponentRuntimeConfigSummary] = Unassigned()
14933
14943
  creation_time: Optional[datetime.datetime] = Unassigned()
14934
14944
  last_modified_time: Optional[datetime.datetime] = Unassigned()
14935
14945
  inference_component_status: Optional[str] = Unassigned()
14936
- last_deployment_config: Optional[InferenceComponentDeploymentConfig] = Unassigned()
14946
+ last_deployment_config: Optional[shapes.InferenceComponentDeploymentConfig] = Unassigned()
14937
14947
 
14938
14948
  def get_name(self) -> str:
14939
14949
  attributes = vars(self)
@@ -14957,10 +14967,10 @@ class InferenceComponent(Base):
14957
14967
  cls,
14958
14968
  inference_component_name: str,
14959
14969
  endpoint_name: Union[str, object],
14960
- specification: InferenceComponentSpecification,
14970
+ specification: shapes.InferenceComponentSpecification,
14961
14971
  variant_name: Optional[str] = Unassigned(),
14962
- runtime_config: Optional[InferenceComponentRuntimeConfig] = Unassigned(),
14963
- tags: Optional[List[Tag]] = Unassigned(),
14972
+ runtime_config: Optional[shapes.InferenceComponentRuntimeConfig] = Unassigned(),
14973
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
14964
14974
  session: Optional[Session] = None,
14965
14975
  region: Optional[str] = None,
14966
14976
  ) -> Optional["InferenceComponent"]:
@@ -15116,9 +15126,9 @@ class InferenceComponent(Base):
15116
15126
  @Base.add_validate_call
15117
15127
  def update(
15118
15128
  self,
15119
- specification: Optional[InferenceComponentSpecification] = Unassigned(),
15120
- runtime_config: Optional[InferenceComponentRuntimeConfig] = Unassigned(),
15121
- deployment_config: Optional[InferenceComponentDeploymentConfig] = Unassigned(),
15129
+ specification: Optional[shapes.InferenceComponentSpecification] = Unassigned(),
15130
+ runtime_config: Optional[shapes.InferenceComponentRuntimeConfig] = Unassigned(),
15131
+ deployment_config: Optional[shapes.InferenceComponentDeploymentConfig] = Unassigned(),
15122
15132
  ) -> Optional["InferenceComponent"]:
15123
15133
  """
15124
15134
  Update a InferenceComponent resource
@@ -15404,7 +15414,7 @@ class InferenceComponent(Base):
15404
15414
  @Base.add_validate_call
15405
15415
  def update_runtime_configs(
15406
15416
  self,
15407
- desired_runtime_config: InferenceComponentRuntimeConfig,
15417
+ desired_runtime_config: shapes.InferenceComponentRuntimeConfig,
15408
15418
  session: Optional[Session] = None,
15409
15419
  region: Optional[str] = None,
15410
15420
  ) -> None:
@@ -15473,7 +15483,7 @@ class InferenceExperiment(Base):
15473
15483
  name: str
15474
15484
  arn: Optional[str] = Unassigned()
15475
15485
  type: Optional[str] = Unassigned()
15476
- schedule: Optional[InferenceExperimentSchedule] = Unassigned()
15486
+ schedule: Optional[shapes.InferenceExperimentSchedule] = Unassigned()
15477
15487
  status: Optional[str] = Unassigned()
15478
15488
  status_reason: Optional[str] = Unassigned()
15479
15489
  description: Optional[str] = Unassigned()
@@ -15481,10 +15491,10 @@ class InferenceExperiment(Base):
15481
15491
  completion_time: Optional[datetime.datetime] = Unassigned()
15482
15492
  last_modified_time: Optional[datetime.datetime] = Unassigned()
15483
15493
  role_arn: Optional[str] = Unassigned()
15484
- endpoint_metadata: Optional[EndpointMetadata] = Unassigned()
15485
- model_variants: Optional[List[ModelVariantConfigSummary]] = Unassigned()
15486
- data_storage_config: Optional[InferenceExperimentDataStorageConfig] = Unassigned()
15487
- shadow_mode_config: Optional[ShadowModeConfig] = Unassigned()
15494
+ endpoint_metadata: Optional[shapes.EndpointMetadata] = Unassigned()
15495
+ model_variants: Optional[List[shapes.ModelVariantConfigSummary]] = Unassigned()
15496
+ data_storage_config: Optional[shapes.InferenceExperimentDataStorageConfig] = Unassigned()
15497
+ shadow_mode_config: Optional[shapes.ShadowModeConfig] = Unassigned()
15488
15498
  kms_key: Optional[str] = Unassigned()
15489
15499
 
15490
15500
  def get_name(self) -> str:
@@ -15529,13 +15539,13 @@ class InferenceExperiment(Base):
15529
15539
  type: str,
15530
15540
  role_arn: str,
15531
15541
  endpoint_name: Union[str, object],
15532
- model_variants: List[ModelVariantConfig],
15533
- shadow_mode_config: ShadowModeConfig,
15534
- schedule: Optional[InferenceExperimentSchedule] = Unassigned(),
15542
+ model_variants: List[shapes.ModelVariantConfig],
15543
+ shadow_mode_config: shapes.ShadowModeConfig,
15544
+ schedule: Optional[shapes.InferenceExperimentSchedule] = Unassigned(),
15535
15545
  description: Optional[str] = Unassigned(),
15536
- data_storage_config: Optional[InferenceExperimentDataStorageConfig] = Unassigned(),
15546
+ data_storage_config: Optional[shapes.InferenceExperimentDataStorageConfig] = Unassigned(),
15537
15547
  kms_key: Optional[str] = Unassigned(),
15538
- tags: Optional[List[Tag]] = Unassigned(),
15548
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
15539
15549
  session: Optional[Session] = None,
15540
15550
  region: Optional[str] = None,
15541
15551
  ) -> Optional["InferenceExperiment"]:
@@ -15703,11 +15713,11 @@ class InferenceExperiment(Base):
15703
15713
  @Base.add_validate_call
15704
15714
  def update(
15705
15715
  self,
15706
- schedule: Optional[InferenceExperimentSchedule] = Unassigned(),
15716
+ schedule: Optional[shapes.InferenceExperimentSchedule] = Unassigned(),
15707
15717
  description: Optional[str] = Unassigned(),
15708
- model_variants: Optional[List[ModelVariantConfig]] = Unassigned(),
15709
- data_storage_config: Optional[InferenceExperimentDataStorageConfig] = Unassigned(),
15710
- shadow_mode_config: Optional[ShadowModeConfig] = Unassigned(),
15718
+ model_variants: Optional[List[shapes.ModelVariantConfig]] = Unassigned(),
15719
+ data_storage_config: Optional[shapes.InferenceExperimentDataStorageConfig] = Unassigned(),
15720
+ shadow_mode_config: Optional[shapes.ShadowModeConfig] = Unassigned(),
15711
15721
  ) -> Optional["InferenceExperiment"]:
15712
15722
  """
15713
15723
  Update a InferenceExperiment resource
@@ -15997,10 +16007,10 @@ class InferenceRecommendationsJob(Base):
15997
16007
  completion_time: Optional[datetime.datetime] = Unassigned()
15998
16008
  last_modified_time: Optional[datetime.datetime] = Unassigned()
15999
16009
  failure_reason: Optional[str] = Unassigned()
16000
- input_config: Optional[RecommendationJobInputConfig] = Unassigned()
16001
- stopping_conditions: Optional[RecommendationJobStoppingConditions] = Unassigned()
16002
- inference_recommendations: Optional[List[InferenceRecommendation]] = Unassigned()
16003
- endpoint_performances: Optional[List[EndpointPerformance]] = Unassigned()
16010
+ input_config: Optional[shapes.RecommendationJobInputConfig] = Unassigned()
16011
+ stopping_conditions: Optional[shapes.RecommendationJobStoppingConditions] = Unassigned()
16012
+ inference_recommendations: Optional[List[shapes.InferenceRecommendation]] = Unassigned()
16013
+ endpoint_performances: Optional[List[shapes.EndpointPerformance]] = Unassigned()
16004
16014
 
16005
16015
  def get_name(self) -> str:
16006
16016
  attributes = vars(self)
@@ -16048,11 +16058,11 @@ class InferenceRecommendationsJob(Base):
16048
16058
  job_name: str,
16049
16059
  job_type: str,
16050
16060
  role_arn: str,
16051
- input_config: RecommendationJobInputConfig,
16061
+ input_config: shapes.RecommendationJobInputConfig,
16052
16062
  job_description: Optional[str] = Unassigned(),
16053
- stopping_conditions: Optional[RecommendationJobStoppingConditions] = Unassigned(),
16054
- output_config: Optional[RecommendationJobOutputConfig] = Unassigned(),
16055
- tags: Optional[List[Tag]] = Unassigned(),
16063
+ stopping_conditions: Optional[shapes.RecommendationJobStoppingConditions] = Unassigned(),
16064
+ output_config: Optional[shapes.RecommendationJobOutputConfig] = Unassigned(),
16065
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
16056
16066
  session: Optional[Session] = None,
16057
16067
  region: Optional[str] = None,
16058
16068
  ) -> Optional["InferenceRecommendationsJob"]:
@@ -16456,7 +16466,7 @@ class InferenceRecommendationsJob(Base):
16456
16466
  step_type: Optional[str] = Unassigned(),
16457
16467
  session: Optional[Session] = None,
16458
16468
  region: Optional[str] = None,
16459
- ) -> ResourceIterator[InferenceRecommendationsJobStep]:
16469
+ ) -> ResourceIterator[shapes.InferenceRecommendationsJobStep]:
16460
16470
  """
16461
16471
  Returns a list of the subtasks for an Inference Recommender job.
16462
16472
 
@@ -16501,7 +16511,7 @@ class InferenceRecommendationsJob(Base):
16501
16511
  list_method="list_inference_recommendations_job_steps",
16502
16512
  summaries_key="Steps",
16503
16513
  summary_name="InferenceRecommendationsJobStep",
16504
- resource_cls=InferenceRecommendationsJobStep,
16514
+ resource_cls=shapes.InferenceRecommendationsJobStep,
16505
16515
  list_method_kwargs=operation_input_args,
16506
16516
  )
16507
16517
 
@@ -16534,22 +16544,22 @@ class LabelingJob(Base):
16534
16544
 
16535
16545
  labeling_job_name: str
16536
16546
  labeling_job_status: Optional[str] = Unassigned()
16537
- label_counters: Optional[LabelCounters] = Unassigned()
16547
+ label_counters: Optional[shapes.LabelCounters] = Unassigned()
16538
16548
  failure_reason: Optional[str] = Unassigned()
16539
16549
  creation_time: Optional[datetime.datetime] = Unassigned()
16540
16550
  last_modified_time: Optional[datetime.datetime] = Unassigned()
16541
16551
  job_reference_code: Optional[str] = Unassigned()
16542
16552
  labeling_job_arn: Optional[str] = Unassigned()
16543
16553
  label_attribute_name: Optional[str] = Unassigned()
16544
- input_config: Optional[LabelingJobInputConfig] = Unassigned()
16545
- output_config: Optional[LabelingJobOutputConfig] = Unassigned()
16554
+ input_config: Optional[shapes.LabelingJobInputConfig] = Unassigned()
16555
+ output_config: Optional[shapes.LabelingJobOutputConfig] = Unassigned()
16546
16556
  role_arn: Optional[str] = Unassigned()
16547
16557
  label_category_config_s3_uri: Optional[str] = Unassigned()
16548
- stopping_conditions: Optional[LabelingJobStoppingConditions] = Unassigned()
16549
- labeling_job_algorithms_config: Optional[LabelingJobAlgorithmsConfig] = Unassigned()
16550
- human_task_config: Optional[HumanTaskConfig] = Unassigned()
16551
- tags: Optional[List[Tag]] = Unassigned()
16552
- labeling_job_output: Optional[LabelingJobOutput] = Unassigned()
16558
+ stopping_conditions: Optional[shapes.LabelingJobStoppingConditions] = Unassigned()
16559
+ labeling_job_algorithms_config: Optional[shapes.LabelingJobAlgorithmsConfig] = Unassigned()
16560
+ human_task_config: Optional[shapes.HumanTaskConfig] = Unassigned()
16561
+ tags: Optional[List[shapes.Tag]] = Unassigned()
16562
+ labeling_job_output: Optional[shapes.LabelingJobOutput] = Unassigned()
16553
16563
 
16554
16564
  def get_name(self) -> str:
16555
16565
  attributes = vars(self)
@@ -16608,14 +16618,14 @@ class LabelingJob(Base):
16608
16618
  cls,
16609
16619
  labeling_job_name: str,
16610
16620
  label_attribute_name: str,
16611
- input_config: LabelingJobInputConfig,
16612
- output_config: LabelingJobOutputConfig,
16621
+ input_config: shapes.LabelingJobInputConfig,
16622
+ output_config: shapes.LabelingJobOutputConfig,
16613
16623
  role_arn: str,
16614
- human_task_config: HumanTaskConfig,
16624
+ human_task_config: shapes.HumanTaskConfig,
16615
16625
  label_category_config_s3_uri: Optional[str] = Unassigned(),
16616
- stopping_conditions: Optional[LabelingJobStoppingConditions] = Unassigned(),
16617
- labeling_job_algorithms_config: Optional[LabelingJobAlgorithmsConfig] = Unassigned(),
16618
- tags: Optional[List[Tag]] = Unassigned(),
16626
+ stopping_conditions: Optional[shapes.LabelingJobStoppingConditions] = Unassigned(),
16627
+ labeling_job_algorithms_config: Optional[shapes.LabelingJobAlgorithmsConfig] = Unassigned(),
16628
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
16619
16629
  session: Optional[Session] = None,
16620
16630
  region: Optional[str] = None,
16621
16631
  ) -> Optional["LabelingJob"]:
@@ -16964,9 +16974,9 @@ class LineageGroup(Base):
16964
16974
  display_name: Optional[str] = Unassigned()
16965
16975
  description: Optional[str] = Unassigned()
16966
16976
  creation_time: Optional[datetime.datetime] = Unassigned()
16967
- created_by: Optional[UserContext] = Unassigned()
16977
+ created_by: Optional[shapes.UserContext] = Unassigned()
16968
16978
  last_modified_time: Optional[datetime.datetime] = Unassigned()
16969
- last_modified_by: Optional[UserContext] = Unassigned()
16979
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
16970
16980
 
16971
16981
  def get_name(self) -> str:
16972
16982
  attributes = vars(self)
@@ -17140,7 +17150,7 @@ class LineageGroup(Base):
17140
17150
  self,
17141
17151
  session: Optional[Session] = None,
17142
17152
  region: Optional[str] = None,
17143
- ) -> Optional[GetLineageGroupPolicyResponse]:
17153
+ ) -> Optional[shapes.GetLineageGroupPolicyResponse]:
17144
17154
  """
17145
17155
  The resource policy for the lineage group.
17146
17156
 
@@ -17149,7 +17159,7 @@ class LineageGroup(Base):
17149
17159
  region: Region name.
17150
17160
 
17151
17161
  Returns:
17152
- GetLineageGroupPolicyResponse
17162
+ shapes.GetLineageGroupPolicyResponse
17153
17163
 
17154
17164
  Raises:
17155
17165
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -17180,7 +17190,7 @@ class LineageGroup(Base):
17180
17190
  logger.debug(f"Response: {response}")
17181
17191
 
17182
17192
  transformed_response = transform(response, "GetLineageGroupPolicyResponse")
17183
- return GetLineageGroupPolicyResponse(**transformed_response)
17193
+ return shapes.GetLineageGroupPolicyResponse(**transformed_response)
17184
17194
 
17185
17195
 
17186
17196
  class MlflowTrackingServer(Base):
@@ -17218,9 +17228,9 @@ class MlflowTrackingServer(Base):
17218
17228
  weekly_maintenance_window_start: Optional[str] = Unassigned()
17219
17229
  automatic_model_registration: Optional[bool] = Unassigned()
17220
17230
  creation_time: Optional[datetime.datetime] = Unassigned()
17221
- created_by: Optional[UserContext] = Unassigned()
17231
+ created_by: Optional[shapes.UserContext] = Unassigned()
17222
17232
  last_modified_time: Optional[datetime.datetime] = Unassigned()
17223
- last_modified_by: Optional[UserContext] = Unassigned()
17233
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
17224
17234
 
17225
17235
  def get_name(self) -> str:
17226
17236
  attributes = vars(self)
@@ -17263,7 +17273,7 @@ class MlflowTrackingServer(Base):
17263
17273
  mlflow_version: Optional[str] = Unassigned(),
17264
17274
  automatic_model_registration: Optional[bool] = Unassigned(),
17265
17275
  weekly_maintenance_window_start: Optional[str] = Unassigned(),
17266
- tags: Optional[List[Tag]] = Unassigned(),
17276
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
17267
17277
  session: Optional[Session] = None,
17268
17278
  region: Optional[str] = None,
17269
17279
  ) -> Optional["MlflowTrackingServer"]:
@@ -17769,15 +17779,15 @@ class Model(Base):
17769
17779
  """
17770
17780
 
17771
17781
  model_name: str
17772
- primary_container: Optional[ContainerDefinition] = Unassigned()
17773
- containers: Optional[List[ContainerDefinition]] = Unassigned()
17774
- inference_execution_config: Optional[InferenceExecutionConfig] = Unassigned()
17782
+ primary_container: Optional[shapes.ContainerDefinition] = Unassigned()
17783
+ containers: Optional[List[shapes.ContainerDefinition]] = Unassigned()
17784
+ inference_execution_config: Optional[shapes.InferenceExecutionConfig] = Unassigned()
17775
17785
  execution_role_arn: Optional[str] = Unassigned()
17776
- vpc_config: Optional[VpcConfig] = Unassigned()
17786
+ vpc_config: Optional[shapes.VpcConfig] = Unassigned()
17777
17787
  creation_time: Optional[datetime.datetime] = Unassigned()
17778
17788
  model_arn: Optional[str] = Unassigned()
17779
17789
  enable_network_isolation: Optional[bool] = Unassigned()
17780
- deployment_recommendation: Optional[DeploymentRecommendation] = Unassigned()
17790
+ deployment_recommendation: Optional[shapes.DeploymentRecommendation] = Unassigned()
17781
17791
 
17782
17792
  def get_name(self) -> str:
17783
17793
  attributes = vars(self)
@@ -17829,12 +17839,12 @@ class Model(Base):
17829
17839
  def create(
17830
17840
  cls,
17831
17841
  model_name: str,
17832
- primary_container: Optional[ContainerDefinition] = Unassigned(),
17833
- containers: Optional[List[ContainerDefinition]] = Unassigned(),
17834
- inference_execution_config: Optional[InferenceExecutionConfig] = Unassigned(),
17842
+ primary_container: Optional[shapes.ContainerDefinition] = Unassigned(),
17843
+ containers: Optional[List[shapes.ContainerDefinition]] = Unassigned(),
17844
+ inference_execution_config: Optional[shapes.InferenceExecutionConfig] = Unassigned(),
17835
17845
  execution_role_arn: Optional[str] = Unassigned(),
17836
- tags: Optional[List[Tag]] = Unassigned(),
17837
- vpc_config: Optional[VpcConfig] = Unassigned(),
17846
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
17847
+ vpc_config: Optional[shapes.VpcConfig] = Unassigned(),
17838
17848
  enable_network_isolation: Optional[bool] = Unassigned(),
17839
17849
  session: Optional[Session] = None,
17840
17850
  region: Optional[str] = None,
@@ -18091,10 +18101,10 @@ class Model(Base):
18091
18101
  @Base.add_validate_call
18092
18102
  def get_all_metadata(
18093
18103
  self,
18094
- search_expression: Optional[ModelMetadataSearchExpression] = Unassigned(),
18104
+ search_expression: Optional[shapes.ModelMetadataSearchExpression] = Unassigned(),
18095
18105
  session: Optional[Session] = None,
18096
18106
  region: Optional[str] = None,
18097
- ) -> ResourceIterator[ModelMetadataSummary]:
18107
+ ) -> ResourceIterator[shapes.ModelMetadataSummary]:
18098
18108
  """
18099
18109
  Lists the domain, framework, task, and model name of standard machine learning models found in common model zoos.
18100
18110
 
@@ -18136,7 +18146,7 @@ class Model(Base):
18136
18146
  list_method="list_model_metadata",
18137
18147
  summaries_key="ModelMetadataSummaries",
18138
18148
  summary_name="ModelMetadataSummary",
18139
- resource_cls=ModelMetadataSummary,
18149
+ resource_cls=shapes.ModelMetadataSummary,
18140
18150
  list_method_kwargs=operation_input_args,
18141
18151
  )
18142
18152
 
@@ -18163,14 +18173,14 @@ class ModelBiasJobDefinition(Base):
18163
18173
  job_definition_name: str
18164
18174
  job_definition_arn: Optional[str] = Unassigned()
18165
18175
  creation_time: Optional[datetime.datetime] = Unassigned()
18166
- model_bias_baseline_config: Optional[ModelBiasBaselineConfig] = Unassigned()
18167
- model_bias_app_specification: Optional[ModelBiasAppSpecification] = Unassigned()
18168
- model_bias_job_input: Optional[ModelBiasJobInput] = Unassigned()
18169
- model_bias_job_output_config: Optional[MonitoringOutputConfig] = Unassigned()
18170
- job_resources: Optional[MonitoringResources] = Unassigned()
18171
- network_config: Optional[MonitoringNetworkConfig] = Unassigned()
18176
+ model_bias_baseline_config: Optional[shapes.ModelBiasBaselineConfig] = Unassigned()
18177
+ model_bias_app_specification: Optional[shapes.ModelBiasAppSpecification] = Unassigned()
18178
+ model_bias_job_input: Optional[shapes.ModelBiasJobInput] = Unassigned()
18179
+ model_bias_job_output_config: Optional[shapes.MonitoringOutputConfig] = Unassigned()
18180
+ job_resources: Optional[shapes.MonitoringResources] = Unassigned()
18181
+ network_config: Optional[shapes.MonitoringNetworkConfig] = Unassigned()
18172
18182
  role_arn: Optional[str] = Unassigned()
18173
- stopping_condition: Optional[MonitoringStoppingCondition] = Unassigned()
18183
+ stopping_condition: Optional[shapes.MonitoringStoppingCondition] = Unassigned()
18174
18184
 
18175
18185
  def get_name(self) -> str:
18176
18186
  attributes = vars(self)
@@ -18232,15 +18242,15 @@ class ModelBiasJobDefinition(Base):
18232
18242
  def create(
18233
18243
  cls,
18234
18244
  job_definition_name: str,
18235
- model_bias_app_specification: ModelBiasAppSpecification,
18236
- model_bias_job_input: ModelBiasJobInput,
18237
- model_bias_job_output_config: MonitoringOutputConfig,
18238
- job_resources: MonitoringResources,
18245
+ model_bias_app_specification: shapes.ModelBiasAppSpecification,
18246
+ model_bias_job_input: shapes.ModelBiasJobInput,
18247
+ model_bias_job_output_config: shapes.MonitoringOutputConfig,
18248
+ job_resources: shapes.MonitoringResources,
18239
18249
  role_arn: str,
18240
- model_bias_baseline_config: Optional[ModelBiasBaselineConfig] = Unassigned(),
18241
- network_config: Optional[MonitoringNetworkConfig] = Unassigned(),
18242
- stopping_condition: Optional[MonitoringStoppingCondition] = Unassigned(),
18243
- tags: Optional[List[Tag]] = Unassigned(),
18250
+ model_bias_baseline_config: Optional[shapes.ModelBiasBaselineConfig] = Unassigned(),
18251
+ network_config: Optional[shapes.MonitoringNetworkConfig] = Unassigned(),
18252
+ stopping_condition: Optional[shapes.MonitoringStoppingCondition] = Unassigned(),
18253
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
18244
18254
  session: Optional[Session] = None,
18245
18255
  region: Optional[str] = None,
18246
18256
  ) -> Optional["ModelBiasJobDefinition"]:
@@ -18533,11 +18543,11 @@ class ModelCard(Base):
18533
18543
  model_card_version: Optional[int] = Unassigned()
18534
18544
  content: Optional[str] = Unassigned()
18535
18545
  model_card_status: Optional[str] = Unassigned()
18536
- security_config: Optional[ModelCardSecurityConfig] = Unassigned()
18546
+ security_config: Optional[shapes.ModelCardSecurityConfig] = Unassigned()
18537
18547
  creation_time: Optional[datetime.datetime] = Unassigned()
18538
- created_by: Optional[UserContext] = Unassigned()
18548
+ created_by: Optional[shapes.UserContext] = Unassigned()
18539
18549
  last_modified_time: Optional[datetime.datetime] = Unassigned()
18540
- last_modified_by: Optional[UserContext] = Unassigned()
18550
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
18541
18551
  model_card_processing_status: Optional[str] = Unassigned()
18542
18552
 
18543
18553
  def get_name(self) -> str:
@@ -18577,8 +18587,8 @@ class ModelCard(Base):
18577
18587
  model_card_name: str,
18578
18588
  content: str,
18579
18589
  model_card_status: str,
18580
- security_config: Optional[ModelCardSecurityConfig] = Unassigned(),
18581
- tags: Optional[List[Tag]] = Unassigned(),
18590
+ security_config: Optional[shapes.ModelCardSecurityConfig] = Unassigned(),
18591
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
18582
18592
  session: Optional[Session] = None,
18583
18593
  region: Optional[str] = None,
18584
18594
  ) -> Optional["ModelCard"]:
@@ -18945,7 +18955,7 @@ class ModelCard(Base):
18945
18955
  sort_order: Optional[str] = Unassigned(),
18946
18956
  session: Optional[Session] = None,
18947
18957
  region: Optional[str] = None,
18948
- ) -> ResourceIterator[ModelCardVersionSummary]:
18958
+ ) -> ResourceIterator[shapes.ModelCardVersionSummary]:
18949
18959
  """
18950
18960
  List existing versions of an Amazon SageMaker Model Card.
18951
18961
 
@@ -18996,7 +19006,7 @@ class ModelCard(Base):
18996
19006
  list_method="list_model_card_versions",
18997
19007
  summaries_key="ModelCardVersionSummaryList",
18998
19008
  summary_name="ModelCardVersionSummary",
18999
- resource_cls=ModelCardVersionSummary,
19009
+ resource_cls=shapes.ModelCardVersionSummary,
19000
19010
  list_method_kwargs=operation_input_args,
19001
19011
  )
19002
19012
 
@@ -19024,11 +19034,11 @@ class ModelCardExportJob(Base):
19024
19034
  status: Optional[str] = Unassigned()
19025
19035
  model_card_name: Optional[str] = Unassigned()
19026
19036
  model_card_version: Optional[int] = Unassigned()
19027
- output_config: Optional[ModelCardExportOutputConfig] = Unassigned()
19037
+ output_config: Optional[shapes.ModelCardExportOutputConfig] = Unassigned()
19028
19038
  created_at: Optional[datetime.datetime] = Unassigned()
19029
19039
  last_modified_at: Optional[datetime.datetime] = Unassigned()
19030
19040
  failure_reason: Optional[str] = Unassigned()
19031
- export_artifacts: Optional[ModelCardExportArtifacts] = Unassigned()
19041
+ export_artifacts: Optional[shapes.ModelCardExportArtifacts] = Unassigned()
19032
19042
 
19033
19043
  def get_name(self) -> str:
19034
19044
  attributes = vars(self)
@@ -19069,7 +19079,7 @@ class ModelCardExportJob(Base):
19069
19079
  cls,
19070
19080
  model_card_name: Union[str, object],
19071
19081
  model_card_export_job_name: str,
19072
- output_config: ModelCardExportOutputConfig,
19082
+ output_config: shapes.ModelCardExportOutputConfig,
19073
19083
  model_card_version: Optional[int] = Unassigned(),
19074
19084
  session: Optional[Session] = None,
19075
19085
  region: Optional[str] = None,
@@ -19384,16 +19394,18 @@ class ModelExplainabilityJobDefinition(Base):
19384
19394
  job_definition_name: str
19385
19395
  job_definition_arn: Optional[str] = Unassigned()
19386
19396
  creation_time: Optional[datetime.datetime] = Unassigned()
19387
- model_explainability_baseline_config: Optional[ModelExplainabilityBaselineConfig] = Unassigned()
19388
- model_explainability_app_specification: Optional[ModelExplainabilityAppSpecification] = (
19397
+ model_explainability_baseline_config: Optional[shapes.ModelExplainabilityBaselineConfig] = (
19389
19398
  Unassigned()
19390
19399
  )
19391
- model_explainability_job_input: Optional[ModelExplainabilityJobInput] = Unassigned()
19392
- model_explainability_job_output_config: Optional[MonitoringOutputConfig] = Unassigned()
19393
- job_resources: Optional[MonitoringResources] = Unassigned()
19394
- network_config: Optional[MonitoringNetworkConfig] = Unassigned()
19400
+ model_explainability_app_specification: Optional[shapes.ModelExplainabilityAppSpecification] = (
19401
+ Unassigned()
19402
+ )
19403
+ model_explainability_job_input: Optional[shapes.ModelExplainabilityJobInput] = Unassigned()
19404
+ model_explainability_job_output_config: Optional[shapes.MonitoringOutputConfig] = Unassigned()
19405
+ job_resources: Optional[shapes.MonitoringResources] = Unassigned()
19406
+ network_config: Optional[shapes.MonitoringNetworkConfig] = Unassigned()
19395
19407
  role_arn: Optional[str] = Unassigned()
19396
- stopping_condition: Optional[MonitoringStoppingCondition] = Unassigned()
19408
+ stopping_condition: Optional[shapes.MonitoringStoppingCondition] = Unassigned()
19397
19409
 
19398
19410
  def get_name(self) -> str:
19399
19411
  attributes = vars(self)
@@ -19454,17 +19466,17 @@ class ModelExplainabilityJobDefinition(Base):
19454
19466
  def create(
19455
19467
  cls,
19456
19468
  job_definition_name: str,
19457
- model_explainability_app_specification: ModelExplainabilityAppSpecification,
19458
- model_explainability_job_input: ModelExplainabilityJobInput,
19459
- model_explainability_job_output_config: MonitoringOutputConfig,
19460
- job_resources: MonitoringResources,
19469
+ model_explainability_app_specification: shapes.ModelExplainabilityAppSpecification,
19470
+ model_explainability_job_input: shapes.ModelExplainabilityJobInput,
19471
+ model_explainability_job_output_config: shapes.MonitoringOutputConfig,
19472
+ job_resources: shapes.MonitoringResources,
19461
19473
  role_arn: str,
19462
19474
  model_explainability_baseline_config: Optional[
19463
- ModelExplainabilityBaselineConfig
19475
+ shapes.ModelExplainabilityBaselineConfig
19464
19476
  ] = Unassigned(),
19465
- network_config: Optional[MonitoringNetworkConfig] = Unassigned(),
19466
- stopping_condition: Optional[MonitoringStoppingCondition] = Unassigned(),
19467
- tags: Optional[List[Tag]] = Unassigned(),
19477
+ network_config: Optional[shapes.MonitoringNetworkConfig] = Unassigned(),
19478
+ stopping_condition: Optional[shapes.MonitoringStoppingCondition] = Unassigned(),
19479
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
19468
19480
  session: Optional[Session] = None,
19469
19481
  region: Optional[str] = None,
19470
19482
  ) -> Optional["ModelExplainabilityJobDefinition"]:
@@ -19780,32 +19792,32 @@ class ModelPackage(Base):
19780
19792
  model_package_arn: Optional[str] = Unassigned()
19781
19793
  model_package_description: Optional[str] = Unassigned()
19782
19794
  creation_time: Optional[datetime.datetime] = Unassigned()
19783
- inference_specification: Optional[InferenceSpecification] = Unassigned()
19784
- source_algorithm_specification: Optional[SourceAlgorithmSpecification] = Unassigned()
19785
- validation_specification: Optional[ModelPackageValidationSpecification] = Unassigned()
19795
+ inference_specification: Optional[shapes.InferenceSpecification] = Unassigned()
19796
+ source_algorithm_specification: Optional[shapes.SourceAlgorithmSpecification] = Unassigned()
19797
+ validation_specification: Optional[shapes.ModelPackageValidationSpecification] = Unassigned()
19786
19798
  model_package_status: Optional[str] = Unassigned()
19787
- model_package_status_details: Optional[ModelPackageStatusDetails] = Unassigned()
19799
+ model_package_status_details: Optional[shapes.ModelPackageStatusDetails] = Unassigned()
19788
19800
  certify_for_marketplace: Optional[bool] = Unassigned()
19789
19801
  model_approval_status: Optional[str] = Unassigned()
19790
- created_by: Optional[UserContext] = Unassigned()
19791
- metadata_properties: Optional[MetadataProperties] = Unassigned()
19792
- model_metrics: Optional[ModelMetrics] = Unassigned()
19802
+ created_by: Optional[shapes.UserContext] = Unassigned()
19803
+ metadata_properties: Optional[shapes.MetadataProperties] = Unassigned()
19804
+ model_metrics: Optional[shapes.ModelMetrics] = Unassigned()
19793
19805
  last_modified_time: Optional[datetime.datetime] = Unassigned()
19794
- last_modified_by: Optional[UserContext] = Unassigned()
19806
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
19795
19807
  approval_description: Optional[str] = Unassigned()
19796
19808
  domain: Optional[str] = Unassigned()
19797
19809
  task: Optional[str] = Unassigned()
19798
19810
  sample_payload_url: Optional[str] = Unassigned()
19799
19811
  customer_metadata_properties: Optional[Dict[str, str]] = Unassigned()
19800
- drift_check_baselines: Optional[DriftCheckBaselines] = Unassigned()
19812
+ drift_check_baselines: Optional[shapes.DriftCheckBaselines] = Unassigned()
19801
19813
  additional_inference_specifications: Optional[
19802
- List[AdditionalInferenceSpecificationDefinition]
19814
+ List[shapes.AdditionalInferenceSpecificationDefinition]
19803
19815
  ] = Unassigned()
19804
19816
  skip_model_validation: Optional[str] = Unassigned()
19805
19817
  source_uri: Optional[str] = Unassigned()
19806
- security_config: Optional[ModelPackageSecurityConfig] = Unassigned()
19807
- model_card: Optional[ModelPackageModelCard] = Unassigned()
19808
- model_life_cycle: Optional[ModelLifeCycle] = Unassigned()
19818
+ security_config: Optional[shapes.ModelPackageSecurityConfig] = Unassigned()
19819
+ model_card: Optional[shapes.ModelPackageModelCard] = Unassigned()
19820
+ model_life_cycle: Optional[shapes.ModelLifeCycle] = Unassigned()
19809
19821
 
19810
19822
  def get_name(self) -> str:
19811
19823
  attributes = vars(self)
@@ -19882,28 +19894,32 @@ class ModelPackage(Base):
19882
19894
  model_package_name: Optional[str] = Unassigned(),
19883
19895
  model_package_group_name: Optional[Union[str, object]] = Unassigned(),
19884
19896
  model_package_description: Optional[str] = Unassigned(),
19885
- inference_specification: Optional[InferenceSpecification] = Unassigned(),
19886
- validation_specification: Optional[ModelPackageValidationSpecification] = Unassigned(),
19887
- source_algorithm_specification: Optional[SourceAlgorithmSpecification] = Unassigned(),
19897
+ inference_specification: Optional[shapes.InferenceSpecification] = Unassigned(),
19898
+ validation_specification: Optional[
19899
+ shapes.ModelPackageValidationSpecification
19900
+ ] = Unassigned(),
19901
+ source_algorithm_specification: Optional[
19902
+ shapes.SourceAlgorithmSpecification
19903
+ ] = Unassigned(),
19888
19904
  certify_for_marketplace: Optional[bool] = Unassigned(),
19889
- tags: Optional[List[Tag]] = Unassigned(),
19905
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
19890
19906
  model_approval_status: Optional[str] = Unassigned(),
19891
- metadata_properties: Optional[MetadataProperties] = Unassigned(),
19892
- model_metrics: Optional[ModelMetrics] = Unassigned(),
19907
+ metadata_properties: Optional[shapes.MetadataProperties] = Unassigned(),
19908
+ model_metrics: Optional[shapes.ModelMetrics] = Unassigned(),
19893
19909
  client_token: Optional[str] = Unassigned(),
19894
19910
  domain: Optional[str] = Unassigned(),
19895
19911
  task: Optional[str] = Unassigned(),
19896
19912
  sample_payload_url: Optional[str] = Unassigned(),
19897
19913
  customer_metadata_properties: Optional[Dict[str, str]] = Unassigned(),
19898
- drift_check_baselines: Optional[DriftCheckBaselines] = Unassigned(),
19914
+ drift_check_baselines: Optional[shapes.DriftCheckBaselines] = Unassigned(),
19899
19915
  additional_inference_specifications: Optional[
19900
- List[AdditionalInferenceSpecificationDefinition]
19916
+ List[shapes.AdditionalInferenceSpecificationDefinition]
19901
19917
  ] = Unassigned(),
19902
19918
  skip_model_validation: Optional[str] = Unassigned(),
19903
19919
  source_uri: Optional[str] = Unassigned(),
19904
- security_config: Optional[ModelPackageSecurityConfig] = Unassigned(),
19905
- model_card: Optional[ModelPackageModelCard] = Unassigned(),
19906
- model_life_cycle: Optional[ModelLifeCycle] = Unassigned(),
19920
+ security_config: Optional[shapes.ModelPackageSecurityConfig] = Unassigned(),
19921
+ model_card: Optional[shapes.ModelPackageModelCard] = Unassigned(),
19922
+ model_life_cycle: Optional[shapes.ModelLifeCycle] = Unassigned(),
19907
19923
  session: Optional[Session] = None,
19908
19924
  region: Optional[str] = None,
19909
19925
  ) -> Optional["ModelPackage"]:
@@ -20100,12 +20116,12 @@ class ModelPackage(Base):
20100
20116
  customer_metadata_properties: Optional[Dict[str, str]] = Unassigned(),
20101
20117
  customer_metadata_properties_to_remove: Optional[List[str]] = Unassigned(),
20102
20118
  additional_inference_specifications_to_add: Optional[
20103
- List[AdditionalInferenceSpecificationDefinition]
20119
+ List[shapes.AdditionalInferenceSpecificationDefinition]
20104
20120
  ] = Unassigned(),
20105
- inference_specification: Optional[InferenceSpecification] = Unassigned(),
20121
+ inference_specification: Optional[shapes.InferenceSpecification] = Unassigned(),
20106
20122
  source_uri: Optional[str] = Unassigned(),
20107
- model_card: Optional[ModelPackageModelCard] = Unassigned(),
20108
- model_life_cycle: Optional[ModelLifeCycle] = Unassigned(),
20123
+ model_card: Optional[shapes.ModelPackageModelCard] = Unassigned(),
20124
+ model_life_cycle: Optional[shapes.ModelLifeCycle] = Unassigned(),
20109
20125
  client_token: Optional[str] = Unassigned(),
20110
20126
  ) -> Optional["ModelPackage"]:
20111
20127
  """
@@ -20393,7 +20409,7 @@ class ModelPackage(Base):
20393
20409
  model_package_arn_list: List[str],
20394
20410
  session: Optional[Session] = None,
20395
20411
  region: Optional[str] = None,
20396
- ) -> Optional[BatchDescribeModelPackageOutput]:
20412
+ ) -> Optional[shapes.BatchDescribeModelPackageOutput]:
20397
20413
  """
20398
20414
  This action batch describes a list of versioned model packages.
20399
20415
 
@@ -20403,7 +20419,7 @@ class ModelPackage(Base):
20403
20419
  region: Region name.
20404
20420
 
20405
20421
  Returns:
20406
- BatchDescribeModelPackageOutput
20422
+ shapes.BatchDescribeModelPackageOutput
20407
20423
 
20408
20424
  Raises:
20409
20425
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -20433,7 +20449,7 @@ class ModelPackage(Base):
20433
20449
  logger.debug(f"Response: {response}")
20434
20450
 
20435
20451
  transformed_response = transform(response, "BatchDescribeModelPackageOutput")
20436
- return BatchDescribeModelPackageOutput(**transformed_response)
20452
+ return shapes.BatchDescribeModelPackageOutput(**transformed_response)
20437
20453
 
20438
20454
 
20439
20455
  class ModelPackageGroup(Base):
@@ -20454,7 +20470,7 @@ class ModelPackageGroup(Base):
20454
20470
  model_package_group_arn: Optional[str] = Unassigned()
20455
20471
  model_package_group_description: Optional[str] = Unassigned()
20456
20472
  creation_time: Optional[datetime.datetime] = Unassigned()
20457
- created_by: Optional[UserContext] = Unassigned()
20473
+ created_by: Optional[shapes.UserContext] = Unassigned()
20458
20474
  model_package_group_status: Optional[str] = Unassigned()
20459
20475
 
20460
20476
  def get_name(self) -> str:
@@ -20479,7 +20495,7 @@ class ModelPackageGroup(Base):
20479
20495
  cls,
20480
20496
  model_package_group_name: str,
20481
20497
  model_package_group_description: Optional[str] = Unassigned(),
20482
- tags: Optional[List[Tag]] = Unassigned(),
20498
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
20483
20499
  session: Optional[Session] = None,
20484
20500
  region: Optional[str] = None,
20485
20501
  ) -> Optional["ModelPackageGroup"]:
@@ -21003,14 +21019,14 @@ class ModelQualityJobDefinition(Base):
21003
21019
  job_definition_name: str
21004
21020
  job_definition_arn: Optional[str] = Unassigned()
21005
21021
  creation_time: Optional[datetime.datetime] = Unassigned()
21006
- model_quality_baseline_config: Optional[ModelQualityBaselineConfig] = Unassigned()
21007
- model_quality_app_specification: Optional[ModelQualityAppSpecification] = Unassigned()
21008
- model_quality_job_input: Optional[ModelQualityJobInput] = Unassigned()
21009
- model_quality_job_output_config: Optional[MonitoringOutputConfig] = Unassigned()
21010
- job_resources: Optional[MonitoringResources] = Unassigned()
21011
- network_config: Optional[MonitoringNetworkConfig] = Unassigned()
21022
+ model_quality_baseline_config: Optional[shapes.ModelQualityBaselineConfig] = Unassigned()
21023
+ model_quality_app_specification: Optional[shapes.ModelQualityAppSpecification] = Unassigned()
21024
+ model_quality_job_input: Optional[shapes.ModelQualityJobInput] = Unassigned()
21025
+ model_quality_job_output_config: Optional[shapes.MonitoringOutputConfig] = Unassigned()
21026
+ job_resources: Optional[shapes.MonitoringResources] = Unassigned()
21027
+ network_config: Optional[shapes.MonitoringNetworkConfig] = Unassigned()
21012
21028
  role_arn: Optional[str] = Unassigned()
21013
- stopping_condition: Optional[MonitoringStoppingCondition] = Unassigned()
21029
+ stopping_condition: Optional[shapes.MonitoringStoppingCondition] = Unassigned()
21014
21030
 
21015
21031
  def get_name(self) -> str:
21016
21032
  attributes = vars(self)
@@ -21072,15 +21088,15 @@ class ModelQualityJobDefinition(Base):
21072
21088
  def create(
21073
21089
  cls,
21074
21090
  job_definition_name: str,
21075
- model_quality_app_specification: ModelQualityAppSpecification,
21076
- model_quality_job_input: ModelQualityJobInput,
21077
- model_quality_job_output_config: MonitoringOutputConfig,
21078
- job_resources: MonitoringResources,
21091
+ model_quality_app_specification: shapes.ModelQualityAppSpecification,
21092
+ model_quality_job_input: shapes.ModelQualityJobInput,
21093
+ model_quality_job_output_config: shapes.MonitoringOutputConfig,
21094
+ job_resources: shapes.MonitoringResources,
21079
21095
  role_arn: str,
21080
- model_quality_baseline_config: Optional[ModelQualityBaselineConfig] = Unassigned(),
21081
- network_config: Optional[MonitoringNetworkConfig] = Unassigned(),
21082
- stopping_condition: Optional[MonitoringStoppingCondition] = Unassigned(),
21083
- tags: Optional[List[Tag]] = Unassigned(),
21096
+ model_quality_baseline_config: Optional[shapes.ModelQualityBaselineConfig] = Unassigned(),
21097
+ network_config: Optional[shapes.MonitoringNetworkConfig] = Unassigned(),
21098
+ stopping_condition: Optional[shapes.MonitoringStoppingCondition] = Unassigned(),
21099
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
21084
21100
  session: Optional[Session] = None,
21085
21101
  region: Optional[str] = None,
21086
21102
  ) -> Optional["ModelQualityJobDefinition"]:
@@ -21370,7 +21386,7 @@ class MonitoringAlert(Base):
21370
21386
  alert_status: str
21371
21387
  datapoints_to_alert: int
21372
21388
  evaluation_period: int
21373
- actions: MonitoringAlertActions
21389
+ actions: shapes.MonitoringAlertActions
21374
21390
 
21375
21391
  def get_name(self) -> str:
21376
21392
  attributes = vars(self)
@@ -21507,7 +21523,7 @@ class MonitoringAlert(Base):
21507
21523
  status_equals: Optional[str] = Unassigned(),
21508
21524
  session: Optional[Session] = None,
21509
21525
  region: Optional[str] = None,
21510
- ) -> Optional[MonitoringAlertHistorySummary]:
21526
+ ) -> Optional[shapes.MonitoringAlertHistorySummary]:
21511
21527
  """
21512
21528
  Gets a list of past alerts in a model monitoring schedule.
21513
21529
 
@@ -21524,7 +21540,7 @@ class MonitoringAlert(Base):
21524
21540
  region: Region name.
21525
21541
 
21526
21542
  Returns:
21527
- MonitoringAlertHistorySummary
21543
+ shapes.MonitoringAlertHistorySummary
21528
21544
 
21529
21545
  Raises:
21530
21546
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -21563,7 +21579,7 @@ class MonitoringAlert(Base):
21563
21579
  logger.debug(f"Response: {response}")
21564
21580
 
21565
21581
  transformed_response = transform(response, "ListMonitoringAlertHistoryResponse")
21566
- return MonitoringAlertHistorySummary(**transformed_response)
21582
+ return shapes.MonitoringAlertHistorySummary(**transformed_response)
21567
21583
 
21568
21584
 
21569
21585
  class MonitoringExecution(Base):
@@ -21727,9 +21743,9 @@ class MonitoringSchedule(Base):
21727
21743
  failure_reason: Optional[str] = Unassigned()
21728
21744
  creation_time: Optional[datetime.datetime] = Unassigned()
21729
21745
  last_modified_time: Optional[datetime.datetime] = Unassigned()
21730
- monitoring_schedule_config: Optional[MonitoringScheduleConfig] = Unassigned()
21746
+ monitoring_schedule_config: Optional[shapes.MonitoringScheduleConfig] = Unassigned()
21731
21747
  endpoint_name: Optional[str] = Unassigned()
21732
- last_monitoring_execution_summary: Optional[MonitoringExecutionSummary] = Unassigned()
21748
+ last_monitoring_execution_summary: Optional[shapes.MonitoringExecutionSummary] = Unassigned()
21733
21749
 
21734
21750
  def get_name(self) -> str:
21735
21751
  attributes = vars(self)
@@ -21789,8 +21805,8 @@ class MonitoringSchedule(Base):
21789
21805
  def create(
21790
21806
  cls,
21791
21807
  monitoring_schedule_name: str,
21792
- monitoring_schedule_config: MonitoringScheduleConfig,
21793
- tags: Optional[List[Tag]] = Unassigned(),
21808
+ monitoring_schedule_config: shapes.MonitoringScheduleConfig,
21809
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
21794
21810
  session: Optional[Session] = None,
21795
21811
  region: Optional[str] = None,
21796
21812
  ) -> Optional["MonitoringSchedule"]:
@@ -21944,7 +21960,7 @@ class MonitoringSchedule(Base):
21944
21960
  @Base.add_validate_call
21945
21961
  def update(
21946
21962
  self,
21947
- monitoring_schedule_config: MonitoringScheduleConfig,
21963
+ monitoring_schedule_config: shapes.MonitoringScheduleConfig,
21948
21964
  ) -> Optional["MonitoringSchedule"]:
21949
21965
  """
21950
21966
  Update a MonitoringSchedule resource
@@ -22247,9 +22263,9 @@ class NotebookInstance(Base):
22247
22263
  additional_code_repositories: Optional[List[str]] = Unassigned()
22248
22264
  root_access: Optional[str] = Unassigned()
22249
22265
  platform_identifier: Optional[str] = Unassigned()
22250
- instance_metadata_service_configuration: Optional[InstanceMetadataServiceConfiguration] = (
22251
- Unassigned()
22252
- )
22266
+ instance_metadata_service_configuration: Optional[
22267
+ shapes.InstanceMetadataServiceConfiguration
22268
+ ] = Unassigned()
22253
22269
 
22254
22270
  def get_name(self) -> str:
22255
22271
  attributes = vars(self)
@@ -22296,7 +22312,7 @@ class NotebookInstance(Base):
22296
22312
  subnet_id: Optional[str] = Unassigned(),
22297
22313
  security_group_ids: Optional[List[str]] = Unassigned(),
22298
22314
  kms_key_id: Optional[str] = Unassigned(),
22299
- tags: Optional[List[Tag]] = Unassigned(),
22315
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
22300
22316
  lifecycle_config_name: Optional[str] = Unassigned(),
22301
22317
  direct_internet_access: Optional[str] = Unassigned(),
22302
22318
  volume_size_in_gb: Optional[int] = Unassigned(),
@@ -22306,7 +22322,7 @@ class NotebookInstance(Base):
22306
22322
  root_access: Optional[str] = Unassigned(),
22307
22323
  platform_identifier: Optional[str] = Unassigned(),
22308
22324
  instance_metadata_service_configuration: Optional[
22309
- InstanceMetadataServiceConfiguration
22325
+ shapes.InstanceMetadataServiceConfiguration
22310
22326
  ] = Unassigned(),
22311
22327
  session: Optional[Session] = None,
22312
22328
  region: Optional[str] = None,
@@ -22497,7 +22513,7 @@ class NotebookInstance(Base):
22497
22513
  disassociate_additional_code_repositories: Optional[bool] = Unassigned(),
22498
22514
  root_access: Optional[str] = Unassigned(),
22499
22515
  instance_metadata_service_configuration: Optional[
22500
- InstanceMetadataServiceConfiguration
22516
+ shapes.InstanceMetadataServiceConfiguration
22501
22517
  ] = Unassigned(),
22502
22518
  ) -> Optional["NotebookInstance"]:
22503
22519
  """
@@ -22845,8 +22861,8 @@ class NotebookInstanceLifecycleConfig(Base):
22845
22861
 
22846
22862
  notebook_instance_lifecycle_config_name: str
22847
22863
  notebook_instance_lifecycle_config_arn: Optional[str] = Unassigned()
22848
- on_create: Optional[List[NotebookInstanceLifecycleHook]] = Unassigned()
22849
- on_start: Optional[List[NotebookInstanceLifecycleHook]] = Unassigned()
22864
+ on_create: Optional[List[shapes.NotebookInstanceLifecycleHook]] = Unassigned()
22865
+ on_start: Optional[List[shapes.NotebookInstanceLifecycleHook]] = Unassigned()
22850
22866
  last_modified_time: Optional[datetime.datetime] = Unassigned()
22851
22867
  creation_time: Optional[datetime.datetime] = Unassigned()
22852
22868
 
@@ -22871,9 +22887,9 @@ class NotebookInstanceLifecycleConfig(Base):
22871
22887
  def create(
22872
22888
  cls,
22873
22889
  notebook_instance_lifecycle_config_name: str,
22874
- on_create: Optional[List[NotebookInstanceLifecycleHook]] = Unassigned(),
22875
- on_start: Optional[List[NotebookInstanceLifecycleHook]] = Unassigned(),
22876
- tags: Optional[List[Tag]] = Unassigned(),
22890
+ on_create: Optional[List[shapes.NotebookInstanceLifecycleHook]] = Unassigned(),
22891
+ on_start: Optional[List[shapes.NotebookInstanceLifecycleHook]] = Unassigned(),
22892
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
22877
22893
  session: Optional[Session] = None,
22878
22894
  region: Optional[str] = None,
22879
22895
  ) -> Optional["NotebookInstanceLifecycleConfig"]:
@@ -23028,8 +23044,8 @@ class NotebookInstanceLifecycleConfig(Base):
23028
23044
  @Base.add_validate_call
23029
23045
  def update(
23030
23046
  self,
23031
- on_create: Optional[List[NotebookInstanceLifecycleHook]] = Unassigned(),
23032
- on_start: Optional[List[NotebookInstanceLifecycleHook]] = Unassigned(),
23047
+ on_create: Optional[List[shapes.NotebookInstanceLifecycleHook]] = Unassigned(),
23048
+ on_start: Optional[List[shapes.NotebookInstanceLifecycleHook]] = Unassigned(),
23033
23049
  ) -> Optional["NotebookInstanceLifecycleConfig"]:
23034
23050
  """
23035
23051
  Update a NotebookInstanceLifecycleConfig resource
@@ -23208,15 +23224,15 @@ class OptimizationJob(Base):
23208
23224
  creation_time: Optional[datetime.datetime] = Unassigned()
23209
23225
  last_modified_time: Optional[datetime.datetime] = Unassigned()
23210
23226
  failure_reason: Optional[str] = Unassigned()
23211
- model_source: Optional[OptimizationJobModelSource] = Unassigned()
23227
+ model_source: Optional[shapes.OptimizationJobModelSource] = Unassigned()
23212
23228
  optimization_environment: Optional[Dict[str, str]] = Unassigned()
23213
23229
  deployment_instance_type: Optional[str] = Unassigned()
23214
- optimization_configs: Optional[List[OptimizationConfig]] = Unassigned()
23215
- output_config: Optional[OptimizationJobOutputConfig] = Unassigned()
23216
- optimization_output: Optional[OptimizationOutput] = Unassigned()
23230
+ optimization_configs: Optional[List[shapes.OptimizationConfig]] = Unassigned()
23231
+ output_config: Optional[shapes.OptimizationJobOutputConfig] = Unassigned()
23232
+ optimization_output: Optional[shapes.OptimizationOutput] = Unassigned()
23217
23233
  role_arn: Optional[str] = Unassigned()
23218
- stopping_condition: Optional[StoppingCondition] = Unassigned()
23219
- vpc_config: Optional[OptimizationVpcConfig] = Unassigned()
23234
+ stopping_condition: Optional[shapes.StoppingCondition] = Unassigned()
23235
+ vpc_config: Optional[shapes.OptimizationVpcConfig] = Unassigned()
23220
23236
 
23221
23237
  def get_name(self) -> str:
23222
23238
  attributes = vars(self)
@@ -23265,14 +23281,14 @@ class OptimizationJob(Base):
23265
23281
  cls,
23266
23282
  optimization_job_name: str,
23267
23283
  role_arn: str,
23268
- model_source: OptimizationJobModelSource,
23284
+ model_source: shapes.OptimizationJobModelSource,
23269
23285
  deployment_instance_type: str,
23270
- optimization_configs: List[OptimizationConfig],
23271
- output_config: OptimizationJobOutputConfig,
23272
- stopping_condition: StoppingCondition,
23286
+ optimization_configs: List[shapes.OptimizationConfig],
23287
+ output_config: shapes.OptimizationJobOutputConfig,
23288
+ stopping_condition: shapes.StoppingCondition,
23273
23289
  optimization_environment: Optional[Dict[str, str]] = Unassigned(),
23274
- tags: Optional[List[Tag]] = Unassigned(),
23275
- vpc_config: Optional[OptimizationVpcConfig] = Unassigned(),
23290
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
23291
+ vpc_config: Optional[shapes.OptimizationVpcConfig] = Unassigned(),
23276
23292
  session: Optional[Session] = None,
23277
23293
  region: Optional[str] = None,
23278
23294
  ) -> Optional["OptimizationJob"]:
@@ -23671,13 +23687,13 @@ class PartnerApp(Base):
23671
23687
  execution_role_arn: Optional[str] = Unassigned()
23672
23688
  kms_key_id: Optional[str] = Unassigned()
23673
23689
  base_url: Optional[str] = Unassigned()
23674
- maintenance_config: Optional[PartnerAppMaintenanceConfig] = Unassigned()
23690
+ maintenance_config: Optional[shapes.PartnerAppMaintenanceConfig] = Unassigned()
23675
23691
  tier: Optional[str] = Unassigned()
23676
23692
  version: Optional[str] = Unassigned()
23677
- application_config: Optional[PartnerAppConfig] = Unassigned()
23693
+ application_config: Optional[shapes.PartnerAppConfig] = Unassigned()
23678
23694
  auth_type: Optional[str] = Unassigned()
23679
23695
  enable_iam_session_based_identity: Optional[bool] = Unassigned()
23680
- error: Optional[ErrorInfo] = Unassigned()
23696
+ error: Optional[shapes.ErrorInfo] = Unassigned()
23681
23697
 
23682
23698
  def get_name(self) -> str:
23683
23699
  attributes = vars(self)
@@ -23722,11 +23738,11 @@ class PartnerApp(Base):
23722
23738
  tier: str,
23723
23739
  auth_type: str,
23724
23740
  kms_key_id: Optional[str] = Unassigned(),
23725
- maintenance_config: Optional[PartnerAppMaintenanceConfig] = Unassigned(),
23726
- application_config: Optional[PartnerAppConfig] = Unassigned(),
23741
+ maintenance_config: Optional[shapes.PartnerAppMaintenanceConfig] = Unassigned(),
23742
+ application_config: Optional[shapes.PartnerAppConfig] = Unassigned(),
23727
23743
  enable_iam_session_based_identity: Optional[bool] = Unassigned(),
23728
23744
  client_token: Optional[str] = Unassigned(),
23729
- tags: Optional[List[Tag]] = Unassigned(),
23745
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
23730
23746
  session: Optional[Session] = None,
23731
23747
  region: Optional[str] = None,
23732
23748
  ) -> Optional["PartnerApp"]:
@@ -23894,12 +23910,12 @@ class PartnerApp(Base):
23894
23910
  @Base.add_validate_call
23895
23911
  def update(
23896
23912
  self,
23897
- maintenance_config: Optional[PartnerAppMaintenanceConfig] = Unassigned(),
23913
+ maintenance_config: Optional[shapes.PartnerAppMaintenanceConfig] = Unassigned(),
23898
23914
  tier: Optional[str] = Unassigned(),
23899
- application_config: Optional[PartnerAppConfig] = Unassigned(),
23915
+ application_config: Optional[shapes.PartnerAppConfig] = Unassigned(),
23900
23916
  enable_iam_session_based_identity: Optional[bool] = Unassigned(),
23901
23917
  client_token: Optional[str] = Unassigned(),
23902
- tags: Optional[List[Tag]] = Unassigned(),
23918
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
23903
23919
  ) -> Optional["PartnerApp"]:
23904
23920
  """
23905
23921
  Update a PartnerApp resource
@@ -24261,9 +24277,9 @@ class Pipeline(Base):
24261
24277
  creation_time: Optional[datetime.datetime] = Unassigned()
24262
24278
  last_modified_time: Optional[datetime.datetime] = Unassigned()
24263
24279
  last_run_time: Optional[datetime.datetime] = Unassigned()
24264
- created_by: Optional[UserContext] = Unassigned()
24265
- last_modified_by: Optional[UserContext] = Unassigned()
24266
- parallelism_configuration: Optional[ParallelismConfiguration] = Unassigned()
24280
+ created_by: Optional[shapes.UserContext] = Unassigned()
24281
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
24282
+ parallelism_configuration: Optional[shapes.ParallelismConfiguration] = Unassigned()
24267
24283
 
24268
24284
  def get_name(self) -> str:
24269
24285
  attributes = vars(self)
@@ -24304,10 +24320,12 @@ class Pipeline(Base):
24304
24320
  role_arn: str,
24305
24321
  pipeline_display_name: Optional[str] = Unassigned(),
24306
24322
  pipeline_definition: Optional[str] = Unassigned(),
24307
- pipeline_definition_s3_location: Optional[PipelineDefinitionS3Location] = Unassigned(),
24323
+ pipeline_definition_s3_location: Optional[
24324
+ shapes.PipelineDefinitionS3Location
24325
+ ] = Unassigned(),
24308
24326
  pipeline_description: Optional[str] = Unassigned(),
24309
- tags: Optional[List[Tag]] = Unassigned(),
24310
- parallelism_configuration: Optional[ParallelismConfiguration] = Unassigned(),
24327
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
24328
+ parallelism_configuration: Optional[shapes.ParallelismConfiguration] = Unassigned(),
24311
24329
  session: Optional[Session] = None,
24312
24330
  region: Optional[str] = None,
24313
24331
  ) -> Optional["Pipeline"]:
@@ -24474,10 +24492,12 @@ class Pipeline(Base):
24474
24492
  self,
24475
24493
  pipeline_display_name: Optional[str] = Unassigned(),
24476
24494
  pipeline_definition: Optional[str] = Unassigned(),
24477
- pipeline_definition_s3_location: Optional[PipelineDefinitionS3Location] = Unassigned(),
24495
+ pipeline_definition_s3_location: Optional[
24496
+ shapes.PipelineDefinitionS3Location
24497
+ ] = Unassigned(),
24478
24498
  pipeline_description: Optional[str] = Unassigned(),
24479
24499
  role_arn: Optional[str] = Unassigned(),
24480
- parallelism_configuration: Optional[ParallelismConfiguration] = Unassigned(),
24500
+ parallelism_configuration: Optional[shapes.ParallelismConfiguration] = Unassigned(),
24481
24501
  ) -> Optional["Pipeline"]:
24482
24502
  """
24483
24503
  Update a Pipeline resource
@@ -24767,14 +24787,14 @@ class PipelineExecution(Base):
24767
24787
  pipeline_execution_display_name: Optional[str] = Unassigned()
24768
24788
  pipeline_execution_status: Optional[str] = Unassigned()
24769
24789
  pipeline_execution_description: Optional[str] = Unassigned()
24770
- pipeline_experiment_config: Optional[PipelineExperimentConfig] = Unassigned()
24790
+ pipeline_experiment_config: Optional[shapes.PipelineExperimentConfig] = Unassigned()
24771
24791
  failure_reason: Optional[str] = Unassigned()
24772
24792
  creation_time: Optional[datetime.datetime] = Unassigned()
24773
24793
  last_modified_time: Optional[datetime.datetime] = Unassigned()
24774
- created_by: Optional[UserContext] = Unassigned()
24775
- last_modified_by: Optional[UserContext] = Unassigned()
24776
- parallelism_configuration: Optional[ParallelismConfiguration] = Unassigned()
24777
- selective_execution_config: Optional[SelectiveExecutionConfig] = Unassigned()
24794
+ created_by: Optional[shapes.UserContext] = Unassigned()
24795
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
24796
+ parallelism_configuration: Optional[shapes.ParallelismConfiguration] = Unassigned()
24797
+ selective_execution_config: Optional[shapes.SelectiveExecutionConfig] = Unassigned()
24778
24798
 
24779
24799
  def get_name(self) -> str:
24780
24800
  attributes = vars(self)
@@ -24885,7 +24905,7 @@ class PipelineExecution(Base):
24885
24905
  self,
24886
24906
  pipeline_execution_description: Optional[str] = Unassigned(),
24887
24907
  pipeline_execution_display_name: Optional[str] = Unassigned(),
24888
- parallelism_configuration: Optional[ParallelismConfiguration] = Unassigned(),
24908
+ parallelism_configuration: Optional[shapes.ParallelismConfiguration] = Unassigned(),
24889
24909
  ) -> Optional["PipelineExecution"]:
24890
24910
  """
24891
24911
  Update a PipelineExecution resource
@@ -25093,7 +25113,7 @@ class PipelineExecution(Base):
25093
25113
  self,
25094
25114
  session: Optional[Session] = None,
25095
25115
  region: Optional[str] = None,
25096
- ) -> Optional[DescribePipelineDefinitionForExecutionResponse]:
25116
+ ) -> Optional[shapes.DescribePipelineDefinitionForExecutionResponse]:
25097
25117
  """
25098
25118
  Describes the details of an execution's pipeline definition.
25099
25119
 
@@ -25102,7 +25122,7 @@ class PipelineExecution(Base):
25102
25122
  region: Region name.
25103
25123
 
25104
25124
  Returns:
25105
- DescribePipelineDefinitionForExecutionResponse
25125
+ shapes.DescribePipelineDefinitionForExecutionResponse
25106
25126
 
25107
25127
  Raises:
25108
25128
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -25133,7 +25153,7 @@ class PipelineExecution(Base):
25133
25153
  logger.debug(f"Response: {response}")
25134
25154
 
25135
25155
  transformed_response = transform(response, "DescribePipelineDefinitionForExecutionResponse")
25136
- return DescribePipelineDefinitionForExecutionResponse(**transformed_response)
25156
+ return shapes.DescribePipelineDefinitionForExecutionResponse(**transformed_response)
25137
25157
 
25138
25158
  @Base.add_validate_call
25139
25159
  def get_all_steps(
@@ -25141,7 +25161,7 @@ class PipelineExecution(Base):
25141
25161
  sort_order: Optional[str] = Unassigned(),
25142
25162
  session: Optional[Session] = None,
25143
25163
  region: Optional[str] = None,
25144
- ) -> ResourceIterator[PipelineExecutionStep]:
25164
+ ) -> ResourceIterator[shapes.PipelineExecutionStep]:
25145
25165
  """
25146
25166
  Gets a list of PipeLineExecutionStep objects.
25147
25167
 
@@ -25185,7 +25205,7 @@ class PipelineExecution(Base):
25185
25205
  list_method="list_pipeline_execution_steps",
25186
25206
  summaries_key="PipelineExecutionSteps",
25187
25207
  summary_name="PipelineExecutionStep",
25188
- resource_cls=PipelineExecutionStep,
25208
+ resource_cls=shapes.PipelineExecutionStep,
25189
25209
  list_method_kwargs=operation_input_args,
25190
25210
  )
25191
25211
 
@@ -25194,7 +25214,7 @@ class PipelineExecution(Base):
25194
25214
  self,
25195
25215
  session: Optional[Session] = None,
25196
25216
  region: Optional[str] = None,
25197
- ) -> ResourceIterator[Parameter]:
25217
+ ) -> ResourceIterator[shapes.Parameter]:
25198
25218
  """
25199
25219
  Gets a list of parameters for a pipeline execution.
25200
25220
 
@@ -25236,7 +25256,7 @@ class PipelineExecution(Base):
25236
25256
  list_method="list_pipeline_parameters_for_execution",
25237
25257
  summaries_key="PipelineParameters",
25238
25258
  summary_name="Parameter",
25239
- resource_cls=Parameter,
25259
+ resource_cls=shapes.Parameter,
25240
25260
  list_method_kwargs=operation_input_args,
25241
25261
  )
25242
25262
 
@@ -25340,7 +25360,7 @@ class PipelineExecution(Base):
25340
25360
  def send_execution_step_success(
25341
25361
  self,
25342
25362
  callback_token: str,
25343
- output_parameters: Optional[List[OutputParameter]] = Unassigned(),
25363
+ output_parameters: Optional[List[shapes.OutputParameter]] = Unassigned(),
25344
25364
  client_request_token: Optional[str] = Unassigned(),
25345
25365
  session: Optional[Session] = None,
25346
25366
  region: Optional[str] = None,
@@ -25700,15 +25720,15 @@ class ProcessingJob(Base):
25700
25720
  """
25701
25721
 
25702
25722
  processing_job_name: str
25703
- processing_inputs: Optional[List[ProcessingInput]] = Unassigned()
25704
- processing_output_config: Optional[ProcessingOutputConfig] = Unassigned()
25705
- processing_resources: Optional[ProcessingResources] = Unassigned()
25706
- stopping_condition: Optional[ProcessingStoppingCondition] = Unassigned()
25707
- app_specification: Optional[AppSpecification] = Unassigned()
25723
+ processing_inputs: Optional[List[shapes.ProcessingInput]] = Unassigned()
25724
+ processing_output_config: Optional[shapes.ProcessingOutputConfig] = Unassigned()
25725
+ processing_resources: Optional[shapes.ProcessingResources] = Unassigned()
25726
+ stopping_condition: Optional[shapes.ProcessingStoppingCondition] = Unassigned()
25727
+ app_specification: Optional[shapes.AppSpecification] = Unassigned()
25708
25728
  environment: Optional[Dict[str, str]] = Unassigned()
25709
- network_config: Optional[NetworkConfig] = Unassigned()
25729
+ network_config: Optional[shapes.NetworkConfig] = Unassigned()
25710
25730
  role_arn: Optional[str] = Unassigned()
25711
- experiment_config: Optional[ExperimentConfig] = Unassigned()
25731
+ experiment_config: Optional[shapes.ExperimentConfig] = Unassigned()
25712
25732
  processing_job_arn: Optional[str] = Unassigned()
25713
25733
  processing_job_status: Optional[str] = Unassigned()
25714
25734
  exit_message: Optional[str] = Unassigned()
@@ -25768,16 +25788,16 @@ class ProcessingJob(Base):
25768
25788
  def create(
25769
25789
  cls,
25770
25790
  processing_job_name: str,
25771
- processing_resources: ProcessingResources,
25772
- app_specification: AppSpecification,
25791
+ processing_resources: shapes.ProcessingResources,
25792
+ app_specification: shapes.AppSpecification,
25773
25793
  role_arn: str,
25774
- processing_inputs: Optional[List[ProcessingInput]] = Unassigned(),
25775
- processing_output_config: Optional[ProcessingOutputConfig] = Unassigned(),
25776
- stopping_condition: Optional[ProcessingStoppingCondition] = Unassigned(),
25794
+ processing_inputs: Optional[List[shapes.ProcessingInput]] = Unassigned(),
25795
+ processing_output_config: Optional[shapes.ProcessingOutputConfig] = Unassigned(),
25796
+ stopping_condition: Optional[shapes.ProcessingStoppingCondition] = Unassigned(),
25777
25797
  environment: Optional[Dict[str, str]] = Unassigned(),
25778
- network_config: Optional[NetworkConfig] = Unassigned(),
25779
- tags: Optional[List[Tag]] = Unassigned(),
25780
- experiment_config: Optional[ExperimentConfig] = Unassigned(),
25798
+ network_config: Optional[shapes.NetworkConfig] = Unassigned(),
25799
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
25800
+ experiment_config: Optional[shapes.ExperimentConfig] = Unassigned(),
25781
25801
  session: Optional[Session] = None,
25782
25802
  region: Optional[str] = None,
25783
25803
  ) -> Optional["ProcessingJob"]:
@@ -26131,10 +26151,10 @@ class Project(Base):
26131
26151
  project_arn: The Amazon Resource Name (ARN) of the project.
26132
26152
  project_name: The name of the project.
26133
26153
  project_id: The ID of the project.
26134
- service_catalog_provisioning_details: Information used to provision a service catalog product. For information, see What is Amazon Web Services Service Catalog.
26135
26154
  project_status: The status of the project.
26136
26155
  creation_time: The time when the project was created.
26137
26156
  project_description: The description of the project.
26157
+ service_catalog_provisioning_details: Information used to provision a service catalog product. For information, see What is Amazon Web Services Service Catalog.
26138
26158
  service_catalog_provisioned_product_details: Information about a provisioned service catalog product.
26139
26159
  created_by:
26140
26160
  last_modified_time: The timestamp when project was last modified.
@@ -26146,15 +26166,17 @@ class Project(Base):
26146
26166
  project_arn: Optional[str] = Unassigned()
26147
26167
  project_id: Optional[str] = Unassigned()
26148
26168
  project_description: Optional[str] = Unassigned()
26149
- service_catalog_provisioning_details: Optional[ServiceCatalogProvisioningDetails] = Unassigned()
26169
+ service_catalog_provisioning_details: Optional[shapes.ServiceCatalogProvisioningDetails] = (
26170
+ Unassigned()
26171
+ )
26150
26172
  service_catalog_provisioned_product_details: Optional[
26151
- ServiceCatalogProvisionedProductDetails
26173
+ shapes.ServiceCatalogProvisionedProductDetails
26152
26174
  ] = Unassigned()
26153
26175
  project_status: Optional[str] = Unassigned()
26154
- created_by: Optional[UserContext] = Unassigned()
26176
+ created_by: Optional[shapes.UserContext] = Unassigned()
26155
26177
  creation_time: Optional[datetime.datetime] = Unassigned()
26156
26178
  last_modified_time: Optional[datetime.datetime] = Unassigned()
26157
- last_modified_by: Optional[UserContext] = Unassigned()
26179
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
26158
26180
 
26159
26181
  def get_name(self) -> str:
26160
26182
  attributes = vars(self)
@@ -26177,9 +26199,11 @@ class Project(Base):
26177
26199
  def create(
26178
26200
  cls,
26179
26201
  project_name: str,
26180
- service_catalog_provisioning_details: ServiceCatalogProvisioningDetails,
26181
26202
  project_description: Optional[str] = Unassigned(),
26182
- tags: Optional[List[Tag]] = Unassigned(),
26203
+ service_catalog_provisioning_details: Optional[
26204
+ shapes.ServiceCatalogProvisioningDetails
26205
+ ] = Unassigned(),
26206
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
26183
26207
  session: Optional[Session] = None,
26184
26208
  region: Optional[str] = None,
26185
26209
  ) -> Optional["Project"]:
@@ -26188,8 +26212,8 @@ class Project(Base):
26188
26212
 
26189
26213
  Parameters:
26190
26214
  project_name: The name of the project.
26191
- service_catalog_provisioning_details: The product ID and provisioning artifact ID to provision a service catalog. The provisioning artifact ID will default to the latest provisioning artifact ID of the product, if you don't provide the provisioning artifact ID. For more information, see What is Amazon Web Services Service Catalog.
26192
26215
  project_description: A description for the project.
26216
+ service_catalog_provisioning_details: The product ID and provisioning artifact ID to provision a service catalog. The provisioning artifact ID will default to the latest provisioning artifact ID of the product, if you don't provide the provisioning artifact ID. For more information, see What is Amazon Web Services Service Catalog.
26193
26217
  tags: An array of key-value pairs that you want to use to organize and track your Amazon Web Services resource costs. For more information, see Tagging Amazon Web Services resources in the Amazon Web Services General Reference Guide.
26194
26218
  session: Boto3 session.
26195
26219
  region: Region name.
@@ -26331,9 +26355,9 @@ class Project(Base):
26331
26355
  self,
26332
26356
  project_description: Optional[str] = Unassigned(),
26333
26357
  service_catalog_provisioning_update_details: Optional[
26334
- ServiceCatalogProvisioningUpdateDetails
26358
+ shapes.ServiceCatalogProvisioningUpdateDetails
26335
26359
  ] = Unassigned(),
26336
- tags: Optional[List[Tag]] = Unassigned(),
26360
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
26337
26361
  ) -> Optional["Project"]:
26338
26362
  """
26339
26363
  Update a Project resource
@@ -26786,9 +26810,9 @@ class Space(Base):
26786
26810
  last_modified_time: Optional[datetime.datetime] = Unassigned()
26787
26811
  creation_time: Optional[datetime.datetime] = Unassigned()
26788
26812
  failure_reason: Optional[str] = Unassigned()
26789
- space_settings: Optional[SpaceSettings] = Unassigned()
26790
- ownership_settings: Optional[OwnershipSettings] = Unassigned()
26791
- space_sharing_settings: Optional[SpaceSharingSettings] = Unassigned()
26813
+ space_settings: Optional[shapes.SpaceSettings] = Unassigned()
26814
+ ownership_settings: Optional[shapes.OwnershipSettings] = Unassigned()
26815
+ space_sharing_settings: Optional[shapes.SpaceSharingSettings] = Unassigned()
26792
26816
  space_display_name: Optional[str] = Unassigned()
26793
26817
  url: Optional[str] = Unassigned()
26794
26818
 
@@ -26814,10 +26838,10 @@ class Space(Base):
26814
26838
  cls,
26815
26839
  domain_id: str,
26816
26840
  space_name: str,
26817
- tags: Optional[List[Tag]] = Unassigned(),
26818
- space_settings: Optional[SpaceSettings] = Unassigned(),
26819
- ownership_settings: Optional[OwnershipSettings] = Unassigned(),
26820
- space_sharing_settings: Optional[SpaceSharingSettings] = Unassigned(),
26841
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
26842
+ space_settings: Optional[shapes.SpaceSettings] = Unassigned(),
26843
+ ownership_settings: Optional[shapes.OwnershipSettings] = Unassigned(),
26844
+ space_sharing_settings: Optional[shapes.SpaceSharingSettings] = Unassigned(),
26821
26845
  space_display_name: Optional[str] = Unassigned(),
26822
26846
  session: Optional[Session] = None,
26823
26847
  region: Optional[str] = None,
@@ -26981,7 +27005,7 @@ class Space(Base):
26981
27005
  @Base.add_validate_call
26982
27006
  def update(
26983
27007
  self,
26984
- space_settings: Optional[SpaceSettings] = Unassigned(),
27008
+ space_settings: Optional[shapes.SpaceSettings] = Unassigned(),
26985
27009
  space_display_name: Optional[str] = Unassigned(),
26986
27010
  ) -> Optional["Space"]:
26987
27011
  """
@@ -27302,7 +27326,7 @@ class StudioLifecycleConfig(Base):
27302
27326
  studio_lifecycle_config_name: str,
27303
27327
  studio_lifecycle_config_content: str,
27304
27328
  studio_lifecycle_config_app_type: str,
27305
- tags: Optional[List[Tag]] = Unassigned(),
27329
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
27306
27330
  session: Optional[Session] = None,
27307
27331
  region: Optional[str] = None,
27308
27332
  ) -> Optional["StudioLifecycleConfig"]:
@@ -27576,7 +27600,7 @@ class SubscribedWorkteam(Base):
27576
27600
  """
27577
27601
 
27578
27602
  workteam_arn: str
27579
- subscribed_workteam: Optional[SubscribedWorkteam] = Unassigned()
27603
+ subscribed_workteam: Optional[shapes.SubscribedWorkteam] = Unassigned()
27580
27604
 
27581
27605
  def get_name(self) -> str:
27582
27606
  attributes = vars(self)
@@ -27823,7 +27847,7 @@ class Tag(Base):
27823
27847
  def add_tags(
27824
27848
  cls,
27825
27849
  resource_arn: str,
27826
- tags: List[Tag],
27850
+ tags: List[shapes.Tag],
27827
27851
  session: Optional[Session] = None,
27828
27852
  region: Optional[str] = None,
27829
27853
  ) -> None:
@@ -27967,44 +27991,46 @@ class TrainingJob(Base):
27967
27991
  tuning_job_arn: Optional[str] = Unassigned()
27968
27992
  labeling_job_arn: Optional[str] = Unassigned()
27969
27993
  auto_ml_job_arn: Optional[str] = Unassigned()
27970
- model_artifacts: Optional[ModelArtifacts] = Unassigned()
27994
+ model_artifacts: Optional[shapes.ModelArtifacts] = Unassigned()
27971
27995
  training_job_status: Optional[str] = Unassigned()
27972
27996
  secondary_status: Optional[str] = Unassigned()
27973
27997
  failure_reason: Optional[str] = Unassigned()
27974
27998
  hyper_parameters: Optional[Dict[str, str]] = Unassigned()
27975
- algorithm_specification: Optional[AlgorithmSpecification] = Unassigned()
27999
+ algorithm_specification: Optional[shapes.AlgorithmSpecification] = Unassigned()
27976
28000
  role_arn: Optional[str] = Unassigned()
27977
- input_data_config: Optional[List[Channel]] = Unassigned()
27978
- output_data_config: Optional[OutputDataConfig] = Unassigned()
27979
- resource_config: Optional[ResourceConfig] = Unassigned()
27980
- warm_pool_status: Optional[WarmPoolStatus] = Unassigned()
27981
- vpc_config: Optional[VpcConfig] = Unassigned()
27982
- stopping_condition: Optional[StoppingCondition] = Unassigned()
28001
+ input_data_config: Optional[List[shapes.Channel]] = Unassigned()
28002
+ output_data_config: Optional[shapes.OutputDataConfig] = Unassigned()
28003
+ resource_config: Optional[shapes.ResourceConfig] = Unassigned()
28004
+ warm_pool_status: Optional[shapes.WarmPoolStatus] = Unassigned()
28005
+ vpc_config: Optional[shapes.VpcConfig] = Unassigned()
28006
+ stopping_condition: Optional[shapes.StoppingCondition] = Unassigned()
27983
28007
  creation_time: Optional[datetime.datetime] = Unassigned()
27984
28008
  training_start_time: Optional[datetime.datetime] = Unassigned()
27985
28009
  training_end_time: Optional[datetime.datetime] = Unassigned()
27986
28010
  last_modified_time: Optional[datetime.datetime] = Unassigned()
27987
- secondary_status_transitions: Optional[List[SecondaryStatusTransition]] = Unassigned()
27988
- final_metric_data_list: Optional[List[MetricData]] = Unassigned()
28011
+ secondary_status_transitions: Optional[List[shapes.SecondaryStatusTransition]] = Unassigned()
28012
+ final_metric_data_list: Optional[List[shapes.MetricData]] = Unassigned()
27989
28013
  enable_network_isolation: Optional[bool] = Unassigned()
27990
28014
  enable_inter_container_traffic_encryption: Optional[bool] = Unassigned()
27991
28015
  enable_managed_spot_training: Optional[bool] = Unassigned()
27992
- checkpoint_config: Optional[CheckpointConfig] = Unassigned()
28016
+ checkpoint_config: Optional[shapes.CheckpointConfig] = Unassigned()
27993
28017
  training_time_in_seconds: Optional[int] = Unassigned()
27994
28018
  billable_time_in_seconds: Optional[int] = Unassigned()
27995
- debug_hook_config: Optional[DebugHookConfig] = Unassigned()
27996
- experiment_config: Optional[ExperimentConfig] = Unassigned()
27997
- debug_rule_configurations: Optional[List[DebugRuleConfiguration]] = Unassigned()
27998
- tensor_board_output_config: Optional[TensorBoardOutputConfig] = Unassigned()
27999
- debug_rule_evaluation_statuses: Optional[List[DebugRuleEvaluationStatus]] = Unassigned()
28000
- profiler_config: Optional[ProfilerConfig] = Unassigned()
28001
- profiler_rule_configurations: Optional[List[ProfilerRuleConfiguration]] = Unassigned()
28002
- profiler_rule_evaluation_statuses: Optional[List[ProfilerRuleEvaluationStatus]] = Unassigned()
28019
+ debug_hook_config: Optional[shapes.DebugHookConfig] = Unassigned()
28020
+ experiment_config: Optional[shapes.ExperimentConfig] = Unassigned()
28021
+ debug_rule_configurations: Optional[List[shapes.DebugRuleConfiguration]] = Unassigned()
28022
+ tensor_board_output_config: Optional[shapes.TensorBoardOutputConfig] = Unassigned()
28023
+ debug_rule_evaluation_statuses: Optional[List[shapes.DebugRuleEvaluationStatus]] = Unassigned()
28024
+ profiler_config: Optional[shapes.ProfilerConfig] = Unassigned()
28025
+ profiler_rule_configurations: Optional[List[shapes.ProfilerRuleConfiguration]] = Unassigned()
28026
+ profiler_rule_evaluation_statuses: Optional[List[shapes.ProfilerRuleEvaluationStatus]] = (
28027
+ Unassigned()
28028
+ )
28003
28029
  profiling_status: Optional[str] = Unassigned()
28004
28030
  environment: Optional[Dict[str, str]] = Unassigned()
28005
- retry_strategy: Optional[RetryStrategy] = Unassigned()
28006
- remote_debug_config: Optional[RemoteDebugConfig] = Unassigned()
28007
- infra_check_config: Optional[InfraCheckConfig] = Unassigned()
28031
+ retry_strategy: Optional[shapes.RetryStrategy] = Unassigned()
28032
+ remote_debug_config: Optional[shapes.RemoteDebugConfig] = Unassigned()
28033
+ infra_check_config: Optional[shapes.InfraCheckConfig] = Unassigned()
28008
28034
 
28009
28035
  def get_name(self) -> str:
28010
28036
  attributes = vars(self)
@@ -28057,30 +28083,32 @@ class TrainingJob(Base):
28057
28083
  def create(
28058
28084
  cls,
28059
28085
  training_job_name: str,
28060
- algorithm_specification: AlgorithmSpecification,
28086
+ algorithm_specification: shapes.AlgorithmSpecification,
28061
28087
  role_arn: str,
28062
- output_data_config: OutputDataConfig,
28063
- resource_config: ResourceConfig,
28064
- stopping_condition: StoppingCondition,
28088
+ output_data_config: shapes.OutputDataConfig,
28089
+ resource_config: shapes.ResourceConfig,
28090
+ stopping_condition: shapes.StoppingCondition,
28065
28091
  hyper_parameters: Optional[Dict[str, str]] = Unassigned(),
28066
- input_data_config: Optional[List[Channel]] = Unassigned(),
28067
- vpc_config: Optional[VpcConfig] = Unassigned(),
28068
- tags: Optional[List[Tag]] = Unassigned(),
28092
+ input_data_config: Optional[List[shapes.Channel]] = Unassigned(),
28093
+ vpc_config: Optional[shapes.VpcConfig] = Unassigned(),
28094
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
28069
28095
  enable_network_isolation: Optional[bool] = Unassigned(),
28070
28096
  enable_inter_container_traffic_encryption: Optional[bool] = Unassigned(),
28071
28097
  enable_managed_spot_training: Optional[bool] = Unassigned(),
28072
- checkpoint_config: Optional[CheckpointConfig] = Unassigned(),
28073
- debug_hook_config: Optional[DebugHookConfig] = Unassigned(),
28074
- debug_rule_configurations: Optional[List[DebugRuleConfiguration]] = Unassigned(),
28075
- tensor_board_output_config: Optional[TensorBoardOutputConfig] = Unassigned(),
28076
- experiment_config: Optional[ExperimentConfig] = Unassigned(),
28077
- profiler_config: Optional[ProfilerConfig] = Unassigned(),
28078
- profiler_rule_configurations: Optional[List[ProfilerRuleConfiguration]] = Unassigned(),
28098
+ checkpoint_config: Optional[shapes.CheckpointConfig] = Unassigned(),
28099
+ debug_hook_config: Optional[shapes.DebugHookConfig] = Unassigned(),
28100
+ debug_rule_configurations: Optional[List[shapes.DebugRuleConfiguration]] = Unassigned(),
28101
+ tensor_board_output_config: Optional[shapes.TensorBoardOutputConfig] = Unassigned(),
28102
+ experiment_config: Optional[shapes.ExperimentConfig] = Unassigned(),
28103
+ profiler_config: Optional[shapes.ProfilerConfig] = Unassigned(),
28104
+ profiler_rule_configurations: Optional[
28105
+ List[shapes.ProfilerRuleConfiguration]
28106
+ ] = Unassigned(),
28079
28107
  environment: Optional[Dict[str, str]] = Unassigned(),
28080
- retry_strategy: Optional[RetryStrategy] = Unassigned(),
28081
- remote_debug_config: Optional[RemoteDebugConfig] = Unassigned(),
28082
- infra_check_config: Optional[InfraCheckConfig] = Unassigned(),
28083
- session_chaining_config: Optional[SessionChainingConfig] = Unassigned(),
28108
+ retry_strategy: Optional[shapes.RetryStrategy] = Unassigned(),
28109
+ remote_debug_config: Optional[shapes.RemoteDebugConfig] = Unassigned(),
28110
+ infra_check_config: Optional[shapes.InfraCheckConfig] = Unassigned(),
28111
+ session_chaining_config: Optional[shapes.SessionChainingConfig] = Unassigned(),
28084
28112
  session: Optional[Session] = None,
28085
28113
  region: Optional[str] = None,
28086
28114
  ) -> Optional["TrainingJob"]:
@@ -28277,10 +28305,12 @@ class TrainingJob(Base):
28277
28305
  @Base.add_validate_call
28278
28306
  def update(
28279
28307
  self,
28280
- profiler_config: Optional[ProfilerConfigForUpdate] = Unassigned(),
28281
- profiler_rule_configurations: Optional[List[ProfilerRuleConfiguration]] = Unassigned(),
28282
- resource_config: Optional[ResourceConfigForUpdate] = Unassigned(),
28283
- remote_debug_config: Optional[RemoteDebugConfigForUpdate] = Unassigned(),
28308
+ profiler_config: Optional[shapes.ProfilerConfigForUpdate] = Unassigned(),
28309
+ profiler_rule_configurations: Optional[
28310
+ List[shapes.ProfilerRuleConfiguration]
28311
+ ] = Unassigned(),
28312
+ resource_config: Optional[shapes.ResourceConfigForUpdate] = Unassigned(),
28313
+ remote_debug_config: Optional[shapes.RemoteDebugConfigForUpdate] = Unassigned(),
28284
28314
  ) -> Optional["TrainingJob"]:
28285
28315
  """
28286
28316
  Update a TrainingJob resource
@@ -28557,7 +28587,7 @@ class TrainingPlan(Base):
28557
28587
  available_instance_count: Optional[int] = Unassigned()
28558
28588
  in_use_instance_count: Optional[int] = Unassigned()
28559
28589
  target_resources: Optional[List[str]] = Unassigned()
28560
- reserved_capacity_summaries: Optional[List[ReservedCapacitySummary]] = Unassigned()
28590
+ reserved_capacity_summaries: Optional[List[shapes.ReservedCapacitySummary]] = Unassigned()
28561
28591
 
28562
28592
  def get_name(self) -> str:
28563
28593
  attributes = vars(self)
@@ -28581,7 +28611,7 @@ class TrainingPlan(Base):
28581
28611
  cls,
28582
28612
  training_plan_name: str,
28583
28613
  training_plan_offering_id: str,
28584
- tags: Optional[List[Tag]] = Unassigned(),
28614
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
28585
28615
  session: Optional[Session] = None,
28586
28616
  region: Optional[str] = None,
28587
28617
  ) -> Optional["TrainingPlan"]:
@@ -28796,7 +28826,7 @@ class TrainingPlan(Base):
28796
28826
  start_time_before: Optional[datetime.datetime] = Unassigned(),
28797
28827
  sort_by: Optional[str] = Unassigned(),
28798
28828
  sort_order: Optional[str] = Unassigned(),
28799
- filters: Optional[List[TrainingPlanFilter]] = Unassigned(),
28829
+ filters: Optional[List[shapes.TrainingPlanFilter]] = Unassigned(),
28800
28830
  session: Optional[Session] = None,
28801
28831
  region: Optional[str] = None,
28802
28832
  ) -> ResourceIterator["TrainingPlan"]:
@@ -28890,21 +28920,21 @@ class TransformJob(Base):
28890
28920
  failure_reason: Optional[str] = Unassigned()
28891
28921
  model_name: Optional[str] = Unassigned()
28892
28922
  max_concurrent_transforms: Optional[int] = Unassigned()
28893
- model_client_config: Optional[ModelClientConfig] = Unassigned()
28923
+ model_client_config: Optional[shapes.ModelClientConfig] = Unassigned()
28894
28924
  max_payload_in_mb: Optional[int] = Unassigned()
28895
28925
  batch_strategy: Optional[str] = Unassigned()
28896
28926
  environment: Optional[Dict[str, str]] = Unassigned()
28897
- transform_input: Optional[TransformInput] = Unassigned()
28898
- transform_output: Optional[TransformOutput] = Unassigned()
28899
- data_capture_config: Optional[BatchDataCaptureConfig] = Unassigned()
28900
- transform_resources: Optional[TransformResources] = Unassigned()
28927
+ transform_input: Optional[shapes.TransformInput] = Unassigned()
28928
+ transform_output: Optional[shapes.TransformOutput] = Unassigned()
28929
+ data_capture_config: Optional[shapes.BatchDataCaptureConfig] = Unassigned()
28930
+ transform_resources: Optional[shapes.TransformResources] = Unassigned()
28901
28931
  creation_time: Optional[datetime.datetime] = Unassigned()
28902
28932
  transform_start_time: Optional[datetime.datetime] = Unassigned()
28903
28933
  transform_end_time: Optional[datetime.datetime] = Unassigned()
28904
28934
  labeling_job_arn: Optional[str] = Unassigned()
28905
28935
  auto_ml_job_arn: Optional[str] = Unassigned()
28906
- data_processing: Optional[DataProcessing] = Unassigned()
28907
- experiment_config: Optional[ExperimentConfig] = Unassigned()
28936
+ data_processing: Optional[shapes.DataProcessing] = Unassigned()
28937
+ experiment_config: Optional[shapes.ExperimentConfig] = Unassigned()
28908
28938
 
28909
28939
  def get_name(self) -> str:
28910
28940
  attributes = vars(self)
@@ -28960,18 +28990,18 @@ class TransformJob(Base):
28960
28990
  cls,
28961
28991
  transform_job_name: str,
28962
28992
  model_name: Union[str, object],
28963
- transform_input: TransformInput,
28964
- transform_output: TransformOutput,
28965
- transform_resources: TransformResources,
28993
+ transform_input: shapes.TransformInput,
28994
+ transform_output: shapes.TransformOutput,
28995
+ transform_resources: shapes.TransformResources,
28966
28996
  max_concurrent_transforms: Optional[int] = Unassigned(),
28967
- model_client_config: Optional[ModelClientConfig] = Unassigned(),
28997
+ model_client_config: Optional[shapes.ModelClientConfig] = Unassigned(),
28968
28998
  max_payload_in_mb: Optional[int] = Unassigned(),
28969
28999
  batch_strategy: Optional[str] = Unassigned(),
28970
29000
  environment: Optional[Dict[str, str]] = Unassigned(),
28971
- data_capture_config: Optional[BatchDataCaptureConfig] = Unassigned(),
28972
- data_processing: Optional[DataProcessing] = Unassigned(),
28973
- tags: Optional[List[Tag]] = Unassigned(),
28974
- experiment_config: Optional[ExperimentConfig] = Unassigned(),
29001
+ data_capture_config: Optional[shapes.BatchDataCaptureConfig] = Unassigned(),
29002
+ data_processing: Optional[shapes.DataProcessing] = Unassigned(),
29003
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
29004
+ experiment_config: Optional[shapes.ExperimentConfig] = Unassigned(),
28975
29005
  session: Optional[Session] = None,
28976
29006
  region: Optional[str] = None,
28977
29007
  ) -> Optional["TransformJob"]:
@@ -29345,12 +29375,12 @@ class Trial(Base):
29345
29375
  trial_arn: Optional[str] = Unassigned()
29346
29376
  display_name: Optional[str] = Unassigned()
29347
29377
  experiment_name: Optional[str] = Unassigned()
29348
- source: Optional[TrialSource] = Unassigned()
29378
+ source: Optional[shapes.TrialSource] = Unassigned()
29349
29379
  creation_time: Optional[datetime.datetime] = Unassigned()
29350
- created_by: Optional[UserContext] = Unassigned()
29380
+ created_by: Optional[shapes.UserContext] = Unassigned()
29351
29381
  last_modified_time: Optional[datetime.datetime] = Unassigned()
29352
- last_modified_by: Optional[UserContext] = Unassigned()
29353
- metadata_properties: Optional[MetadataProperties] = Unassigned()
29382
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
29383
+ metadata_properties: Optional[shapes.MetadataProperties] = Unassigned()
29354
29384
 
29355
29385
  def get_name(self) -> str:
29356
29386
  attributes = vars(self)
@@ -29375,8 +29405,8 @@ class Trial(Base):
29375
29405
  trial_name: str,
29376
29406
  experiment_name: Union[str, object],
29377
29407
  display_name: Optional[str] = Unassigned(),
29378
- metadata_properties: Optional[MetadataProperties] = Unassigned(),
29379
- tags: Optional[List[Tag]] = Unassigned(),
29408
+ metadata_properties: Optional[shapes.MetadataProperties] = Unassigned(),
29409
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
29380
29410
  session: Optional[Session] = None,
29381
29411
  region: Optional[str] = None,
29382
29412
  ) -> Optional["Trial"]:
@@ -29705,21 +29735,21 @@ class TrialComponent(Base):
29705
29735
  trial_component_name: str
29706
29736
  trial_component_arn: Optional[str] = Unassigned()
29707
29737
  display_name: Optional[str] = Unassigned()
29708
- source: Optional[TrialComponentSource] = Unassigned()
29709
- status: Optional[TrialComponentStatus] = Unassigned()
29738
+ source: Optional[shapes.TrialComponentSource] = Unassigned()
29739
+ status: Optional[shapes.TrialComponentStatus] = Unassigned()
29710
29740
  start_time: Optional[datetime.datetime] = Unassigned()
29711
29741
  end_time: Optional[datetime.datetime] = Unassigned()
29712
29742
  creation_time: Optional[datetime.datetime] = Unassigned()
29713
- created_by: Optional[UserContext] = Unassigned()
29743
+ created_by: Optional[shapes.UserContext] = Unassigned()
29714
29744
  last_modified_time: Optional[datetime.datetime] = Unassigned()
29715
- last_modified_by: Optional[UserContext] = Unassigned()
29716
- parameters: Optional[Dict[str, TrialComponentParameterValue]] = Unassigned()
29717
- input_artifacts: Optional[Dict[str, TrialComponentArtifact]] = Unassigned()
29718
- output_artifacts: Optional[Dict[str, TrialComponentArtifact]] = Unassigned()
29719
- metadata_properties: Optional[MetadataProperties] = Unassigned()
29720
- metrics: Optional[List[TrialComponentMetricSummary]] = Unassigned()
29745
+ last_modified_by: Optional[shapes.UserContext] = Unassigned()
29746
+ parameters: Optional[Dict[str, shapes.TrialComponentParameterValue]] = Unassigned()
29747
+ input_artifacts: Optional[Dict[str, shapes.TrialComponentArtifact]] = Unassigned()
29748
+ output_artifacts: Optional[Dict[str, shapes.TrialComponentArtifact]] = Unassigned()
29749
+ metadata_properties: Optional[shapes.MetadataProperties] = Unassigned()
29750
+ metrics: Optional[List[shapes.TrialComponentMetricSummary]] = Unassigned()
29721
29751
  lineage_group_arn: Optional[str] = Unassigned()
29722
- sources: Optional[List[TrialComponentSource]] = Unassigned()
29752
+ sources: Optional[List[shapes.TrialComponentSource]] = Unassigned()
29723
29753
 
29724
29754
  def get_name(self) -> str:
29725
29755
  attributes = vars(self)
@@ -29743,14 +29773,14 @@ class TrialComponent(Base):
29743
29773
  cls,
29744
29774
  trial_component_name: str,
29745
29775
  display_name: Optional[str] = Unassigned(),
29746
- status: Optional[TrialComponentStatus] = Unassigned(),
29776
+ status: Optional[shapes.TrialComponentStatus] = Unassigned(),
29747
29777
  start_time: Optional[datetime.datetime] = Unassigned(),
29748
29778
  end_time: Optional[datetime.datetime] = Unassigned(),
29749
- parameters: Optional[Dict[str, TrialComponentParameterValue]] = Unassigned(),
29750
- input_artifacts: Optional[Dict[str, TrialComponentArtifact]] = Unassigned(),
29751
- output_artifacts: Optional[Dict[str, TrialComponentArtifact]] = Unassigned(),
29752
- metadata_properties: Optional[MetadataProperties] = Unassigned(),
29753
- tags: Optional[List[Tag]] = Unassigned(),
29779
+ parameters: Optional[Dict[str, shapes.TrialComponentParameterValue]] = Unassigned(),
29780
+ input_artifacts: Optional[Dict[str, shapes.TrialComponentArtifact]] = Unassigned(),
29781
+ output_artifacts: Optional[Dict[str, shapes.TrialComponentArtifact]] = Unassigned(),
29782
+ metadata_properties: Optional[shapes.MetadataProperties] = Unassigned(),
29783
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
29754
29784
  session: Optional[Session] = None,
29755
29785
  region: Optional[str] = None,
29756
29786
  ) -> Optional["TrialComponent"]:
@@ -29915,14 +29945,14 @@ class TrialComponent(Base):
29915
29945
  def update(
29916
29946
  self,
29917
29947
  display_name: Optional[str] = Unassigned(),
29918
- status: Optional[TrialComponentStatus] = Unassigned(),
29948
+ status: Optional[shapes.TrialComponentStatus] = Unassigned(),
29919
29949
  start_time: Optional[datetime.datetime] = Unassigned(),
29920
29950
  end_time: Optional[datetime.datetime] = Unassigned(),
29921
- parameters: Optional[Dict[str, TrialComponentParameterValue]] = Unassigned(),
29951
+ parameters: Optional[Dict[str, shapes.TrialComponentParameterValue]] = Unassigned(),
29922
29952
  parameters_to_remove: Optional[List[str]] = Unassigned(),
29923
- input_artifacts: Optional[Dict[str, TrialComponentArtifact]] = Unassigned(),
29953
+ input_artifacts: Optional[Dict[str, shapes.TrialComponentArtifact]] = Unassigned(),
29924
29954
  input_artifacts_to_remove: Optional[List[str]] = Unassigned(),
29925
- output_artifacts: Optional[Dict[str, TrialComponentArtifact]] = Unassigned(),
29955
+ output_artifacts: Optional[Dict[str, shapes.TrialComponentArtifact]] = Unassigned(),
29926
29956
  output_artifacts_to_remove: Optional[List[str]] = Unassigned(),
29927
29957
  ) -> Optional["TrialComponent"]:
29928
29958
  """
@@ -30232,7 +30262,7 @@ class TrialComponent(Base):
30232
30262
  @Base.add_validate_call
30233
30263
  def batch_put_metrics(
30234
30264
  self,
30235
- metric_data: List[RawMetricData],
30265
+ metric_data: List[shapes.RawMetricData],
30236
30266
  session: Optional[Session] = None,
30237
30267
  region: Optional[str] = None,
30238
30268
  ) -> None:
@@ -30276,10 +30306,10 @@ class TrialComponent(Base):
30276
30306
  @Base.add_validate_call
30277
30307
  def batch_get_metrics(
30278
30308
  cls,
30279
- metric_queries: List[MetricQuery],
30309
+ metric_queries: List[shapes.MetricQuery],
30280
30310
  session: Optional[Session] = None,
30281
30311
  region: Optional[str] = None,
30282
- ) -> Optional[BatchGetMetricsResponse]:
30312
+ ) -> Optional[shapes.BatchGetMetricsResponse]:
30283
30313
  """
30284
30314
  Used to retrieve training metrics from SageMaker.
30285
30315
 
@@ -30289,7 +30319,7 @@ class TrialComponent(Base):
30289
30319
  region: Region name.
30290
30320
 
30291
30321
  Returns:
30292
- BatchGetMetricsResponse
30322
+ shapes.BatchGetMetricsResponse
30293
30323
 
30294
30324
  Raises:
30295
30325
  botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
@@ -30319,7 +30349,7 @@ class TrialComponent(Base):
30319
30349
  logger.debug(f"Response: {response}")
30320
30350
 
30321
30351
  transformed_response = transform(response, "BatchGetMetricsResponse")
30322
- return BatchGetMetricsResponse(**transformed_response)
30352
+ return shapes.BatchGetMetricsResponse(**transformed_response)
30323
30353
 
30324
30354
 
30325
30355
  class UserProfile(Base):
@@ -30351,7 +30381,7 @@ class UserProfile(Base):
30351
30381
  failure_reason: Optional[str] = Unassigned()
30352
30382
  single_sign_on_user_identifier: Optional[str] = Unassigned()
30353
30383
  single_sign_on_user_value: Optional[str] = Unassigned()
30354
- user_settings: Optional[UserSettings] = Unassigned()
30384
+ user_settings: Optional[shapes.UserSettings] = Unassigned()
30355
30385
 
30356
30386
  def get_name(self) -> str:
30357
30387
  attributes = vars(self)
@@ -30420,8 +30450,8 @@ class UserProfile(Base):
30420
30450
  user_profile_name: str,
30421
30451
  single_sign_on_user_identifier: Optional[str] = Unassigned(),
30422
30452
  single_sign_on_user_value: Optional[str] = Unassigned(),
30423
- tags: Optional[List[Tag]] = Unassigned(),
30424
- user_settings: Optional[UserSettings] = Unassigned(),
30453
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
30454
+ user_settings: Optional[shapes.UserSettings] = Unassigned(),
30425
30455
  session: Optional[Session] = None,
30426
30456
  region: Optional[str] = None,
30427
30457
  ) -> Optional["UserProfile"]:
@@ -30587,7 +30617,7 @@ class UserProfile(Base):
30587
30617
  @Base.add_validate_call
30588
30618
  def update(
30589
30619
  self,
30590
- user_settings: Optional[UserSettings] = Unassigned(),
30620
+ user_settings: Optional[shapes.UserSettings] = Unassigned(),
30591
30621
  ) -> Optional["UserProfile"]:
30592
30622
  """
30593
30623
  Update a UserProfile resource
@@ -30876,7 +30906,7 @@ class Workforce(Base):
30876
30906
  """
30877
30907
 
30878
30908
  workforce_name: str
30879
- workforce: Optional[Workforce] = Unassigned()
30909
+ workforce: Optional[shapes.Workforce] = Unassigned()
30880
30910
 
30881
30911
  def get_name(self) -> str:
30882
30912
  attributes = vars(self)
@@ -30920,11 +30950,11 @@ class Workforce(Base):
30920
30950
  def create(
30921
30951
  cls,
30922
30952
  workforce_name: str,
30923
- cognito_config: Optional[CognitoConfig] = Unassigned(),
30924
- oidc_config: Optional[OidcConfig] = Unassigned(),
30925
- source_ip_config: Optional[SourceIpConfig] = Unassigned(),
30926
- tags: Optional[List[Tag]] = Unassigned(),
30927
- workforce_vpc_config: Optional[WorkforceVpcConfigRequest] = Unassigned(),
30953
+ cognito_config: Optional[shapes.CognitoConfig] = Unassigned(),
30954
+ oidc_config: Optional[shapes.OidcConfig] = Unassigned(),
30955
+ source_ip_config: Optional[shapes.SourceIpConfig] = Unassigned(),
30956
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
30957
+ workforce_vpc_config: Optional[shapes.WorkforceVpcConfigRequest] = Unassigned(),
30928
30958
  session: Optional[Session] = None,
30929
30959
  region: Optional[str] = None,
30930
30960
  ) -> Optional["Workforce"]:
@@ -31078,9 +31108,9 @@ class Workforce(Base):
31078
31108
  @Base.add_validate_call
31079
31109
  def update(
31080
31110
  self,
31081
- source_ip_config: Optional[SourceIpConfig] = Unassigned(),
31082
- oidc_config: Optional[OidcConfig] = Unassigned(),
31083
- workforce_vpc_config: Optional[WorkforceVpcConfigRequest] = Unassigned(),
31111
+ source_ip_config: Optional[shapes.SourceIpConfig] = Unassigned(),
31112
+ oidc_config: Optional[shapes.OidcConfig] = Unassigned(),
31113
+ workforce_vpc_config: Optional[shapes.WorkforceVpcConfigRequest] = Unassigned(),
31084
31114
  ) -> Optional["Workforce"]:
31085
31115
  """
31086
31116
  Update a Workforce resource
@@ -31347,7 +31377,7 @@ class Workteam(Base):
31347
31377
  """
31348
31378
 
31349
31379
  workteam_name: str
31350
- workteam: Optional[Workteam] = Unassigned()
31380
+ workteam: Optional[shapes.Workteam] = Unassigned()
31351
31381
 
31352
31382
  def get_name(self) -> str:
31353
31383
  attributes = vars(self)
@@ -31370,12 +31400,12 @@ class Workteam(Base):
31370
31400
  def create(
31371
31401
  cls,
31372
31402
  workteam_name: str,
31373
- member_definitions: List[MemberDefinition],
31403
+ member_definitions: List[shapes.MemberDefinition],
31374
31404
  description: str,
31375
31405
  workforce_name: Optional[Union[str, object]] = Unassigned(),
31376
- notification_configuration: Optional[NotificationConfiguration] = Unassigned(),
31377
- worker_access_configuration: Optional[WorkerAccessConfiguration] = Unassigned(),
31378
- tags: Optional[List[Tag]] = Unassigned(),
31406
+ notification_configuration: Optional[shapes.NotificationConfiguration] = Unassigned(),
31407
+ worker_access_configuration: Optional[shapes.WorkerAccessConfiguration] = Unassigned(),
31408
+ tags: Optional[List[shapes.Tag]] = Unassigned(),
31379
31409
  session: Optional[Session] = None,
31380
31410
  region: Optional[str] = None,
31381
31411
  ) -> Optional["Workteam"]:
@@ -31532,10 +31562,10 @@ class Workteam(Base):
31532
31562
  @Base.add_validate_call
31533
31563
  def update(
31534
31564
  self,
31535
- member_definitions: Optional[List[MemberDefinition]] = Unassigned(),
31565
+ member_definitions: Optional[List[shapes.MemberDefinition]] = Unassigned(),
31536
31566
  description: Optional[str] = Unassigned(),
31537
- notification_configuration: Optional[NotificationConfiguration] = Unassigned(),
31538
- worker_access_configuration: Optional[WorkerAccessConfiguration] = Unassigned(),
31567
+ notification_configuration: Optional[shapes.NotificationConfiguration] = Unassigned(),
31568
+ worker_access_configuration: Optional[shapes.WorkerAccessConfiguration] = Unassigned(),
31539
31569
  ) -> Optional["Workteam"]:
31540
31570
  """
31541
31571
  Update a Workteam resource