sagemaker-core 1.0.47__py3-none-any.whl → 1.0.49__py3-none-any.whl

sagemaker_core/__init__.py

@@ -1,4 +0,0 @@
-from sagemaker_core.main.utils import enable_textual_rich_console_and_traceback
-
-
-enable_textual_rich_console_and_traceback()

sagemaker_core/main/code_injection/shape_dag.py

@@ -43,6 +43,22 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "AddClusterNodeSpecification": {
+        "members": [
+            {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
+            {
+                "name": "IncrementTargetCountBy",
+                "shape": "AddClusterNodeSpecificationIncrementTargetCountByInteger",
+                "type": "integer",
+            },
+        ],
+        "type": "structure",
+    },
+    "AddClusterNodeSpecificationList": {
+        "member_shape": "AddClusterNodeSpecification",
+        "member_type": "structure",
+        "type": "list",
+    },
     "AddTagsInput": {
         "members": [
             {"name": "ResourceArn", "shape": "ResourceArn", "type": "string"},
@@ -59,6 +75,10 @@ SHAPE_DAG = {
         "member_type": "string",
         "type": "list",
     },
+    "AdditionalEnis": {
+        "members": [{"name": "EfaEnis", "shape": "EfaEnis", "type": "list"}],
+        "type": "structure",
+    },
     "AdditionalInferenceSpecificationDefinition": {
         "members": [
             {"name": "Name", "shape": "EntityName", "type": "string"},
@@ -413,6 +433,25 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "AttachClusterNodeVolumeRequest": {
+        "members": [
+            {"name": "ClusterArn", "shape": "ClusterArn", "type": "string"},
+            {"name": "NodeId", "shape": "ClusterNodeId", "type": "string"},
+            {"name": "VolumeId", "shape": "VolumeId", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "AttachClusterNodeVolumeResponse": {
+        "members": [
+            {"name": "ClusterArn", "shape": "ClusterArn", "type": "string"},
+            {"name": "NodeId", "shape": "ClusterNodeId", "type": "string"},
+            {"name": "VolumeId", "shape": "VolumeId", "type": "string"},
+            {"name": "AttachTime", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "Status", "shape": "VolumeAttachmentStatus", "type": "string"},
+            {"name": "DeviceName", "shape": "VolumeDeviceName", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "AttributeNames": {"member_shape": "AttributeName", "member_type": "string", "type": "list"},
     "AuthenticationRequestExtraParams": {
         "key_shape": "AuthenticationRequestExtraParamsKey",
@@ -763,6 +802,39 @@ SHAPE_DAG = {
         "members": [{"name": "Mode", "shape": "AutotuneMode", "type": "string"}],
         "type": "structure",
     },
+    "BatchAddClusterNodesError": {
+        "members": [
+            {"name": "InstanceGroupName", "shape": "InstanceGroupName", "type": "string"},
+            {"name": "ErrorCode", "shape": "BatchAddClusterNodesErrorCode", "type": "string"},
+            {"name": "FailedCount", "shape": "BatchAddFailureCount", "type": "integer"},
+            {"name": "Message", "shape": "String", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "BatchAddClusterNodesErrorList": {
+        "member_shape": "BatchAddClusterNodesError",
+        "member_type": "structure",
+        "type": "list",
+    },
+    "BatchAddClusterNodesRequest": {
+        "members": [
+            {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
+            {
+                "name": "ClientToken",
+                "shape": "BatchAddClusterNodesRequestClientTokenString",
+                "type": "string",
+            },
+            {"name": "NodesToAdd", "shape": "AddClusterNodeSpecificationList", "type": "list"},
+        ],
+        "type": "structure",
+    },
+    "BatchAddClusterNodesResponse": {
+        "members": [
+            {"name": "Successful", "shape": "NodeAdditionResultList", "type": "list"},
+            {"name": "Failed", "shape": "BatchAddClusterNodesErrorList", "type": "list"},
+        ],
+        "type": "structure",
+    },
     "BatchDataCaptureConfig": {
         "members": [
             {"name": "DestinationS3Uri", "shape": "S3Uri", "type": "string"},
@@ -771,6 +843,19 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "BatchDeleteClusterNodeLogicalIdsError": {
+        "members": [
+            {"name": "Code", "shape": "BatchDeleteClusterNodesErrorCode", "type": "string"},
+            {"name": "Message", "shape": "String", "type": "string"},
+            {"name": "NodeLogicalId", "shape": "ClusterNodeLogicalId", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "BatchDeleteClusterNodeLogicalIdsErrorList": {
+        "member_shape": "BatchDeleteClusterNodeLogicalIdsError",
+        "member_type": "structure",
+        "type": "list",
+    },
     "BatchDeleteClusterNodesError": {
         "members": [
             {"name": "Code", "shape": "BatchDeleteClusterNodesErrorCode", "type": "string"},
@@ -788,6 +873,7 @@ SHAPE_DAG = {
         "members": [
             {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
             {"name": "NodeIds", "shape": "ClusterNodeIds", "type": "list"},
+            {"name": "NodeLogicalIds", "shape": "ClusterNodeLogicalIdList", "type": "list"},
         ],
         "type": "structure",
     },
@@ -795,6 +881,16 @@ SHAPE_DAG = {
         "members": [
             {"name": "Failed", "shape": "BatchDeleteClusterNodesErrorList", "type": "list"},
             {"name": "Successful", "shape": "ClusterNodeIds", "type": "list"},
+            {
+                "name": "FailedNodeLogicalIds",
+                "shape": "BatchDeleteClusterNodeLogicalIdsErrorList",
+                "type": "list",
+            },
+            {
+                "name": "SuccessfulNodeLogicalIds",
+                "shape": "ClusterNodeLogicalIdList",
+                "type": "list",
+            },
         ],
         "type": "structure",
     },
@@ -1083,6 +1179,13 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "CapacityReservation": {
+        "members": [
+            {"name": "Arn", "shape": "String", "type": "string"},
+            {"name": "Type", "shape": "CapacityReservationType", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "CapacitySize": {
         "members": [
             {"name": "Type", "shape": "CapacitySizeType", "type": "string"},
@@ -1348,6 +1451,38 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "ClusterEventDetail": {
+        "members": [
+            {"name": "EventId", "shape": "EventId", "type": "string"},
+            {"name": "ClusterArn", "shape": "ClusterArn", "type": "string"},
+            {"name": "ClusterName", "shape": "ClusterName", "type": "string"},
+            {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
+            {"name": "InstanceId", "shape": "String", "type": "string"},
+            {"name": "ResourceType", "shape": "ClusterEventResourceType", "type": "string"},
+            {"name": "EventTime", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "EventDetails", "shape": "EventDetails", "type": "structure"},
+            {"name": "Description", "shape": "String", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "ClusterEventSummaries": {
+        "member_shape": "ClusterEventSummary",
+        "member_type": "structure",
+        "type": "list",
+    },
+    "ClusterEventSummary": {
+        "members": [
+            {"name": "EventId", "shape": "EventId", "type": "string"},
+            {"name": "ClusterArn", "shape": "ClusterArn", "type": "string"},
+            {"name": "ClusterName", "shape": "ClusterName", "type": "string"},
+            {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
+            {"name": "InstanceId", "shape": "String", "type": "string"},
+            {"name": "ResourceType", "shape": "ClusterEventResourceType", "type": "string"},
+            {"name": "EventTime", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "Description", "shape": "String", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "ClusterInstanceGroupDetails": {
         "members": [
             {"name": "CurrentCount", "shape": "ClusterNonNegativeInstanceCount", "type": "integer"},
@@ -1376,6 +1511,8 @@ SHAPE_DAG = {
                 "shape": "ScheduledUpdateConfig",
                 "type": "structure",
             },
+            {"name": "CurrentImageId", "shape": "ImageId", "type": "string"},
+            {"name": "DesiredImageId", "shape": "ImageId", "type": "string"},
         ],
         "type": "structure",
     },
@@ -1405,6 +1542,7 @@ SHAPE_DAG = {
                 "shape": "ScheduledUpdateConfig",
                 "type": "structure",
             },
+            {"name": "ImageId", "shape": "ImageId", "type": "string"},
         ],
         "type": "structure",
     },
@@ -1450,10 +1588,19 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "ClusterMetadata": {
+        "members": [
+            {"name": "FailureMessage", "shape": "String", "type": "string"},
+            {"name": "EksRoleAccessEntries", "shape": "EksRoleAccessEntries", "type": "list"},
+            {"name": "SlrAccessEntry", "shape": "String", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "ClusterNodeDetails": {
         "members": [
             {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
             {"name": "InstanceId", "shape": "String", "type": "string"},
+            {"name": "NodeLogicalId", "shape": "ClusterNodeLogicalId", "type": "string"},
             {
                 "name": "InstanceStatus",
                 "shape": "ClusterInstanceStatusDetails",
@@ -1474,10 +1621,17 @@ SHAPE_DAG = {
             {"name": "PrivatePrimaryIpv6", "shape": "ClusterPrivatePrimaryIpv6", "type": "string"},
             {"name": "PrivateDnsHostname", "shape": "ClusterPrivateDnsHostname", "type": "string"},
             {"name": "Placement", "shape": "ClusterInstancePlacement", "type": "structure"},
+            {"name": "CurrentImageId", "shape": "ImageId", "type": "string"},
+            {"name": "DesiredImageId", "shape": "ImageId", "type": "string"},
         ],
         "type": "structure",
     },
     "ClusterNodeIds": {"member_shape": "ClusterNodeId", "member_type": "string", "type": "list"},
+    "ClusterNodeLogicalIdList": {
+        "member_shape": "ClusterNodeLogicalId",
+        "member_type": "string",
+        "type": "list",
+    },
     "ClusterNodeSummaries": {
         "member_shape": "ClusterNodeSummary",
         "member_type": "structure",
@@ -1487,6 +1641,7 @@ SHAPE_DAG = {
         "members": [
             {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
             {"name": "InstanceId", "shape": "String", "type": "string"},
+            {"name": "NodeLogicalId", "shape": "String", "type": "string"},
             {"name": "InstanceType", "shape": "ClusterInstanceType", "type": "string"},
             {"name": "LaunchTime", "shape": "Timestamp", "type": "timestamp"},
             {"name": "LastSoftwareUpdateTime", "shape": "Timestamp", "type": "timestamp"},
@@ -2084,6 +2239,11 @@ SHAPE_DAG = {
             {"name": "Tags", "shape": "TagList", "type": "list"},
             {"name": "Orchestrator", "shape": "ClusterOrchestrator", "type": "structure"},
             {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
+            {
+                "name": "NodeProvisioningMode",
+                "shape": "ClusterNodeProvisioningMode",
+                "type": "string",
+            },
         ],
         "type": "structure",
     },
@@ -4422,10 +4582,22 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "DescribeClusterEventRequest": {
+        "members": [
+            {"name": "EventId", "shape": "EventId", "type": "string"},
+            {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "DescribeClusterEventResponse": {
+        "members": [{"name": "EventDetails", "shape": "ClusterEventDetail", "type": "structure"}],
+        "type": "structure",
+    },
     "DescribeClusterNodeRequest": {
         "members": [
             {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
             {"name": "NodeId", "shape": "ClusterNodeId", "type": "string"},
+            {"name": "NodeLogicalId", "shape": "ClusterNodeLogicalId", "type": "string"},
         ],
         "type": "structure",
     },
@@ -4453,6 +4625,11 @@ SHAPE_DAG = {
             {"name": "VpcConfig", "shape": "VpcConfig", "type": "structure"},
             {"name": "Orchestrator", "shape": "ClusterOrchestrator", "type": "structure"},
             {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
+            {
+                "name": "NodeProvisioningMode",
+                "shape": "ClusterNodeProvisioningMode",
+                "type": "string",
+            },
         ],
         "type": "structure",
     },
@@ -6262,6 +6439,25 @@ SHAPE_DAG = {
         "member_type": "structure",
         "type": "list",
     },
+    "DetachClusterNodeVolumeRequest": {
+        "members": [
+            {"name": "ClusterArn", "shape": "ClusterArn", "type": "string"},
+            {"name": "NodeId", "shape": "ClusterNodeId", "type": "string"},
+            {"name": "VolumeId", "shape": "VolumeId", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "DetachClusterNodeVolumeResponse": {
+        "members": [
+            {"name": "ClusterArn", "shape": "ClusterArn", "type": "string"},
+            {"name": "NodeId", "shape": "ClusterNodeId", "type": "string"},
+            {"name": "VolumeId", "shape": "VolumeId", "type": "string"},
+            {"name": "AttachTime", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "Status", "shape": "VolumeAttachmentStatus", "type": "string"},
+            {"name": "DeviceName", "shape": "VolumeDeviceName", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "Device": {
         "members": [
             {"name": "DeviceName", "shape": "DeviceName", "type": "string"},
@@ -6655,6 +6851,8 @@ SHAPE_DAG = {
         "type": "structure",
     },
     "Edges": {"member_shape": "Edge", "member_type": "structure", "type": "list"},
+    "EfaEnis": {"member_shape": "String", "member_type": "string", "type": "list"},
+    "EksRoleAccessEntries": {"member_shape": "String", "member_type": "string", "type": "list"},
     "EmrServerlessComputeConfig": {
         "members": [{"name": "ExecutionRoleARN", "shape": "RoleArn", "type": "string"}],
         "type": "structure",
@@ -6870,6 +7068,23 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "EventDetails": {
+        "members": [{"name": "EventMetadata", "shape": "EventMetadata", "type": "structure"}],
+        "type": "structure",
+    },
+    "EventMetadata": {
+        "members": [
+            {"name": "Cluster", "shape": "ClusterMetadata", "type": "structure"},
+            {"name": "InstanceGroup", "shape": "InstanceGroupMetadata", "type": "structure"},
+            {
+                "name": "InstanceGroupScaling",
+                "shape": "InstanceGroupScalingMetadata",
+                "type": "structure",
+            },
+            {"name": "Instance", "shape": "InstanceMetadata", "type": "structure"},
+        ],
+        "type": "structure",
+    },
     "ExecutionRoleArns": {"member_shape": "RoleArn", "member_type": "string", "type": "list"},
     "Experiment": {
         "members": [
@@ -8328,12 +8543,42 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "InstanceGroupMetadata": {
+        "members": [
+            {"name": "FailureMessage", "shape": "String", "type": "string"},
+            {"name": "AvailabilityZoneId", "shape": "String", "type": "string"},
+            {"name": "CapacityReservation", "shape": "CapacityReservation", "type": "structure"},
+            {"name": "SubnetId", "shape": "String", "type": "string"},
+            {"name": "SecurityGroupIds", "shape": "SecurityGroupIds", "type": "list"},
+            {"name": "AmiOverride", "shape": "String", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "InstanceGroupNames": {
         "member_shape": "InstanceGroupName",
         "member_type": "string",
         "type": "list",
     },
+    "InstanceGroupScalingMetadata": {
+        "members": [
+            {"name": "InstanceCount", "shape": "InstanceCount", "type": "integer"},
+            {"name": "TargetCount", "shape": "TargetCount", "type": "integer"},
+            {"name": "FailureMessage", "shape": "String", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "InstanceGroups": {"member_shape": "InstanceGroup", "member_type": "structure", "type": "list"},
+    "InstanceMetadata": {
+        "members": [
+            {"name": "CustomerEni", "shape": "String", "type": "string"},
+            {"name": "AdditionalEnis", "shape": "AdditionalEnis", "type": "structure"},
+            {"name": "CapacityReservation", "shape": "CapacityReservation", "type": "structure"},
+            {"name": "FailureMessage", "shape": "String", "type": "string"},
+            {"name": "LcsExecutionState", "shape": "String", "type": "string"},
+            {"name": "NodeLogicalId", "shape": "ClusterNodeLogicalId", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "InstanceMetadataServiceConfiguration": {
         "members": [
             {
@@ -8905,6 +9150,28 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "ListClusterEventsRequest": {
+        "members": [
+            {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
+            {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
+            {"name": "NodeId", "shape": "ClusterNodeId", "type": "string"},
+            {"name": "EventTimeAfter", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "EventTimeBefore", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "SortBy", "shape": "EventSortBy", "type": "string"},
+            {"name": "SortOrder", "shape": "SortOrder", "type": "string"},
+            {"name": "ResourceType", "shape": "ClusterEventResourceType", "type": "string"},
+            {"name": "MaxResults", "shape": "ClusterEventMaxResults", "type": "integer"},
+            {"name": "NextToken", "shape": "NextToken", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "ListClusterEventsResponse": {
+        "members": [
+            {"name": "NextToken", "shape": "NextToken", "type": "string"},
+            {"name": "Events", "shape": "ClusterEventSummaries", "type": "list"},
+        ],
+        "type": "structure",
+    },
     "ListClusterNodesRequest": {
         "members": [
             {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
@@ -8919,6 +9186,11 @@ SHAPE_DAG = {
             {"name": "NextToken", "shape": "NextToken", "type": "string"},
             {"name": "SortBy", "shape": "ClusterSortBy", "type": "string"},
             {"name": "SortOrder", "shape": "SortOrder", "type": "string"},
+            {
+                "name": "IncludeNodeLogicalIds",
+                "shape": "IncludeNodeLogicalIdsBoolean",
+                "type": "boolean",
+            },
         ],
         "type": "structure",
     },
@@ -11765,6 +12037,19 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "NodeAdditionResult": {
+        "members": [
+            {"name": "NodeLogicalId", "shape": "ClusterNodeLogicalId", "type": "string"},
+            {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
+            {"name": "Status", "shape": "ClusterInstanceStatus", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "NodeAdditionResultList": {
+        "member_shape": "NodeAdditionResult",
+        "member_type": "structure",
+        "type": "list",
+    },
     "NotebookInstanceAcceleratorTypes": {
         "member_shape": "NotebookInstanceAcceleratorType",
         "member_type": "string",
@@ -15419,6 +15704,7 @@ SHAPE_DAG = {
                 "type": "list",
             },
             {"name": "DeploymentConfig", "shape": "DeploymentConfiguration", "type": "structure"},
+            {"name": "ImageId", "shape": "ImageId", "type": "string"},
         ],
         "type": "structure",
     },

sagemaker_core/main/resources.py

@@ -3223,6 +3223,7 @@ class Cluster(Base):
         vpc_config:
         orchestrator: The type of orchestrator used for the SageMaker HyperPod cluster.
         node_recovery: The node recovery mode configured for the SageMaker HyperPod cluster.
+        node_provisioning_mode: The mode used for provisioning nodes in the cluster.
 
     """
 
@@ -3238,6 +3239,7 @@ class Cluster(Base):
     vpc_config: Optional[shapes.VpcConfig] = Unassigned()
     orchestrator: Optional[shapes.ClusterOrchestrator] = Unassigned()
     node_recovery: Optional[str] = Unassigned()
+    node_provisioning_mode: Optional[str] = Unassigned()
 
     def get_name(self) -> str:
         attributes = vars(self)
@@ -3287,6 +3289,7 @@ class Cluster(Base):
         tags: Optional[List[shapes.Tag]] = Unassigned(),
         orchestrator: Optional[shapes.ClusterOrchestrator] = Unassigned(),
         node_recovery: Optional[str] = Unassigned(),
+        node_provisioning_mode: Optional[str] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> Optional["Cluster"]:
@@ -3301,6 +3304,7 @@ class Cluster(Base):
             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.
             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.
             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.
+            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.
             session: Boto3 session.
             region: Region name.
 
@@ -3337,6 +3341,7 @@ class Cluster(Base):
             "Tags": tags,
             "Orchestrator": orchestrator,
             "NodeRecovery": node_recovery,
+            "NodeProvisioningMode": node_provisioning_mode,
         }
 
         operation_input_args = Base.populate_chained_attributes(
@@ -3731,6 +3736,7 @@ class Cluster(Base):
     def get_node(
         self,
         node_id: Optional[str] = Unassigned(),
+        node_logical_id: Optional[str] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> Optional[shapes.ClusterNodeDetails]:
@@ -3739,6 +3745,7 @@ class Cluster(Base):
 
         Parameters:
             node_id: The ID of the SageMaker HyperPod cluster node.
+            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.
             session: Boto3 session.
             region: Region name.
 
@@ -3761,6 +3768,7 @@ class Cluster(Base):
         operation_input_args = {
             "ClusterName": self.cluster_name,
             "NodeId": node_id,
+            "NodeLogicalId": node_logical_id,
         }
         # serialize the input request
         operation_input_args = serialize(operation_input_args)
@@ -3785,6 +3793,7 @@ class Cluster(Base):
         instance_group_name_contains: Optional[str] = Unassigned(),
         sort_by: Optional[str] = Unassigned(),
         sort_order: Optional[str] = Unassigned(),
+        include_node_logical_ids: Optional[bool] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> ResourceIterator[shapes.ClusterNodeDetails]:
@@ -3799,6 +3808,7 @@ class Cluster(Base):
             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.
             sort_by: The field by which to sort results. The default value is CREATION_TIME.
             sort_order: The sort order for results. The default value is Ascending.
+            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.
             session: Boto3 session.
             region: Region name.
 
@@ -3825,6 +3835,7 @@ class Cluster(Base):
             "InstanceGroupNameContains": instance_group_name_contains,
             "SortBy": sort_by,
             "SortOrder": sort_order,
+            "IncludeNodeLogicalIds": include_node_logical_ids,
         }
         # serialize the input request
         operation_input_args = serialize(operation_input_args)
@@ -3847,6 +3858,7 @@ class Cluster(Base):
     def update_software(
         self,
         deployment_config: Optional[shapes.DeploymentConfiguration] = Unassigned(),
+        image_id: Optional[str] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> None:
@@ -3855,6 +3867,7 @@ class Cluster(Base):
 
         Parameters:
             deployment_config: The configuration to use when updating the AMI versions.
+            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.
             session: Boto3 session.
             region: Region name.
 
@@ -3876,6 +3889,7 @@ class Cluster(Base):
             "ClusterName": self.cluster_name,
             "InstanceGroups": self.instance_groups,
             "DeploymentConfig": deployment_config,
+            "ImageId": image_id,
         }
         # serialize the input request
         operation_input_args = serialize(operation_input_args)
@@ -3893,6 +3907,7 @@ class Cluster(Base):
     def batch_delete_nodes(
         self,
         node_ids: Optional[List[str]] = Unassigned(),
+        node_logical_ids: Optional[List[str]] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> Optional[shapes.BatchDeleteClusterNodesResponse]:
@@ -3901,6 +3916,7 @@ class Cluster(Base):
 
         Parameters:
             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.
+            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.
             session: Boto3 session.
             region: Region name.
 
@@ -3923,6 +3939,7 @@ class Cluster(Base):
         operation_input_args = {
             "ClusterName": self.cluster_name,
             "NodeIds": node_ids,
+            "NodeLogicalIds": node_logical_ids,
         }
         # serialize the input request
         operation_input_args = serialize(operation_input_args)

sagemaker_core/main/shapes.py

@@ -494,6 +494,21 @@ class ActionSummary(Base):
     last_modified_time: Optional[datetime.datetime] = Unassigned()
 
 
+class AddClusterNodeSpecification(Base):
+    """
+    AddClusterNodeSpecification
+      Specifies an instance group and the number of nodes to add to it.
+
+    Attributes
+    ----------------------
+    instance_group_name: The name of the instance group to which you want to add nodes.
+    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.
+    """
+
+    instance_group_name: str
+    increment_target_count_by: int
+
+
 class Tag(Base):
     """
     Tag
@@ -509,6 +524,19 @@ class Tag(Base):
     value: str
 
 
+class AdditionalEnis(Base):
+    """
+    AdditionalEnis
+      Information about additional Elastic Network Interfaces (ENIs) associated with an instance.
+
+    Attributes
+    ----------------------
+    efa_enis: A list of Elastic Fabric Adapter (EFA) ENIs associated with the instance.
+    """
+
+    efa_enis: Optional[List[str]] = Unassigned()
+
+
 class ModelAccessConfig(Base):
     """
     ModelAccessConfig
@@ -2400,6 +2428,42 @@ class Autotune(Base):
     mode: str
 
 
+class BatchAddClusterNodesError(Base):
+    """
+    BatchAddClusterNodesError
+      Information about an error that occurred during the node addition operation.
+
+    Attributes
+    ----------------------
+    instance_group_name: The name of the instance group for which the error occurred.
+    error_code: The error code associated with the failure. Possible values include InstanceGroupNotFound and InvalidInstanceGroupState.
+    failed_count: The number of nodes that failed to be added to the specified instance group.
+    message: A descriptive message providing additional details about the error.
+    """
+
+    instance_group_name: str
+    error_code: str
+    failed_count: int
+    message: Optional[str] = Unassigned()
+
+
+class NodeAdditionResult(Base):
+    """
+    NodeAdditionResult
+      Information about a node that was successfully added to the cluster.
+
+    Attributes
+    ----------------------
+    node_logical_id: A unique identifier assigned to the node that can be used to track its provisioning status through the DescribeClusterNode operation.
+    instance_group_name: The name of the instance group to which the node was added.
+    status: The current status of the node. Possible values include Pending, Running, Failed, ShuttingDown, SystemUpdating, DeepHealthCheckInProgress, and NotFound.
+    """
+
+    node_logical_id: str
+    instance_group_name: str
+    status: str
+
+
 class BatchDataCaptureConfig(Base):
     """
     BatchDataCaptureConfig
@@ -2417,6 +2481,23 @@ class BatchDataCaptureConfig(Base):
     generate_inference_id: Optional[bool] = Unassigned()
 
 
+class BatchDeleteClusterNodeLogicalIdsError(Base):
+    """
+    BatchDeleteClusterNodeLogicalIdsError
+      Information about an error that occurred when attempting to delete a node identified by its NodeLogicalId.
+
+    Attributes
+    ----------------------
+    code: The error code associated with the failure. Possible values include NodeLogicalIdNotFound, InvalidNodeStatus, and InternalError.
+    message: A descriptive message providing additional details about the error.
+    node_logical_id: The NodeLogicalId of the node that could not be deleted.
+    """
+
+    code: str
+    message: str
+    node_logical_id: str
+
+
 class BatchDeleteClusterNodesError(Base):
     """
     BatchDeleteClusterNodesError
@@ -2442,10 +2523,14 @@ class BatchDeleteClusterNodesResponse(Base):
     ----------------------
     failed: A list of errors encountered when deleting the specified nodes.
     successful: A list of node IDs that were successfully deleted from the specified cluster.
+    failed_node_logical_ids: A list of NodeLogicalIds that could not be deleted, along with error information explaining why the deletion failed.
+    successful_node_logical_ids: A list of NodeLogicalIds that were successfully deleted from the cluster.
     """
 
     failed: Optional[List[BatchDeleteClusterNodesError]] = Unassigned()
     successful: Optional[List[str]] = Unassigned()
+    failed_node_logical_ids: Optional[List[BatchDeleteClusterNodeLogicalIdsError]] = Unassigned()
+    successful_node_logical_ids: Optional[List[str]] = Unassigned()
 
 
 class BatchDescribeModelPackageError(Base):
@@ -2901,6 +2986,21 @@ class CanvasAppSettings(Base):
     emr_serverless_settings: Optional[EmrServerlessSettings] = Unassigned()
 
 
+class CapacityReservation(Base):
+    """
+    CapacityReservation
+      Information about the Capacity Reservation used by an instance or instance group.
+
+    Attributes
+    ----------------------
+    arn: The Amazon Resource Name (ARN) of the Capacity Reservation.
+    type: The type of Capacity Reservation. Valid values are ODCR (On-Demand Capacity Reservation) or CRG (Capacity Reservation Group).
+    """
+
+    arn: Optional[str] = Unassigned()
+    type: Optional[str] = Unassigned()
+
+
 class CapacitySizeConfig(Base):
     """
     CapacitySizeConfig
@@ -3287,6 +3387,174 @@ class ClusterEbsVolumeConfig(Base):
     volume_size_in_gb: Optional[int] = Unassigned()
 
 
+class ClusterMetadata(Base):
+    """
+    ClusterMetadata
+      Metadata information about a SageMaker HyperPod cluster showing information about the cluster level operations, such as creating, updating, and deleting.
+
+    Attributes
+    ----------------------
+    failure_message: An error message describing why the cluster level operation (such as creating, updating, or deleting) failed.
+    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.
+    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.
+    """
+
+    failure_message: Optional[str] = Unassigned()
+    eks_role_access_entries: Optional[List[str]] = Unassigned()
+    slr_access_entry: Optional[str] = Unassigned()
+
+
+class InstanceGroupMetadata(Base):
+    """
+    InstanceGroupMetadata
+      Metadata information about an instance group in a SageMaker HyperPod cluster.
+
+    Attributes
+    ----------------------
+    failure_message: An error message describing why the instance group level operation (such as creating, scaling, or deleting) failed.
+    availability_zone_id: The ID of the Availability Zone where the instance group is located.
+    capacity_reservation: Information about the Capacity Reservation used by the instance group.
+    subnet_id: The ID of the subnet where the instance group is located.
+    security_group_ids: A list of security group IDs associated with the instance group.
+    ami_override: If you use a custom Amazon Machine Image (AMI) for the instance group, this field shows the ID of the custom AMI.
+    """
+
+    failure_message: Optional[str] = Unassigned()
+    availability_zone_id: Optional[str] = Unassigned()
+    capacity_reservation: Optional[CapacityReservation] = Unassigned()
+    subnet_id: Optional[str] = Unassigned()
+    security_group_ids: Optional[List[str]] = Unassigned()
+    ami_override: Optional[str] = Unassigned()
+
+
+class InstanceGroupScalingMetadata(Base):
+    """
+    InstanceGroupScalingMetadata
+      Metadata information about scaling operations for an instance group.
+
+    Attributes
+    ----------------------
+    instance_count: The current number of instances in the group.
+    target_count: The desired number of instances for the group after scaling.
+    failure_message: An error message describing why the scaling operation failed, if applicable.
+    """
+
+    instance_count: Optional[int] = Unassigned()
+    target_count: Optional[int] = Unassigned()
+    failure_message: Optional[str] = Unassigned()
+
+
+class InstanceMetadata(Base):
+    """
+    InstanceMetadata
+      Metadata information about an instance in a HyperPod cluster.
+
+    Attributes
+    ----------------------
+    customer_eni: The ID of the customer-managed Elastic Network Interface (ENI) associated with the instance.
+    additional_enis: Information about additional Elastic Network Interfaces (ENIs) associated with the instance.
+    capacity_reservation: Information about the Capacity Reservation used by the instance.
+    failure_message: An error message describing why the instance creation or update failed, if applicable.
+    lcs_execution_state: The execution state of the Lifecycle Script (LCS) for the instance.
+    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.
+    """
+
+    customer_eni: Optional[str] = Unassigned()
+    additional_enis: Optional[AdditionalEnis] = Unassigned()
+    capacity_reservation: Optional[CapacityReservation] = Unassigned()
+    failure_message: Optional[str] = Unassigned()
+    lcs_execution_state: Optional[str] = Unassigned()
+    node_logical_id: Optional[str] = Unassigned()
+
+
+class EventMetadata(Base):
+    """
+    EventMetadata
+      Metadata associated with a cluster event, which may include details about various resource types.
+
+    Attributes
+    ----------------------
+    cluster: Metadata specific to cluster-level events.
+    instance_group: Metadata specific to instance group-level events.
+    instance_group_scaling: Metadata related to instance group scaling events.
+    instance: Metadata specific to instance-level events.
+    """
+
+    cluster: Optional[ClusterMetadata] = Unassigned()
+    instance_group: Optional[InstanceGroupMetadata] = Unassigned()
+    instance_group_scaling: Optional[InstanceGroupScalingMetadata] = Unassigned()
+    instance: Optional[InstanceMetadata] = Unassigned()
+
+
+class EventDetails(Base):
+    """
+    EventDetails
+      Detailed information about a specific event, including event metadata.
+
+    Attributes
+    ----------------------
+    event_metadata: Metadata specific to the event, which may include information about the cluster, instance group, or instance involved.
+    """
+
+    event_metadata: Optional[EventMetadata] = Unassigned()
+
+
+class ClusterEventDetail(Base):
+    """
+    ClusterEventDetail
+      Detailed information about a specific event in a HyperPod cluster.
+
+    Attributes
+    ----------------------
+    event_id: The unique identifier (UUID) of the event.
+    cluster_arn: The Amazon Resource Name (ARN) of the SageMaker HyperPod cluster associated with the event.
+    cluster_name: The name of the SageMaker HyperPod cluster associated with the event.
+    instance_group_name: The name of the instance group associated with the event, if applicable.
+    instance_id: The EC2 instance ID associated with the event, if applicable.
+    resource_type: The type of resource associated with the event. Valid values are "Cluster", "InstanceGroup", or "Instance".
+    event_time: The timestamp when the event occurred.
+    event_details: Additional details about the event, including event-specific metadata.
+    description: A human-readable description of the event.
+    """
+
+    event_id: str
+    cluster_arn: str
+    cluster_name: Union[str, object]
+    resource_type: str
+    event_time: datetime.datetime
+    instance_group_name: Optional[str] = Unassigned()
+    instance_id: Optional[str] = Unassigned()
+    event_details: Optional[EventDetails] = Unassigned()
+    description: Optional[str] = Unassigned()
+
+
+class ClusterEventSummary(Base):
+    """
+    ClusterEventSummary
+      A summary of an event in a SageMaker HyperPod cluster.
+
+    Attributes
+    ----------------------
+    event_id: The unique identifier (UUID) of the event.
+    cluster_arn: The Amazon Resource Name (ARN) of the SageMaker HyperPod cluster associated with the event.
+    cluster_name: The name of the SageMaker HyperPod cluster associated with the event.
+    instance_group_name: The name of the instance group associated with the event, if applicable.
+    instance_id: The EC2 instance ID associated with the event, if applicable.
+    resource_type: The type of resource associated with the event. Valid values are "Cluster", "InstanceGroup", or "Instance".
+    event_time: The timestamp when the event occurred.
+    description: A brief, human-readable description of the event.
+    """
+
+    event_id: str
+    cluster_arn: str
+    cluster_name: Union[str, object]
+    resource_type: str
+    event_time: datetime.datetime
+    instance_group_name: Optional[str] = Unassigned()
+    instance_id: Optional[str] = Unassigned()
+    description: Optional[str] = Unassigned()
+
+
 class ClusterLifeCycleConfig(Base):
     """
     ClusterLifeCycleConfig
@@ -3383,6 +3651,8 @@ class ClusterInstanceGroupDetails(Base):
     training_plan_status: The current status of the training plan associated with this cluster instance group.
     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.
     scheduled_update_config: The configuration object of the schedule that SageMaker follows when updating the AMI.
+    current_image_id: The ID of the Amazon Machine Image (AMI) currently in use by the instance group.
+    desired_image_id: The ID of the Amazon Machine Image (AMI) desired for the instance group.
     """
 
     current_count: Optional[int] = Unassigned()
@@ -3399,6 +3669,8 @@ class ClusterInstanceGroupDetails(Base):
     training_plan_status: Optional[str] = Unassigned()
     override_vpc_config: Optional[VpcConfig] = Unassigned()
     scheduled_update_config: Optional[ScheduledUpdateConfig] = Unassigned()
+    current_image_id: Optional[str] = Unassigned()
+    desired_image_id: Optional[str] = Unassigned()
 
 
 class ClusterInstanceGroupSpecification(Base):
@@ -3419,6 +3691,7 @@ class ClusterInstanceGroupSpecification(Base):
     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 .
     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.
     scheduled_update_config: The configuration object of the schedule that SageMaker uses to update the AMI.
+    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.
     """
 
     instance_count: int
@@ -3432,6 +3705,7 @@ class ClusterInstanceGroupSpecification(Base):
     training_plan_arn: Optional[str] = Unassigned()
     override_vpc_config: Optional[VpcConfig] = Unassigned()
     scheduled_update_config: Optional[ScheduledUpdateConfig] = Unassigned()
+    image_id: Optional[str] = Unassigned()
 
 
 class ClusterInstancePlacement(Base):
@@ -3473,6 +3747,7 @@ class ClusterNodeDetails(Base):
     ----------------------
     instance_group_name: The instance group name in which the instance is.
     instance_id: The ID of the instance.
+    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.
     instance_status: The status of the instance.
     instance_type: The type of the instance.
     launch_time: The time when the instance is launched.
@@ -3485,10 +3760,13 @@ class ClusterNodeDetails(Base):
     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.
     private_dns_hostname: The private DNS hostname of the SageMaker HyperPod cluster node.
     placement: The placement details of the SageMaker HyperPod cluster node.
+    current_image_id: The ID of the Amazon Machine Image (AMI) currently in use by the node.
+    desired_image_id: The ID of the Amazon Machine Image (AMI) desired for the node.
     """
 
     instance_group_name: Optional[str] = Unassigned()
     instance_id: Optional[str] = Unassigned()
+    node_logical_id: Optional[str] = Unassigned()
     instance_status: Optional[ClusterInstanceStatusDetails] = Unassigned()
     instance_type: Optional[str] = Unassigned()
     launch_time: Optional[datetime.datetime] = Unassigned()
@@ -3501,6 +3779,8 @@ class ClusterNodeDetails(Base):
     private_primary_ipv6: Optional[str] = Unassigned()
     private_dns_hostname: Optional[str] = Unassigned()
     placement: Optional[ClusterInstancePlacement] = Unassigned()
+    current_image_id: Optional[str] = Unassigned()
+    desired_image_id: Optional[str] = Unassigned()
 
 
 class ClusterNodeSummary(Base):
@@ -3512,6 +3792,7 @@ class ClusterNodeSummary(Base):
     ----------------------
     instance_group_name: The name of the instance group in which the instance is.
     instance_id: The ID of the instance.
+    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.
     instance_type: The type of the instance.
     launch_time: The time when the instance is launched.
     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):
     instance_type: str
     launch_time: datetime.datetime
     instance_status: ClusterInstanceStatusDetails
+    node_logical_id: Optional[str] = Unassigned()
     last_software_update_time: Optional[datetime.datetime] = Unassigned()
 
 
@@ -5045,7 +5327,7 @@ class UnifiedStudioSettings(Base):
     project_id: The ID of the Amazon SageMaker Unified Studio project that corresponds to the domain.
     environment_id: The ID of the environment that Amazon SageMaker Unified Studio associates with the domain.
     project_s3_path: The location where Amazon S3 stores temporary execution data and other artifacts for the project that corresponds to the domain.
-    single_sign_on_application_arn: The ARN of the application managed by SageMaker AI and SageMaker Unified Studio in the Amazon Web Services IAM Identity Center.
+    single_sign_on_application_arn: The ARN of the Amazon DataZone application managed by Amazon SageMaker Unified Studio in the Amazon Web Services IAM Identity Center.
     """
 
     studio_web_portal_access: Optional[str] = Unassigned()

sagemaker_core/main/utils.py

@@ -146,7 +146,7 @@ def get_textual_rich_theme() -> Theme:
 textual_rich_console_and_traceback_enabled = False
 
 
-def enable_textual_rich_console_and_traceback():
+def enable_rich_logging():
     """
     Reconfigure the global textual rich console with the customized theme
         and enable textual rich error traceback
@@ -180,7 +180,7 @@ def get_textual_rich_logger(name: str, log_level: str = "INFO") -> logging.Logge
         logging.Logger: A textial rich logger.
 
     """
-    enable_textual_rich_console_and_traceback()
+    enable_rich_logging()
     handler = get_rich_handler()
     logger = logging.getLogger(name)
     for handler in logger.handlers:

{sagemaker_core-1.0.47.dist-info → sagemaker_core-1.0.49.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.47
+Version: 1.0.49
 Summary: An python package for sagemaker core functionalities
 Author-email: AWS <sagemaker-interests@amazon.com>
 Project-URL: Repository, https://github.com/aws/sagemaker-core.git

{sagemaker_core-1.0.47.dist-info → sagemaker_core-1.0.49.dist-info}/RECORD

@@ -1,4 +1,4 @@
-sagemaker_core/__init__.py,sha256=kM7971seaG8p8wFv2T0kOrFLRBhjOpjAHo3DGCaqRHc,126
+sagemaker_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sagemaker_core/_version.py,sha256=4UH5LevZBa3Kl1OiDbkaeIzXUQNSYVFrc-2su2TCco4,86
 sagemaker_core/helper/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sagemaker_core/helper/session_helper.py,sha256=GO1UJgpN1L9a25nYlVb-KWk4KvmFzVkLqFMqw-VaI4c,33126
@@ -7,15 +7,15 @@ sagemaker_core/main/config_schema.py,sha256=Wxe2gJash1rrxBomGhSYmII1LmJ3E70LIuSW
 sagemaker_core/main/default_configs_helper.py,sha256=bg_tgczX_bYzNiSlalJ6TWPTgrQYsI0uZguP5TIbPiw,8324
 sagemaker_core/main/exceptions.py,sha256=CsiM3V_Gb16grBotnu59LB6tznryPcSvAQDAOOYGc10,5563
 sagemaker_core/main/logs.py,sha256=yfEH7uP91nbE1lefymOlBr81ziBzsDSIOF2Qyd54FJE,6241
-sagemaker_core/main/resources.py,sha256=lDuKu3EHRhEjOE6MVLHfPABxGzBNHh8R2j961ITdMA0,1437580
-sagemaker_core/main/shapes.py,sha256=P1uS3zyLHSgrOU87gAPv4pmLCkE4mQyK0ORgVkoqq9I,762937
+sagemaker_core/main/resources.py,sha256=nldjn7RZ2C0Myq-R-nvYBjouuQFPAJ0CrAhA3e61cNI,1440403
+sagemaker_core/main/shapes.py,sha256=cjTQKK-UZg_SkNEDAR8EUlul-046-s14pD6zt1PEo4I,776265
 sagemaker_core/main/user_agent.py,sha256=BPYDAfDd70ObP-VAjl7aDHALHyGknkpRP21ktVr_LDw,2744
-sagemaker_core/main/utils.py,sha256=Ge_KdFKWo-qoIXlXkgLbaJbfXMVdUVHTq6BbDQn48Tw,19201
+sagemaker_core/main/utils.py,sha256=y1aZzztCApczZSqT0U7_H8zLlpqp1vgoUKcp40mTM2o,19157
 sagemaker_core/main/code_injection/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sagemaker_core/main/code_injection/base.py,sha256=11_Jif0nOzfbLGlXaacKf-wcizzfS64U0OSZGoVffFU,1733
 sagemaker_core/main/code_injection/codec.py,sha256=2YzJ-iYEzmguVaJVcZeyCR0OpTSR7UOixATrOm4MiBk,8885
 sagemaker_core/main/code_injection/constants.py,sha256=2ICExGge8vAWx7lSTW0JGh-bH1korkvpOpDu5M63eI4,980
-sagemaker_core/main/code_injection/shape_dag.py,sha256=4vVwP5AlNSeGqX8_Ye5aRmLYKNpRiQOUjDT-2Xt7PYQ,724635
+sagemaker_core/main/code_injection/shape_dag.py,sha256=W33MDZhfRmGo1EMI25QVbQyqdc9N1lb50Lr1mnErDxY,737697
 sagemaker_core/resources/__init__.py,sha256=EAYTFMN-nPjnPjjBbhIUeaL67FLKNPd7qbcbl9VIrws,31
 sagemaker_core/shapes/__init__.py,sha256=RnbIu9eTxKt-DNsOFJabrWIgrrtS9_SdAozP9JBl_ic,28
 sagemaker_core/tools/__init__.py,sha256=xX79JImxCVzrWMnjgntLCve2G5I-R4pRar5s20kT9Rs,56
@@ -28,8 +28,8 @@ sagemaker_core/tools/resources_extractor.py,sha256=hN61ehZbPnhFW-2FIVDi7NsEz4rLv
 sagemaker_core/tools/shapes_codegen.py,sha256=4lsePZpjk7M6RpJs5yar_m4z5MzwGHFrvCkdS_-R12c,12172
 sagemaker_core/tools/shapes_extractor.py,sha256=vxVKjXD3lmjrkoKiexjUnOt8ITbFxQSeiDtx7P6Qtkw,14226
 sagemaker_core/tools/templates.py,sha256=0lOIH3Rq2CXWkQhK6VenN_TE_v5p852s2kQyb_BeQxA,23460
-sagemaker_core-1.0.47.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.47.dist-info/METADATA,sha256=7rgdyuAZWa8wvlnIFRZaqiQldgDIyITqyY7Y2lNdlz0,4871
-sagemaker_core-1.0.47.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sagemaker_core-1.0.47.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.47.dist-info/RECORD,,
+sagemaker_core-1.0.49.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sagemaker_core-1.0.49.dist-info/METADATA,sha256=zHe0l5x-M6jMwtIEYhbhoVO82faXhsgNHaf0XmUL5GU,4871
+sagemaker_core-1.0.49.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sagemaker_core-1.0.49.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
+sagemaker_core-1.0.49.dist-info/RECORD,,