feat(client-sagemaker): Added new capability in the UpdateCluster ope…

…ration to remove instance groups from your SageMaker HyperPod cluster.
aws · Feb 20, 2025 · 98ed404 · 98ed404
1 parent 4b613e3
commit 98ed404
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 0 deletions.
diff --git a/clients/client-sagemaker/src/commands/UpdateClusterCommand.ts b/clients/client-sagemaker/src/commands/UpdateClusterCommand.ts
@@ -70,6 +70,9 @@ export interface UpdateClusterCommandOutput extends UpdateClusterResponse, __Met
  *     },
  *   ],
  *   NodeRecovery: "Automatic" || "None",
+ *   InstanceGroupsToDelete: [ // ClusterInstanceGroupsToDelete
+ *     "STRING_VALUE",
+ *   ],
  * };
  * const command = new UpdateClusterCommand(input);
  * const response = await client.send(command);

diff --git a/clients/client-sagemaker/src/models/models_5.ts b/clients/client-sagemaker/src/models/models_5.ts
@@ -205,6 +205,13 @@ export interface UpdateClusterRequest {
    * @public
    */
   NodeRecovery?: ClusterNodeRecovery | undefined;
+
+  /**
+   * <p>Specify the names of the instance groups to delete.
+   *       Use a single <code>,</code> as the separator between multiple names.</p>
+   * @public
+   */
+  InstanceGroupsToDelete?: string[] | undefined;
 }
 
 /**

diff --git a/clients/client-sagemaker/src/protocols/Aws_json1_1.ts b/clients/client-sagemaker/src/protocols/Aws_json1_1.ts
@@ -14144,6 +14144,8 @@ const se_BatchTransformInput = (input: BatchTransformInput, context: __SerdeCont
 
 // se_ClusterInstanceGroupSpecifications omitted.
 
+// se_ClusterInstanceGroupsToDelete omitted.
+
 // se_ClusterInstanceStorageConfig omitted.
 
 // se_ClusterInstanceStorageConfigs omitted.

diff --git a/codegen/sdk-codegen/aws-models/sagemaker.json b/codegen/sdk-codegen/aws-models/sagemaker.json
@@ -7076,6 +7076,18 @@
         }
       }
     },
+    "com.amazonaws.sagemaker#ClusterInstanceGroupsToDelete": {
+      "type": "list",
+      "member": {
+        "target": "com.amazonaws.sagemaker#ClusterInstanceGroupName"
+      },
+      "traits": {
+        "smithy.api#length": {
+          "min": 0,
+          "max": 100
+        }
+      }
+    },
     "com.amazonaws.sagemaker#ClusterInstancePlacement": {
       "type": "structure",
       "members": {
@@ -72193,6 +72205,12 @@
           "traits": {
             "smithy.api#documentation": "<p>The node recovery mode to be applied to the SageMaker HyperPod cluster.</p>"
           }
+        },
+        "InstanceGroupsToDelete": {
+          "target": "com.amazonaws.sagemaker#ClusterInstanceGroupsToDelete",
+          "traits": {
+            "smithy.api#documentation": "<p>Specify the names of the instance groups to delete. \n      Use a single <code>,</code> as the separator between multiple names.</p>"
+          }
         }
       },
       "traits": {