sagemaker-core 1.0.48__py3-none-any.whl → 1.0.49__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.

@@ -43,6 +43,22 @@ SHAPE_DAG = {
43
43
  ],
44
44
  "type": "structure",
45
45
  },
46
+ "AddClusterNodeSpecification": {
47
+ "members": [
48
+ {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
49
+ {
50
+ "name": "IncrementTargetCountBy",
51
+ "shape": "AddClusterNodeSpecificationIncrementTargetCountByInteger",
52
+ "type": "integer",
53
+ },
54
+ ],
55
+ "type": "structure",
56
+ },
57
+ "AddClusterNodeSpecificationList": {
58
+ "member_shape": "AddClusterNodeSpecification",
59
+ "member_type": "structure",
60
+ "type": "list",
61
+ },
46
62
  "AddTagsInput": {
47
63
  "members": [
48
64
  {"name": "ResourceArn", "shape": "ResourceArn", "type": "string"},
@@ -59,6 +75,10 @@ SHAPE_DAG = {
59
75
  "member_type": "string",
60
76
  "type": "list",
61
77
  },
78
+ "AdditionalEnis": {
79
+ "members": [{"name": "EfaEnis", "shape": "EfaEnis", "type": "list"}],
80
+ "type": "structure",
81
+ },
62
82
  "AdditionalInferenceSpecificationDefinition": {
63
83
  "members": [
64
84
  {"name": "Name", "shape": "EntityName", "type": "string"},
@@ -782,6 +802,39 @@ SHAPE_DAG = {
782
802
  "members": [{"name": "Mode", "shape": "AutotuneMode", "type": "string"}],
783
803
  "type": "structure",
784
804
  },
805
+ "BatchAddClusterNodesError": {
806
+ "members": [
807
+ {"name": "InstanceGroupName", "shape": "InstanceGroupName", "type": "string"},
808
+ {"name": "ErrorCode", "shape": "BatchAddClusterNodesErrorCode", "type": "string"},
809
+ {"name": "FailedCount", "shape": "BatchAddFailureCount", "type": "integer"},
810
+ {"name": "Message", "shape": "String", "type": "string"},
811
+ ],
812
+ "type": "structure",
813
+ },
814
+ "BatchAddClusterNodesErrorList": {
815
+ "member_shape": "BatchAddClusterNodesError",
816
+ "member_type": "structure",
817
+ "type": "list",
818
+ },
819
+ "BatchAddClusterNodesRequest": {
820
+ "members": [
821
+ {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
822
+ {
823
+ "name": "ClientToken",
824
+ "shape": "BatchAddClusterNodesRequestClientTokenString",
825
+ "type": "string",
826
+ },
827
+ {"name": "NodesToAdd", "shape": "AddClusterNodeSpecificationList", "type": "list"},
828
+ ],
829
+ "type": "structure",
830
+ },
831
+ "BatchAddClusterNodesResponse": {
832
+ "members": [
833
+ {"name": "Successful", "shape": "NodeAdditionResultList", "type": "list"},
834
+ {"name": "Failed", "shape": "BatchAddClusterNodesErrorList", "type": "list"},
835
+ ],
836
+ "type": "structure",
837
+ },
785
838
  "BatchDataCaptureConfig": {
786
839
  "members": [
787
840
  {"name": "DestinationS3Uri", "shape": "S3Uri", "type": "string"},
@@ -790,6 +843,19 @@ SHAPE_DAG = {
790
843
  ],
791
844
  "type": "structure",
792
845
  },
846
+ "BatchDeleteClusterNodeLogicalIdsError": {
847
+ "members": [
848
+ {"name": "Code", "shape": "BatchDeleteClusterNodesErrorCode", "type": "string"},
849
+ {"name": "Message", "shape": "String", "type": "string"},
850
+ {"name": "NodeLogicalId", "shape": "ClusterNodeLogicalId", "type": "string"},
851
+ ],
852
+ "type": "structure",
853
+ },
854
+ "BatchDeleteClusterNodeLogicalIdsErrorList": {
855
+ "member_shape": "BatchDeleteClusterNodeLogicalIdsError",
856
+ "member_type": "structure",
857
+ "type": "list",
858
+ },
793
859
  "BatchDeleteClusterNodesError": {
794
860
  "members": [
795
861
  {"name": "Code", "shape": "BatchDeleteClusterNodesErrorCode", "type": "string"},
@@ -807,6 +873,7 @@ SHAPE_DAG = {
807
873
  "members": [
808
874
  {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
809
875
  {"name": "NodeIds", "shape": "ClusterNodeIds", "type": "list"},
876
+ {"name": "NodeLogicalIds", "shape": "ClusterNodeLogicalIdList", "type": "list"},
810
877
  ],
811
878
  "type": "structure",
812
879
  },
@@ -814,6 +881,16 @@ SHAPE_DAG = {
814
881
  "members": [
815
882
  {"name": "Failed", "shape": "BatchDeleteClusterNodesErrorList", "type": "list"},
816
883
  {"name": "Successful", "shape": "ClusterNodeIds", "type": "list"},
884
+ {
885
+ "name": "FailedNodeLogicalIds",
886
+ "shape": "BatchDeleteClusterNodeLogicalIdsErrorList",
887
+ "type": "list",
888
+ },
889
+ {
890
+ "name": "SuccessfulNodeLogicalIds",
891
+ "shape": "ClusterNodeLogicalIdList",
892
+ "type": "list",
893
+ },
817
894
  ],
818
895
  "type": "structure",
819
896
  },
@@ -1102,6 +1179,13 @@ SHAPE_DAG = {
1102
1179
  ],
1103
1180
  "type": "structure",
1104
1181
  },
1182
+ "CapacityReservation": {
1183
+ "members": [
1184
+ {"name": "Arn", "shape": "String", "type": "string"},
1185
+ {"name": "Type", "shape": "CapacityReservationType", "type": "string"},
1186
+ ],
1187
+ "type": "structure",
1188
+ },
1105
1189
  "CapacitySize": {
1106
1190
  "members": [
1107
1191
  {"name": "Type", "shape": "CapacitySizeType", "type": "string"},
@@ -1367,6 +1451,38 @@ SHAPE_DAG = {
1367
1451
  ],
1368
1452
  "type": "structure",
1369
1453
  },
1454
+ "ClusterEventDetail": {
1455
+ "members": [
1456
+ {"name": "EventId", "shape": "EventId", "type": "string"},
1457
+ {"name": "ClusterArn", "shape": "ClusterArn", "type": "string"},
1458
+ {"name": "ClusterName", "shape": "ClusterName", "type": "string"},
1459
+ {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
1460
+ {"name": "InstanceId", "shape": "String", "type": "string"},
1461
+ {"name": "ResourceType", "shape": "ClusterEventResourceType", "type": "string"},
1462
+ {"name": "EventTime", "shape": "Timestamp", "type": "timestamp"},
1463
+ {"name": "EventDetails", "shape": "EventDetails", "type": "structure"},
1464
+ {"name": "Description", "shape": "String", "type": "string"},
1465
+ ],
1466
+ "type": "structure",
1467
+ },
1468
+ "ClusterEventSummaries": {
1469
+ "member_shape": "ClusterEventSummary",
1470
+ "member_type": "structure",
1471
+ "type": "list",
1472
+ },
1473
+ "ClusterEventSummary": {
1474
+ "members": [
1475
+ {"name": "EventId", "shape": "EventId", "type": "string"},
1476
+ {"name": "ClusterArn", "shape": "ClusterArn", "type": "string"},
1477
+ {"name": "ClusterName", "shape": "ClusterName", "type": "string"},
1478
+ {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
1479
+ {"name": "InstanceId", "shape": "String", "type": "string"},
1480
+ {"name": "ResourceType", "shape": "ClusterEventResourceType", "type": "string"},
1481
+ {"name": "EventTime", "shape": "Timestamp", "type": "timestamp"},
1482
+ {"name": "Description", "shape": "String", "type": "string"},
1483
+ ],
1484
+ "type": "structure",
1485
+ },
1370
1486
  "ClusterInstanceGroupDetails": {
1371
1487
  "members": [
1372
1488
  {"name": "CurrentCount", "shape": "ClusterNonNegativeInstanceCount", "type": "integer"},
@@ -1395,6 +1511,8 @@ SHAPE_DAG = {
1395
1511
  "shape": "ScheduledUpdateConfig",
1396
1512
  "type": "structure",
1397
1513
  },
1514
+ {"name": "CurrentImageId", "shape": "ImageId", "type": "string"},
1515
+ {"name": "DesiredImageId", "shape": "ImageId", "type": "string"},
1398
1516
  ],
1399
1517
  "type": "structure",
1400
1518
  },
@@ -1424,6 +1542,7 @@ SHAPE_DAG = {
1424
1542
  "shape": "ScheduledUpdateConfig",
1425
1543
  "type": "structure",
1426
1544
  },
1545
+ {"name": "ImageId", "shape": "ImageId", "type": "string"},
1427
1546
  ],
1428
1547
  "type": "structure",
1429
1548
  },
@@ -1469,10 +1588,19 @@ SHAPE_DAG = {
1469
1588
  ],
1470
1589
  "type": "structure",
1471
1590
  },
1591
+ "ClusterMetadata": {
1592
+ "members": [
1593
+ {"name": "FailureMessage", "shape": "String", "type": "string"},
1594
+ {"name": "EksRoleAccessEntries", "shape": "EksRoleAccessEntries", "type": "list"},
1595
+ {"name": "SlrAccessEntry", "shape": "String", "type": "string"},
1596
+ ],
1597
+ "type": "structure",
1598
+ },
1472
1599
  "ClusterNodeDetails": {
1473
1600
  "members": [
1474
1601
  {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
1475
1602
  {"name": "InstanceId", "shape": "String", "type": "string"},
1603
+ {"name": "NodeLogicalId", "shape": "ClusterNodeLogicalId", "type": "string"},
1476
1604
  {
1477
1605
  "name": "InstanceStatus",
1478
1606
  "shape": "ClusterInstanceStatusDetails",
@@ -1493,10 +1621,17 @@ SHAPE_DAG = {
1493
1621
  {"name": "PrivatePrimaryIpv6", "shape": "ClusterPrivatePrimaryIpv6", "type": "string"},
1494
1622
  {"name": "PrivateDnsHostname", "shape": "ClusterPrivateDnsHostname", "type": "string"},
1495
1623
  {"name": "Placement", "shape": "ClusterInstancePlacement", "type": "structure"},
1624
+ {"name": "CurrentImageId", "shape": "ImageId", "type": "string"},
1625
+ {"name": "DesiredImageId", "shape": "ImageId", "type": "string"},
1496
1626
  ],
1497
1627
  "type": "structure",
1498
1628
  },
1499
1629
  "ClusterNodeIds": {"member_shape": "ClusterNodeId", "member_type": "string", "type": "list"},
1630
+ "ClusterNodeLogicalIdList": {
1631
+ "member_shape": "ClusterNodeLogicalId",
1632
+ "member_type": "string",
1633
+ "type": "list",
1634
+ },
1500
1635
  "ClusterNodeSummaries": {
1501
1636
  "member_shape": "ClusterNodeSummary",
1502
1637
  "member_type": "structure",
@@ -1506,6 +1641,7 @@ SHAPE_DAG = {
1506
1641
  "members": [
1507
1642
  {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
1508
1643
  {"name": "InstanceId", "shape": "String", "type": "string"},
1644
+ {"name": "NodeLogicalId", "shape": "String", "type": "string"},
1509
1645
  {"name": "InstanceType", "shape": "ClusterInstanceType", "type": "string"},
1510
1646
  {"name": "LaunchTime", "shape": "Timestamp", "type": "timestamp"},
1511
1647
  {"name": "LastSoftwareUpdateTime", "shape": "Timestamp", "type": "timestamp"},
@@ -2103,6 +2239,11 @@ SHAPE_DAG = {
2103
2239
  {"name": "Tags", "shape": "TagList", "type": "list"},
2104
2240
  {"name": "Orchestrator", "shape": "ClusterOrchestrator", "type": "structure"},
2105
2241
  {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
2242
+ {
2243
+ "name": "NodeProvisioningMode",
2244
+ "shape": "ClusterNodeProvisioningMode",
2245
+ "type": "string",
2246
+ },
2106
2247
  ],
2107
2248
  "type": "structure",
2108
2249
  },
@@ -4441,10 +4582,22 @@ SHAPE_DAG = {
4441
4582
  ],
4442
4583
  "type": "structure",
4443
4584
  },
4585
+ "DescribeClusterEventRequest": {
4586
+ "members": [
4587
+ {"name": "EventId", "shape": "EventId", "type": "string"},
4588
+ {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
4589
+ ],
4590
+ "type": "structure",
4591
+ },
4592
+ "DescribeClusterEventResponse": {
4593
+ "members": [{"name": "EventDetails", "shape": "ClusterEventDetail", "type": "structure"}],
4594
+ "type": "structure",
4595
+ },
4444
4596
  "DescribeClusterNodeRequest": {
4445
4597
  "members": [
4446
4598
  {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
4447
4599
  {"name": "NodeId", "shape": "ClusterNodeId", "type": "string"},
4600
+ {"name": "NodeLogicalId", "shape": "ClusterNodeLogicalId", "type": "string"},
4448
4601
  ],
4449
4602
  "type": "structure",
4450
4603
  },
@@ -4472,6 +4625,11 @@ SHAPE_DAG = {
4472
4625
  {"name": "VpcConfig", "shape": "VpcConfig", "type": "structure"},
4473
4626
  {"name": "Orchestrator", "shape": "ClusterOrchestrator", "type": "structure"},
4474
4627
  {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
4628
+ {
4629
+ "name": "NodeProvisioningMode",
4630
+ "shape": "ClusterNodeProvisioningMode",
4631
+ "type": "string",
4632
+ },
4475
4633
  ],
4476
4634
  "type": "structure",
4477
4635
  },
@@ -6693,6 +6851,8 @@ SHAPE_DAG = {
6693
6851
  "type": "structure",
6694
6852
  },
6695
6853
  "Edges": {"member_shape": "Edge", "member_type": "structure", "type": "list"},
6854
+ "EfaEnis": {"member_shape": "String", "member_type": "string", "type": "list"},
6855
+ "EksRoleAccessEntries": {"member_shape": "String", "member_type": "string", "type": "list"},
6696
6856
  "EmrServerlessComputeConfig": {
6697
6857
  "members": [{"name": "ExecutionRoleARN", "shape": "RoleArn", "type": "string"}],
6698
6858
  "type": "structure",
@@ -6908,6 +7068,23 @@ SHAPE_DAG = {
6908
7068
  ],
6909
7069
  "type": "structure",
6910
7070
  },
7071
+ "EventDetails": {
7072
+ "members": [{"name": "EventMetadata", "shape": "EventMetadata", "type": "structure"}],
7073
+ "type": "structure",
7074
+ },
7075
+ "EventMetadata": {
7076
+ "members": [
7077
+ {"name": "Cluster", "shape": "ClusterMetadata", "type": "structure"},
7078
+ {"name": "InstanceGroup", "shape": "InstanceGroupMetadata", "type": "structure"},
7079
+ {
7080
+ "name": "InstanceGroupScaling",
7081
+ "shape": "InstanceGroupScalingMetadata",
7082
+ "type": "structure",
7083
+ },
7084
+ {"name": "Instance", "shape": "InstanceMetadata", "type": "structure"},
7085
+ ],
7086
+ "type": "structure",
7087
+ },
6911
7088
  "ExecutionRoleArns": {"member_shape": "RoleArn", "member_type": "string", "type": "list"},
6912
7089
  "Experiment": {
6913
7090
  "members": [
@@ -8366,12 +8543,42 @@ SHAPE_DAG = {
8366
8543
  ],
8367
8544
  "type": "structure",
8368
8545
  },
8546
+ "InstanceGroupMetadata": {
8547
+ "members": [
8548
+ {"name": "FailureMessage", "shape": "String", "type": "string"},
8549
+ {"name": "AvailabilityZoneId", "shape": "String", "type": "string"},
8550
+ {"name": "CapacityReservation", "shape": "CapacityReservation", "type": "structure"},
8551
+ {"name": "SubnetId", "shape": "String", "type": "string"},
8552
+ {"name": "SecurityGroupIds", "shape": "SecurityGroupIds", "type": "list"},
8553
+ {"name": "AmiOverride", "shape": "String", "type": "string"},
8554
+ ],
8555
+ "type": "structure",
8556
+ },
8369
8557
  "InstanceGroupNames": {
8370
8558
  "member_shape": "InstanceGroupName",
8371
8559
  "member_type": "string",
8372
8560
  "type": "list",
8373
8561
  },
8562
+ "InstanceGroupScalingMetadata": {
8563
+ "members": [
8564
+ {"name": "InstanceCount", "shape": "InstanceCount", "type": "integer"},
8565
+ {"name": "TargetCount", "shape": "TargetCount", "type": "integer"},
8566
+ {"name": "FailureMessage", "shape": "String", "type": "string"},
8567
+ ],
8568
+ "type": "structure",
8569
+ },
8374
8570
  "InstanceGroups": {"member_shape": "InstanceGroup", "member_type": "structure", "type": "list"},
8571
+ "InstanceMetadata": {
8572
+ "members": [
8573
+ {"name": "CustomerEni", "shape": "String", "type": "string"},
8574
+ {"name": "AdditionalEnis", "shape": "AdditionalEnis", "type": "structure"},
8575
+ {"name": "CapacityReservation", "shape": "CapacityReservation", "type": "structure"},
8576
+ {"name": "FailureMessage", "shape": "String", "type": "string"},
8577
+ {"name": "LcsExecutionState", "shape": "String", "type": "string"},
8578
+ {"name": "NodeLogicalId", "shape": "ClusterNodeLogicalId", "type": "string"},
8579
+ ],
8580
+ "type": "structure",
8581
+ },
8375
8582
  "InstanceMetadataServiceConfiguration": {
8376
8583
  "members": [
8377
8584
  {
@@ -8943,6 +9150,28 @@ SHAPE_DAG = {
8943
9150
  ],
8944
9151
  "type": "structure",
8945
9152
  },
9153
+ "ListClusterEventsRequest": {
9154
+ "members": [
9155
+ {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
9156
+ {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
9157
+ {"name": "NodeId", "shape": "ClusterNodeId", "type": "string"},
9158
+ {"name": "EventTimeAfter", "shape": "Timestamp", "type": "timestamp"},
9159
+ {"name": "EventTimeBefore", "shape": "Timestamp", "type": "timestamp"},
9160
+ {"name": "SortBy", "shape": "EventSortBy", "type": "string"},
9161
+ {"name": "SortOrder", "shape": "SortOrder", "type": "string"},
9162
+ {"name": "ResourceType", "shape": "ClusterEventResourceType", "type": "string"},
9163
+ {"name": "MaxResults", "shape": "ClusterEventMaxResults", "type": "integer"},
9164
+ {"name": "NextToken", "shape": "NextToken", "type": "string"},
9165
+ ],
9166
+ "type": "structure",
9167
+ },
9168
+ "ListClusterEventsResponse": {
9169
+ "members": [
9170
+ {"name": "NextToken", "shape": "NextToken", "type": "string"},
9171
+ {"name": "Events", "shape": "ClusterEventSummaries", "type": "list"},
9172
+ ],
9173
+ "type": "structure",
9174
+ },
8946
9175
  "ListClusterNodesRequest": {
8947
9176
  "members": [
8948
9177
  {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
@@ -8957,6 +9186,11 @@ SHAPE_DAG = {
8957
9186
  {"name": "NextToken", "shape": "NextToken", "type": "string"},
8958
9187
  {"name": "SortBy", "shape": "ClusterSortBy", "type": "string"},
8959
9188
  {"name": "SortOrder", "shape": "SortOrder", "type": "string"},
9189
+ {
9190
+ "name": "IncludeNodeLogicalIds",
9191
+ "shape": "IncludeNodeLogicalIdsBoolean",
9192
+ "type": "boolean",
9193
+ },
8960
9194
  ],
8961
9195
  "type": "structure",
8962
9196
  },
@@ -11803,6 +12037,19 @@ SHAPE_DAG = {
11803
12037
  ],
11804
12038
  "type": "structure",
11805
12039
  },
12040
+ "NodeAdditionResult": {
12041
+ "members": [
12042
+ {"name": "NodeLogicalId", "shape": "ClusterNodeLogicalId", "type": "string"},
12043
+ {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
12044
+ {"name": "Status", "shape": "ClusterInstanceStatus", "type": "string"},
12045
+ ],
12046
+ "type": "structure",
12047
+ },
12048
+ "NodeAdditionResultList": {
12049
+ "member_shape": "NodeAdditionResult",
12050
+ "member_type": "structure",
12051
+ "type": "list",
12052
+ },
11806
12053
  "NotebookInstanceAcceleratorTypes": {
11807
12054
  "member_shape": "NotebookInstanceAcceleratorType",
11808
12055
  "member_type": "string",
@@ -15457,6 +15704,7 @@ SHAPE_DAG = {
15457
15704
  "type": "list",
15458
15705
  },
15459
15706
  {"name": "DeploymentConfig", "shape": "DeploymentConfiguration", "type": "structure"},
15707
+ {"name": "ImageId", "shape": "ImageId", "type": "string"},
15460
15708
  ],
15461
15709
  "type": "structure",
15462
15710
  },
@@ -3223,6 +3223,7 @@ class Cluster(Base):
3223
3223
  vpc_config:
3224
3224
  orchestrator: The type of orchestrator used for the SageMaker HyperPod cluster.
3225
3225
  node_recovery: The node recovery mode configured for the SageMaker HyperPod cluster.
3226
+ node_provisioning_mode: The mode used for provisioning nodes in the cluster.
3226
3227
 
3227
3228
  """
3228
3229
 
@@ -3238,6 +3239,7 @@ class Cluster(Base):
3238
3239
  vpc_config: Optional[shapes.VpcConfig] = Unassigned()
3239
3240
  orchestrator: Optional[shapes.ClusterOrchestrator] = Unassigned()
3240
3241
  node_recovery: Optional[str] = Unassigned()
3242
+ node_provisioning_mode: Optional[str] = Unassigned()
3241
3243
 
3242
3244
  def get_name(self) -> str:
3243
3245
  attributes = vars(self)
@@ -3287,6 +3289,7 @@ class Cluster(Base):
3287
3289
  tags: Optional[List[shapes.Tag]] = Unassigned(),
3288
3290
  orchestrator: Optional[shapes.ClusterOrchestrator] = Unassigned(),
3289
3291
  node_recovery: Optional[str] = Unassigned(),
3292
+ node_provisioning_mode: Optional[str] = Unassigned(),
3290
3293
  session: Optional[Session] = None,
3291
3294
  region: Optional[str] = None,
3292
3295
  ) -> Optional["Cluster"]:
@@ -3301,6 +3304,7 @@ class Cluster(Base):
3301
3304
  tags: Custom tags for managing the SageMaker HyperPod cluster as an Amazon Web Services resource. You can add tags to your cluster in the same way you add them in other Amazon Web Services services that support tagging. To learn more about tagging Amazon Web Services resources in general, see Tagging Amazon Web Services Resources User Guide.
3302
3305
  orchestrator: The type of orchestrator to use for the SageMaker HyperPod cluster. Currently, the only supported value is "eks", which is to use an Amazon Elastic Kubernetes Service (EKS) cluster as the orchestrator.
3303
3306
  node_recovery: The node recovery mode for the SageMaker HyperPod cluster. When set to Automatic, SageMaker HyperPod will automatically reboot or replace faulty nodes when issues are detected. When set to None, cluster administrators will need to manually manage any faulty cluster instances.
3307
+ node_provisioning_mode: The mode for provisioning nodes in the cluster. You can specify the following modes: Continuous: Scaling behavior that enables 1) concurrent operation execution within instance groups, 2) continuous retry mechanisms for failed operations, 3) enhanced customer visibility into cluster events through detailed event streams, 4) partial provisioning capabilities. Your clusters and instance groups remain InService while scaling. This mode is only supported for EKS orchestrated clusters.
3304
3308
  session: Boto3 session.
3305
3309
  region: Region name.
3306
3310
 
@@ -3337,6 +3341,7 @@ class Cluster(Base):
3337
3341
  "Tags": tags,
3338
3342
  "Orchestrator": orchestrator,
3339
3343
  "NodeRecovery": node_recovery,
3344
+ "NodeProvisioningMode": node_provisioning_mode,
3340
3345
  }
3341
3346
 
3342
3347
  operation_input_args = Base.populate_chained_attributes(
@@ -3731,6 +3736,7 @@ class Cluster(Base):
3731
3736
  def get_node(
3732
3737
  self,
3733
3738
  node_id: Optional[str] = Unassigned(),
3739
+ node_logical_id: Optional[str] = Unassigned(),
3734
3740
  session: Optional[Session] = None,
3735
3741
  region: Optional[str] = None,
3736
3742
  ) -> Optional[shapes.ClusterNodeDetails]:
@@ -3739,6 +3745,7 @@ class Cluster(Base):
3739
3745
 
3740
3746
  Parameters:
3741
3747
  node_id: The ID of the SageMaker HyperPod cluster node.
3748
+ node_logical_id: The logical identifier of the node to describe. You can specify either NodeLogicalId or InstanceId, but not both. NodeLogicalId can be used to describe nodes that are still being provisioned and don't yet have an InstanceId assigned.
3742
3749
  session: Boto3 session.
3743
3750
  region: Region name.
3744
3751
 
@@ -3761,6 +3768,7 @@ class Cluster(Base):
3761
3768
  operation_input_args = {
3762
3769
  "ClusterName": self.cluster_name,
3763
3770
  "NodeId": node_id,
3771
+ "NodeLogicalId": node_logical_id,
3764
3772
  }
3765
3773
  # serialize the input request
3766
3774
  operation_input_args = serialize(operation_input_args)
@@ -3785,6 +3793,7 @@ class Cluster(Base):
3785
3793
  instance_group_name_contains: Optional[str] = Unassigned(),
3786
3794
  sort_by: Optional[str] = Unassigned(),
3787
3795
  sort_order: Optional[str] = Unassigned(),
3796
+ include_node_logical_ids: Optional[bool] = Unassigned(),
3788
3797
  session: Optional[Session] = None,
3789
3798
  region: Optional[str] = None,
3790
3799
  ) -> ResourceIterator[shapes.ClusterNodeDetails]:
@@ -3799,6 +3808,7 @@ class Cluster(Base):
3799
3808
  next_token: If the result of the previous ListClusterNodes request was truncated, the response includes a NextToken. To retrieve the next set of cluster nodes, use the token in the next request.
3800
3809
  sort_by: The field by which to sort results. The default value is CREATION_TIME.
3801
3810
  sort_order: The sort order for results. The default value is Ascending.
3811
+ include_node_logical_ids: Specifies whether to include nodes that are still being provisioned in the response. When set to true, the response includes all nodes regardless of their provisioning status. When set to False (default), only nodes with assigned InstanceIds are returned.
3802
3812
  session: Boto3 session.
3803
3813
  region: Region name.
3804
3814
 
@@ -3825,6 +3835,7 @@ class Cluster(Base):
3825
3835
  "InstanceGroupNameContains": instance_group_name_contains,
3826
3836
  "SortBy": sort_by,
3827
3837
  "SortOrder": sort_order,
3838
+ "IncludeNodeLogicalIds": include_node_logical_ids,
3828
3839
  }
3829
3840
  # serialize the input request
3830
3841
  operation_input_args = serialize(operation_input_args)
@@ -3847,6 +3858,7 @@ class Cluster(Base):
3847
3858
  def update_software(
3848
3859
  self,
3849
3860
  deployment_config: Optional[shapes.DeploymentConfiguration] = Unassigned(),
3861
+ image_id: Optional[str] = Unassigned(),
3850
3862
  session: Optional[Session] = None,
3851
3863
  region: Optional[str] = None,
3852
3864
  ) -> None:
@@ -3855,6 +3867,7 @@ class Cluster(Base):
3855
3867
 
3856
3868
  Parameters:
3857
3869
  deployment_config: The configuration to use when updating the AMI versions.
3870
+ image_id: When configuring your HyperPod cluster, you can specify an image ID using one of the following options: HyperPodPublicAmiId: Use a HyperPod public AMI CustomAmiId: Use your custom AMI default: Use the default latest system image f you choose to use a custom AMI (CustomAmiId), ensure it meets the following requirements: Encryption: The custom AMI must be unencrypted. Ownership: The custom AMI must be owned by the same Amazon Web Services account that is creating the HyperPod cluster. Volume support: Only the primary AMI snapshot volume is supported; additional AMI volumes are not supported. When updating the instance group's AMI through the UpdateClusterSoftware operation, if an instance group uses a custom AMI, you must provide an ImageId or use the default as input.
3858
3871
  session: Boto3 session.
3859
3872
  region: Region name.
3860
3873
 
@@ -3876,6 +3889,7 @@ class Cluster(Base):
3876
3889
  "ClusterName": self.cluster_name,
3877
3890
  "InstanceGroups": self.instance_groups,
3878
3891
  "DeploymentConfig": deployment_config,
3892
+ "ImageId": image_id,
3879
3893
  }
3880
3894
  # serialize the input request
3881
3895
  operation_input_args = serialize(operation_input_args)
@@ -3893,6 +3907,7 @@ class Cluster(Base):
3893
3907
  def batch_delete_nodes(
3894
3908
  self,
3895
3909
  node_ids: Optional[List[str]] = Unassigned(),
3910
+ node_logical_ids: Optional[List[str]] = Unassigned(),
3896
3911
  session: Optional[Session] = None,
3897
3912
  region: Optional[str] = None,
3898
3913
  ) -> Optional[shapes.BatchDeleteClusterNodesResponse]:
@@ -3901,6 +3916,7 @@ class Cluster(Base):
3901
3916
 
3902
3917
  Parameters:
3903
3918
  node_ids: A list of node IDs to be deleted from the specified cluster. For SageMaker HyperPod clusters using the Slurm workload manager, you cannot remove instances that are configured as Slurm controller nodes. If you need to delete more than 99 instances, contact Support for assistance.
3919
+ node_logical_ids: A list of NodeLogicalIds identifying the nodes to be deleted. You can specify up to 50 NodeLogicalIds. You must specify either NodeLogicalIds, InstanceIds, or both, with a combined maximum of 50 identifiers.
3904
3920
  session: Boto3 session.
3905
3921
  region: Region name.
3906
3922
 
@@ -3923,6 +3939,7 @@ class Cluster(Base):
3923
3939
  operation_input_args = {
3924
3940
  "ClusterName": self.cluster_name,
3925
3941
  "NodeIds": node_ids,
3942
+ "NodeLogicalIds": node_logical_ids,
3926
3943
  }
3927
3944
  # serialize the input request
3928
3945
  operation_input_args = serialize(operation_input_args)
@@ -494,6 +494,21 @@ class ActionSummary(Base):
494
494
  last_modified_time: Optional[datetime.datetime] = Unassigned()
495
495
 
496
496
 
497
+ class AddClusterNodeSpecification(Base):
498
+ """
499
+ AddClusterNodeSpecification
500
+ Specifies an instance group and the number of nodes to add to it.
501
+
502
+ Attributes
503
+ ----------------------
504
+ instance_group_name: The name of the instance group to which you want to add nodes.
505
+ increment_target_count_by: The number of nodes to add to the specified instance group. The total number of nodes across all instance groups in a single request cannot exceed 50.
506
+ """
507
+
508
+ instance_group_name: str
509
+ increment_target_count_by: int
510
+
511
+
497
512
  class Tag(Base):
498
513
  """
499
514
  Tag
@@ -509,6 +524,19 @@ class Tag(Base):
509
524
  value: str
510
525
 
511
526
 
527
+ class AdditionalEnis(Base):
528
+ """
529
+ AdditionalEnis
530
+ Information about additional Elastic Network Interfaces (ENIs) associated with an instance.
531
+
532
+ Attributes
533
+ ----------------------
534
+ efa_enis: A list of Elastic Fabric Adapter (EFA) ENIs associated with the instance.
535
+ """
536
+
537
+ efa_enis: Optional[List[str]] = Unassigned()
538
+
539
+
512
540
  class ModelAccessConfig(Base):
513
541
  """
514
542
  ModelAccessConfig
@@ -2400,6 +2428,42 @@ class Autotune(Base):
2400
2428
  mode: str
2401
2429
 
2402
2430
 
2431
+ class BatchAddClusterNodesError(Base):
2432
+ """
2433
+ BatchAddClusterNodesError
2434
+ Information about an error that occurred during the node addition operation.
2435
+
2436
+ Attributes
2437
+ ----------------------
2438
+ instance_group_name: The name of the instance group for which the error occurred.
2439
+ error_code: The error code associated with the failure. Possible values include InstanceGroupNotFound and InvalidInstanceGroupState.
2440
+ failed_count: The number of nodes that failed to be added to the specified instance group.
2441
+ message: A descriptive message providing additional details about the error.
2442
+ """
2443
+
2444
+ instance_group_name: str
2445
+ error_code: str
2446
+ failed_count: int
2447
+ message: Optional[str] = Unassigned()
2448
+
2449
+
2450
+ class NodeAdditionResult(Base):
2451
+ """
2452
+ NodeAdditionResult
2453
+ Information about a node that was successfully added to the cluster.
2454
+
2455
+ Attributes
2456
+ ----------------------
2457
+ node_logical_id: A unique identifier assigned to the node that can be used to track its provisioning status through the DescribeClusterNode operation.
2458
+ instance_group_name: The name of the instance group to which the node was added.
2459
+ status: The current status of the node. Possible values include Pending, Running, Failed, ShuttingDown, SystemUpdating, DeepHealthCheckInProgress, and NotFound.
2460
+ """
2461
+
2462
+ node_logical_id: str
2463
+ instance_group_name: str
2464
+ status: str
2465
+
2466
+
2403
2467
  class BatchDataCaptureConfig(Base):
2404
2468
  """
2405
2469
  BatchDataCaptureConfig
@@ -2417,6 +2481,23 @@ class BatchDataCaptureConfig(Base):
2417
2481
  generate_inference_id: Optional[bool] = Unassigned()
2418
2482
 
2419
2483
 
2484
+ class BatchDeleteClusterNodeLogicalIdsError(Base):
2485
+ """
2486
+ BatchDeleteClusterNodeLogicalIdsError
2487
+ Information about an error that occurred when attempting to delete a node identified by its NodeLogicalId.
2488
+
2489
+ Attributes
2490
+ ----------------------
2491
+ code: The error code associated with the failure. Possible values include NodeLogicalIdNotFound, InvalidNodeStatus, and InternalError.
2492
+ message: A descriptive message providing additional details about the error.
2493
+ node_logical_id: The NodeLogicalId of the node that could not be deleted.
2494
+ """
2495
+
2496
+ code: str
2497
+ message: str
2498
+ node_logical_id: str
2499
+
2500
+
2420
2501
  class BatchDeleteClusterNodesError(Base):
2421
2502
  """
2422
2503
  BatchDeleteClusterNodesError
@@ -2442,10 +2523,14 @@ class BatchDeleteClusterNodesResponse(Base):
2442
2523
  ----------------------
2443
2524
  failed: A list of errors encountered when deleting the specified nodes.
2444
2525
  successful: A list of node IDs that were successfully deleted from the specified cluster.
2526
+ failed_node_logical_ids: A list of NodeLogicalIds that could not be deleted, along with error information explaining why the deletion failed.
2527
+ successful_node_logical_ids: A list of NodeLogicalIds that were successfully deleted from the cluster.
2445
2528
  """
2446
2529
 
2447
2530
  failed: Optional[List[BatchDeleteClusterNodesError]] = Unassigned()
2448
2531
  successful: Optional[List[str]] = Unassigned()
2532
+ failed_node_logical_ids: Optional[List[BatchDeleteClusterNodeLogicalIdsError]] = Unassigned()
2533
+ successful_node_logical_ids: Optional[List[str]] = Unassigned()
2449
2534
 
2450
2535
 
2451
2536
  class BatchDescribeModelPackageError(Base):
@@ -2901,6 +2986,21 @@ class CanvasAppSettings(Base):
2901
2986
  emr_serverless_settings: Optional[EmrServerlessSettings] = Unassigned()
2902
2987
 
2903
2988
 
2989
+ class CapacityReservation(Base):
2990
+ """
2991
+ CapacityReservation
2992
+ Information about the Capacity Reservation used by an instance or instance group.
2993
+
2994
+ Attributes
2995
+ ----------------------
2996
+ arn: The Amazon Resource Name (ARN) of the Capacity Reservation.
2997
+ type: The type of Capacity Reservation. Valid values are ODCR (On-Demand Capacity Reservation) or CRG (Capacity Reservation Group).
2998
+ """
2999
+
3000
+ arn: Optional[str] = Unassigned()
3001
+ type: Optional[str] = Unassigned()
3002
+
3003
+
2904
3004
  class CapacitySizeConfig(Base):
2905
3005
  """
2906
3006
  CapacitySizeConfig
@@ -3287,6 +3387,174 @@ class ClusterEbsVolumeConfig(Base):
3287
3387
  volume_size_in_gb: Optional[int] = Unassigned()
3288
3388
 
3289
3389
 
3390
+ class ClusterMetadata(Base):
3391
+ """
3392
+ ClusterMetadata
3393
+ Metadata information about a SageMaker HyperPod cluster showing information about the cluster level operations, such as creating, updating, and deleting.
3394
+
3395
+ Attributes
3396
+ ----------------------
3397
+ failure_message: An error message describing why the cluster level operation (such as creating, updating, or deleting) failed.
3398
+ eks_role_access_entries: A list of Amazon EKS IAM role ARNs associated with the cluster. This is created by SageMaker HyperPod on your behalf and only applies for EKS-orchestrated clusters.
3399
+ slr_access_entry: The Service-Linked Role (SLR) associated with the cluster. This is created by SageMaker HyperPod on your behalf and only applies for EKS-orchestrated clusters.
3400
+ """
3401
+
3402
+ failure_message: Optional[str] = Unassigned()
3403
+ eks_role_access_entries: Optional[List[str]] = Unassigned()
3404
+ slr_access_entry: Optional[str] = Unassigned()
3405
+
3406
+
3407
+ class InstanceGroupMetadata(Base):
3408
+ """
3409
+ InstanceGroupMetadata
3410
+ Metadata information about an instance group in a SageMaker HyperPod cluster.
3411
+
3412
+ Attributes
3413
+ ----------------------
3414
+ failure_message: An error message describing why the instance group level operation (such as creating, scaling, or deleting) failed.
3415
+ availability_zone_id: The ID of the Availability Zone where the instance group is located.
3416
+ capacity_reservation: Information about the Capacity Reservation used by the instance group.
3417
+ subnet_id: The ID of the subnet where the instance group is located.
3418
+ security_group_ids: A list of security group IDs associated with the instance group.
3419
+ ami_override: If you use a custom Amazon Machine Image (AMI) for the instance group, this field shows the ID of the custom AMI.
3420
+ """
3421
+
3422
+ failure_message: Optional[str] = Unassigned()
3423
+ availability_zone_id: Optional[str] = Unassigned()
3424
+ capacity_reservation: Optional[CapacityReservation] = Unassigned()
3425
+ subnet_id: Optional[str] = Unassigned()
3426
+ security_group_ids: Optional[List[str]] = Unassigned()
3427
+ ami_override: Optional[str] = Unassigned()
3428
+
3429
+
3430
+ class InstanceGroupScalingMetadata(Base):
3431
+ """
3432
+ InstanceGroupScalingMetadata
3433
+ Metadata information about scaling operations for an instance group.
3434
+
3435
+ Attributes
3436
+ ----------------------
3437
+ instance_count: The current number of instances in the group.
3438
+ target_count: The desired number of instances for the group after scaling.
3439
+ failure_message: An error message describing why the scaling operation failed, if applicable.
3440
+ """
3441
+
3442
+ instance_count: Optional[int] = Unassigned()
3443
+ target_count: Optional[int] = Unassigned()
3444
+ failure_message: Optional[str] = Unassigned()
3445
+
3446
+
3447
+ class InstanceMetadata(Base):
3448
+ """
3449
+ InstanceMetadata
3450
+ Metadata information about an instance in a HyperPod cluster.
3451
+
3452
+ Attributes
3453
+ ----------------------
3454
+ customer_eni: The ID of the customer-managed Elastic Network Interface (ENI) associated with the instance.
3455
+ additional_enis: Information about additional Elastic Network Interfaces (ENIs) associated with the instance.
3456
+ capacity_reservation: Information about the Capacity Reservation used by the instance.
3457
+ failure_message: An error message describing why the instance creation or update failed, if applicable.
3458
+ lcs_execution_state: The execution state of the Lifecycle Script (LCS) for the instance.
3459
+ node_logical_id: The unique logical identifier of the node within the cluster. The ID used here is the same object as in the BatchAddClusterNodes API.
3460
+ """
3461
+
3462
+ customer_eni: Optional[str] = Unassigned()
3463
+ additional_enis: Optional[AdditionalEnis] = Unassigned()
3464
+ capacity_reservation: Optional[CapacityReservation] = Unassigned()
3465
+ failure_message: Optional[str] = Unassigned()
3466
+ lcs_execution_state: Optional[str] = Unassigned()
3467
+ node_logical_id: Optional[str] = Unassigned()
3468
+
3469
+
3470
+ class EventMetadata(Base):
3471
+ """
3472
+ EventMetadata
3473
+ Metadata associated with a cluster event, which may include details about various resource types.
3474
+
3475
+ Attributes
3476
+ ----------------------
3477
+ cluster: Metadata specific to cluster-level events.
3478
+ instance_group: Metadata specific to instance group-level events.
3479
+ instance_group_scaling: Metadata related to instance group scaling events.
3480
+ instance: Metadata specific to instance-level events.
3481
+ """
3482
+
3483
+ cluster: Optional[ClusterMetadata] = Unassigned()
3484
+ instance_group: Optional[InstanceGroupMetadata] = Unassigned()
3485
+ instance_group_scaling: Optional[InstanceGroupScalingMetadata] = Unassigned()
3486
+ instance: Optional[InstanceMetadata] = Unassigned()
3487
+
3488
+
3489
+ class EventDetails(Base):
3490
+ """
3491
+ EventDetails
3492
+ Detailed information about a specific event, including event metadata.
3493
+
3494
+ Attributes
3495
+ ----------------------
3496
+ event_metadata: Metadata specific to the event, which may include information about the cluster, instance group, or instance involved.
3497
+ """
3498
+
3499
+ event_metadata: Optional[EventMetadata] = Unassigned()
3500
+
3501
+
3502
+ class ClusterEventDetail(Base):
3503
+ """
3504
+ ClusterEventDetail
3505
+ Detailed information about a specific event in a HyperPod cluster.
3506
+
3507
+ Attributes
3508
+ ----------------------
3509
+ event_id: The unique identifier (UUID) of the event.
3510
+ cluster_arn: The Amazon Resource Name (ARN) of the SageMaker HyperPod cluster associated with the event.
3511
+ cluster_name: The name of the SageMaker HyperPod cluster associated with the event.
3512
+ instance_group_name: The name of the instance group associated with the event, if applicable.
3513
+ instance_id: The EC2 instance ID associated with the event, if applicable.
3514
+ resource_type: The type of resource associated with the event. Valid values are "Cluster", "InstanceGroup", or "Instance".
3515
+ event_time: The timestamp when the event occurred.
3516
+ event_details: Additional details about the event, including event-specific metadata.
3517
+ description: A human-readable description of the event.
3518
+ """
3519
+
3520
+ event_id: str
3521
+ cluster_arn: str
3522
+ cluster_name: Union[str, object]
3523
+ resource_type: str
3524
+ event_time: datetime.datetime
3525
+ instance_group_name: Optional[str] = Unassigned()
3526
+ instance_id: Optional[str] = Unassigned()
3527
+ event_details: Optional[EventDetails] = Unassigned()
3528
+ description: Optional[str] = Unassigned()
3529
+
3530
+
3531
+ class ClusterEventSummary(Base):
3532
+ """
3533
+ ClusterEventSummary
3534
+ A summary of an event in a SageMaker HyperPod cluster.
3535
+
3536
+ Attributes
3537
+ ----------------------
3538
+ event_id: The unique identifier (UUID) of the event.
3539
+ cluster_arn: The Amazon Resource Name (ARN) of the SageMaker HyperPod cluster associated with the event.
3540
+ cluster_name: The name of the SageMaker HyperPod cluster associated with the event.
3541
+ instance_group_name: The name of the instance group associated with the event, if applicable.
3542
+ instance_id: The EC2 instance ID associated with the event, if applicable.
3543
+ resource_type: The type of resource associated with the event. Valid values are "Cluster", "InstanceGroup", or "Instance".
3544
+ event_time: The timestamp when the event occurred.
3545
+ description: A brief, human-readable description of the event.
3546
+ """
3547
+
3548
+ event_id: str
3549
+ cluster_arn: str
3550
+ cluster_name: Union[str, object]
3551
+ resource_type: str
3552
+ event_time: datetime.datetime
3553
+ instance_group_name: Optional[str] = Unassigned()
3554
+ instance_id: Optional[str] = Unassigned()
3555
+ description: Optional[str] = Unassigned()
3556
+
3557
+
3290
3558
  class ClusterLifeCycleConfig(Base):
3291
3559
  """
3292
3560
  ClusterLifeCycleConfig
@@ -3383,6 +3651,8 @@ class ClusterInstanceGroupDetails(Base):
3383
3651
  training_plan_status: The current status of the training plan associated with this cluster instance group.
3384
3652
  override_vpc_config: The customized Amazon VPC configuration at the instance group level that overrides the default Amazon VPC configuration of the SageMaker HyperPod cluster.
3385
3653
  scheduled_update_config: The configuration object of the schedule that SageMaker follows when updating the AMI.
3654
+ current_image_id: The ID of the Amazon Machine Image (AMI) currently in use by the instance group.
3655
+ desired_image_id: The ID of the Amazon Machine Image (AMI) desired for the instance group.
3386
3656
  """
3387
3657
 
3388
3658
  current_count: Optional[int] = Unassigned()
@@ -3399,6 +3669,8 @@ class ClusterInstanceGroupDetails(Base):
3399
3669
  training_plan_status: Optional[str] = Unassigned()
3400
3670
  override_vpc_config: Optional[VpcConfig] = Unassigned()
3401
3671
  scheduled_update_config: Optional[ScheduledUpdateConfig] = Unassigned()
3672
+ current_image_id: Optional[str] = Unassigned()
3673
+ desired_image_id: Optional[str] = Unassigned()
3402
3674
 
3403
3675
 
3404
3676
  class ClusterInstanceGroupSpecification(Base):
@@ -3419,6 +3691,7 @@ class ClusterInstanceGroupSpecification(Base):
3419
3691
  training_plan_arn: The Amazon Resource Name (ARN); of the training plan to use for this cluster instance group. For more information about how to reserve GPU capacity for your SageMaker HyperPod clusters using Amazon SageMaker Training Plan, see CreateTrainingPlan .
3420
3692
  override_vpc_config: To configure multi-AZ deployments, customize the Amazon VPC configuration at the instance group level. You can specify different subnets and security groups across different AZs in the instance group specification to override a SageMaker HyperPod cluster's default Amazon VPC configuration. For more information about deploying a cluster in multiple AZs, see Setting up SageMaker HyperPod clusters across multiple AZs. When your Amazon VPC and subnets support IPv6, network communications differ based on the cluster orchestration platform: Slurm-orchestrated clusters automatically configure nodes with dual IPv6 and IPv4 addresses, allowing immediate IPv6 network communications. In Amazon EKS-orchestrated clusters, nodes receive dual-stack addressing, but pods can only use IPv6 when the Amazon EKS cluster is explicitly IPv6-enabled. For information about deploying an IPv6 Amazon EKS cluster, see Amazon EKS IPv6 Cluster Deployment. Additional resources for IPv6 configuration: For information about adding IPv6 support to your VPC, see to IPv6 Support for VPC. For information about creating a new IPv6-compatible VPC, see Amazon VPC Creation Guide. To configure SageMaker HyperPod with a custom Amazon VPC, see Custom Amazon VPC Setup for SageMaker HyperPod.
3421
3693
  scheduled_update_config: The configuration object of the schedule that SageMaker uses to update the AMI.
3694
+ image_id: When configuring your HyperPod cluster, you can specify an image ID using one of the following options: HyperPodPublicAmiId: Use a HyperPod public AMI CustomAmiId: Use your custom AMI default: Use the default latest system image f you choose to use a custom AMI (CustomAmiId), ensure it meets the following requirements: Encryption: The custom AMI must be unencrypted. Ownership: The custom AMI must be owned by the same Amazon Web Services account that is creating the HyperPod cluster. Volume support: Only the primary AMI snapshot volume is supported; additional AMI volumes are not supported. When updating the instance group's AMI through the UpdateClusterSoftware operation, if an instance group uses a custom AMI, you must provide an ImageId or use the default as input.
3422
3695
  """
3423
3696
 
3424
3697
  instance_count: int
@@ -3432,6 +3705,7 @@ class ClusterInstanceGroupSpecification(Base):
3432
3705
  training_plan_arn: Optional[str] = Unassigned()
3433
3706
  override_vpc_config: Optional[VpcConfig] = Unassigned()
3434
3707
  scheduled_update_config: Optional[ScheduledUpdateConfig] = Unassigned()
3708
+ image_id: Optional[str] = Unassigned()
3435
3709
 
3436
3710
 
3437
3711
  class ClusterInstancePlacement(Base):
@@ -3473,6 +3747,7 @@ class ClusterNodeDetails(Base):
3473
3747
  ----------------------
3474
3748
  instance_group_name: The instance group name in which the instance is.
3475
3749
  instance_id: The ID of the instance.
3750
+ node_logical_id: A unique identifier for the node that persists throughout its lifecycle, from provisioning request to termination. This identifier can be used to track the node even before it has an assigned InstanceId.
3476
3751
  instance_status: The status of the instance.
3477
3752
  instance_type: The type of the instance.
3478
3753
  launch_time: The time when the instance is launched.
@@ -3485,10 +3760,13 @@ class ClusterNodeDetails(Base):
3485
3760
  private_primary_ipv6: The private primary IPv6 address of the SageMaker HyperPod cluster node when configured with an Amazon VPC that supports IPv6 and includes subnets with IPv6 addressing enabled in either the cluster Amazon VPC configuration or the instance group Amazon VPC configuration.
3486
3761
  private_dns_hostname: The private DNS hostname of the SageMaker HyperPod cluster node.
3487
3762
  placement: The placement details of the SageMaker HyperPod cluster node.
3763
+ current_image_id: The ID of the Amazon Machine Image (AMI) currently in use by the node.
3764
+ desired_image_id: The ID of the Amazon Machine Image (AMI) desired for the node.
3488
3765
  """
3489
3766
 
3490
3767
  instance_group_name: Optional[str] = Unassigned()
3491
3768
  instance_id: Optional[str] = Unassigned()
3769
+ node_logical_id: Optional[str] = Unassigned()
3492
3770
  instance_status: Optional[ClusterInstanceStatusDetails] = Unassigned()
3493
3771
  instance_type: Optional[str] = Unassigned()
3494
3772
  launch_time: Optional[datetime.datetime] = Unassigned()
@@ -3501,6 +3779,8 @@ class ClusterNodeDetails(Base):
3501
3779
  private_primary_ipv6: Optional[str] = Unassigned()
3502
3780
  private_dns_hostname: Optional[str] = Unassigned()
3503
3781
  placement: Optional[ClusterInstancePlacement] = Unassigned()
3782
+ current_image_id: Optional[str] = Unassigned()
3783
+ desired_image_id: Optional[str] = Unassigned()
3504
3784
 
3505
3785
 
3506
3786
  class ClusterNodeSummary(Base):
@@ -3512,6 +3792,7 @@ class ClusterNodeSummary(Base):
3512
3792
  ----------------------
3513
3793
  instance_group_name: The name of the instance group in which the instance is.
3514
3794
  instance_id: The ID of the instance.
3795
+ node_logical_id: A unique identifier for the node that persists throughout its lifecycle, from provisioning request to termination. This identifier can be used to track the node even before it has an assigned InstanceId. This field is only included when IncludeNodeLogicalIds is set to True in the ListClusterNodes request.
3515
3796
  instance_type: The type of the instance.
3516
3797
  launch_time: The time when the instance is launched.
3517
3798
  last_software_update_time: The time when SageMaker last updated the software of the instances in the cluster.
@@ -3523,6 +3804,7 @@ class ClusterNodeSummary(Base):
3523
3804
  instance_type: str
3524
3805
  launch_time: datetime.datetime
3525
3806
  instance_status: ClusterInstanceStatusDetails
3807
+ node_logical_id: Optional[str] = Unassigned()
3526
3808
  last_software_update_time: Optional[datetime.datetime] = Unassigned()
3527
3809
 
3528
3810
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: sagemaker-core
3
- Version: 1.0.48
3
+ Version: 1.0.49
4
4
  Summary: An python package for sagemaker core functionalities
5
5
  Author-email: AWS <sagemaker-interests@amazon.com>
6
6
  Project-URL: Repository, https://github.com/aws/sagemaker-core.git
@@ -7,15 +7,15 @@ sagemaker_core/main/config_schema.py,sha256=Wxe2gJash1rrxBomGhSYmII1LmJ3E70LIuSW
7
7
  sagemaker_core/main/default_configs_helper.py,sha256=bg_tgczX_bYzNiSlalJ6TWPTgrQYsI0uZguP5TIbPiw,8324
8
8
  sagemaker_core/main/exceptions.py,sha256=CsiM3V_Gb16grBotnu59LB6tznryPcSvAQDAOOYGc10,5563
9
9
  sagemaker_core/main/logs.py,sha256=yfEH7uP91nbE1lefymOlBr81ziBzsDSIOF2Qyd54FJE,6241
10
- sagemaker_core/main/resources.py,sha256=lDuKu3EHRhEjOE6MVLHfPABxGzBNHh8R2j961ITdMA0,1437580
11
- sagemaker_core/main/shapes.py,sha256=sGjB8ICd0F5bYmvFx1Y1yvscFS00sg6RCjrMR3CXx9s,762943
10
+ sagemaker_core/main/resources.py,sha256=nldjn7RZ2C0Myq-R-nvYBjouuQFPAJ0CrAhA3e61cNI,1440403
11
+ sagemaker_core/main/shapes.py,sha256=cjTQKK-UZg_SkNEDAR8EUlul-046-s14pD6zt1PEo4I,776265
12
12
  sagemaker_core/main/user_agent.py,sha256=BPYDAfDd70ObP-VAjl7aDHALHyGknkpRP21ktVr_LDw,2744
13
13
  sagemaker_core/main/utils.py,sha256=y1aZzztCApczZSqT0U7_H8zLlpqp1vgoUKcp40mTM2o,19157
14
14
  sagemaker_core/main/code_injection/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
15
15
  sagemaker_core/main/code_injection/base.py,sha256=11_Jif0nOzfbLGlXaacKf-wcizzfS64U0OSZGoVffFU,1733
16
16
  sagemaker_core/main/code_injection/codec.py,sha256=2YzJ-iYEzmguVaJVcZeyCR0OpTSR7UOixATrOm4MiBk,8885
17
17
  sagemaker_core/main/code_injection/constants.py,sha256=2ICExGge8vAWx7lSTW0JGh-bH1korkvpOpDu5M63eI4,980
18
- sagemaker_core/main/code_injection/shape_dag.py,sha256=m19s6YykKJgkM5mX4t_aCoMlmZVpXR2h6nwy_mYVM5Q,726467
18
+ sagemaker_core/main/code_injection/shape_dag.py,sha256=W33MDZhfRmGo1EMI25QVbQyqdc9N1lb50Lr1mnErDxY,737697
19
19
  sagemaker_core/resources/__init__.py,sha256=EAYTFMN-nPjnPjjBbhIUeaL67FLKNPd7qbcbl9VIrws,31
20
20
  sagemaker_core/shapes/__init__.py,sha256=RnbIu9eTxKt-DNsOFJabrWIgrrtS9_SdAozP9JBl_ic,28
21
21
  sagemaker_core/tools/__init__.py,sha256=xX79JImxCVzrWMnjgntLCve2G5I-R4pRar5s20kT9Rs,56
@@ -28,8 +28,8 @@ sagemaker_core/tools/resources_extractor.py,sha256=hN61ehZbPnhFW-2FIVDi7NsEz4rLv
28
28
  sagemaker_core/tools/shapes_codegen.py,sha256=4lsePZpjk7M6RpJs5yar_m4z5MzwGHFrvCkdS_-R12c,12172
29
29
  sagemaker_core/tools/shapes_extractor.py,sha256=vxVKjXD3lmjrkoKiexjUnOt8ITbFxQSeiDtx7P6Qtkw,14226
30
30
  sagemaker_core/tools/templates.py,sha256=0lOIH3Rq2CXWkQhK6VenN_TE_v5p852s2kQyb_BeQxA,23460
31
- sagemaker_core-1.0.48.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
32
- sagemaker_core-1.0.48.dist-info/METADATA,sha256=enC8HCSHEqfxtM05NfKFnWazNYuv49H95De1E4mMKTY,4871
33
- sagemaker_core-1.0.48.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
34
- sagemaker_core-1.0.48.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
35
- sagemaker_core-1.0.48.dist-info/RECORD,,
31
+ sagemaker_core-1.0.49.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
32
+ sagemaker_core-1.0.49.dist-info/METADATA,sha256=zHe0l5x-M6jMwtIEYhbhoVO82faXhsgNHaf0XmUL5GU,4871
33
+ sagemaker_core-1.0.49.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
34
+ sagemaker_core-1.0.49.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
35
+ sagemaker_core-1.0.49.dist-info/RECORD,,