Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 17 additions & 1 deletion sample/sagemaker/2017-07-24/service-2.json
Original file line number Diff line number Diff line change
Expand Up @@ -42642,7 +42642,15 @@
"ml.inf2.xlarge",
"ml.inf2.8xlarge",
"ml.inf2.24xlarge",
"ml.inf2.48xlarge"
"ml.inf2.48xlarge",
"ml.g6.xlarge",
"ml.g6.2xlarge",
"ml.g6.4xlarge",
"ml.g6.8xlarge",
"ml.g6.12xlarge",
"ml.g6.16xlarge",
"ml.g6.24xlarge",
"ml.g6.48xlarge"
]
},
"TransformInstanceTypes":{
Expand Down Expand Up @@ -43801,6 +43809,10 @@
"shape":"ClusterInstanceGroupsToDelete",
"documentation":"<p>Specify the names of the instance groups to delete. Use a single <code>,</code> as the separator between multiple names.</p>"
},
"NodeProvisioningMode":{
"shape":"ClusterNodeProvisioningMode",
"documentation":"<p>Determines how instance provisioning is handled during cluster operations. In <code>Continuous</code> mode, the cluster provisions available instances incrementally and retries until the target count is reached. The cluster becomes operational once cluster-level resources are ready. Use <code>CurrentCount</code> and <code>TargetCount</code> in <code>DescribeCluster</code> to track provisioning progress.</p>"
},
"ClusterRole":{
"shape":"RoleArn",
"documentation":"<p>The Amazon Resource Name (ARN) of the IAM role that HyperPod assumes for cluster autoscaling operations. Cannot be updated while autoscaling is enabled.</p>"
Expand Down Expand Up @@ -44102,6 +44114,10 @@
"TagPropagation":{
"shape":"TagPropagation",
"documentation":"<p>Indicates whether custom tag propagation is supported for the domain. Defaults to <code>DISABLED</code>.</p>"
},
"VpcId":{
"shape":"VpcId",
"documentation":"<p>The identifier for the VPC used by the domain for network communication. Use this field only when adding VPC configuration to a SageMaker AI domain used in Amazon SageMaker Unified Studio that was created without VPC settings. SageMaker AI doesn't automatically apply VPC updates to existing applications. Stop and restart your applications to apply the changes.</p>"
}
}
},
Expand Down
6 changes: 6 additions & 0 deletions src/sagemaker_core/main/code_injection/shape_dag.py
Original file line number Diff line number Diff line change
Expand Up @@ -15910,6 +15910,11 @@
"shape": "ClusterInstanceGroupsToDelete",
"type": "list",
},
{
"name": "NodeProvisioningMode",
"shape": "ClusterNodeProvisioningMode",
"type": "string",
},
{"name": "ClusterRole", "shape": "RoleArn", "type": "string"},
{"name": "AutoScaling", "shape": "ClusterAutoScalingConfig", "type": "structure"},
],
Expand Down Expand Up @@ -16052,6 +16057,7 @@
{"name": "SubnetIds", "shape": "Subnets", "type": "list"},
{"name": "AppNetworkAccessType", "shape": "AppNetworkAccessType", "type": "string"},
{"name": "TagPropagation", "shape": "TagPropagation", "type": "string"},
{"name": "VpcId", "shape": "VpcId", "type": "string"},
],
"type": "structure",
},
Expand Down
4 changes: 4 additions & 0 deletions src/sagemaker_core/main/resources.py
Original file line number Diff line number Diff line change
Expand Up @@ -3476,6 +3476,7 @@ def update(
tiered_storage_config: Optional[shapes.ClusterTieredStorageConfig] = Unassigned(),
node_recovery: Optional[str] = Unassigned(),
instance_groups_to_delete: Optional[List[str]] = Unassigned(),
node_provisioning_mode: Optional[str] = Unassigned(),
cluster_role: Optional[str] = Unassigned(),
auto_scaling: Optional[shapes.ClusterAutoScalingConfig] = Unassigned(),
) -> Optional["Cluster"]:
Expand Down Expand Up @@ -3513,6 +3514,7 @@ def update(
"TieredStorageConfig": tiered_storage_config,
"NodeRecovery": node_recovery,
"InstanceGroupsToDelete": instance_groups_to_delete,
"NodeProvisioningMode": node_provisioning_mode,
"ClusterRole": cluster_role,
"AutoScaling": auto_scaling,
}
Expand Down Expand Up @@ -7644,6 +7646,7 @@ def update(
subnet_ids: Optional[List[str]] = Unassigned(),
app_network_access_type: Optional[str] = Unassigned(),
tag_propagation: Optional[str] = Unassigned(),
vpc_id: Optional[str] = Unassigned(),
) -> Optional["Domain"]:
"""
Update a Domain resource
Expand Down Expand Up @@ -7681,6 +7684,7 @@ def update(
"SubnetIds": subnet_ids,
"AppNetworkAccessType": app_network_access_type,
"TagPropagation": tag_propagation,
"VpcId": vpc_id,
}
logger.debug(f"Input request: {operation_input_args}")
# serialize the input request
Expand Down
Loading