sagemaker-core 1.0.3__py3-none-any.whl → 1.0.4__py3-none-any.whl

sagemaker_core/main/code_injection/shape_dag.py

@@ -1131,13 +1131,14 @@ SHAPE_DAG = {
             {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
             {"name": "InstanceType", "shape": "ClusterInstanceType", "type": "string"},
             {"name": "LifeCycleConfig", "shape": "ClusterLifeCycleConfig", "type": "structure"},
-            {"name": "ExecutionRole", "shape": "IAMRoleArn", "type": "string"},
+            {"name": "ExecutionRole", "shape": "RoleArn", "type": "string"},
             {"name": "ThreadsPerCore", "shape": "ClusterThreadsPerCore", "type": "integer"},
             {
                 "name": "InstanceStorageConfigs",
                 "shape": "ClusterInstanceStorageConfigs",
                 "type": "list",
             },
+            {"name": "OnStartDeepHealthChecks", "shape": "OnStartDeepHealthChecks", "type": "list"},
         ],
         "type": "structure",
     },
@@ -1152,13 +1153,14 @@ SHAPE_DAG = {
             {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"},
             {"name": "InstanceType", "shape": "ClusterInstanceType", "type": "string"},
             {"name": "LifeCycleConfig", "shape": "ClusterLifeCycleConfig", "type": "structure"},
-            {"name": "ExecutionRole", "shape": "IAMRoleArn", "type": "string"},
+            {"name": "ExecutionRole", "shape": "RoleArn", "type": "string"},
             {"name": "ThreadsPerCore", "shape": "ClusterThreadsPerCore", "type": "integer"},
             {
                 "name": "InstanceStorageConfigs",
                 "shape": "ClusterInstanceStorageConfigs",
                 "type": "list",
             },
+            {"name": "OnStartDeepHealthChecks", "shape": "OnStartDeepHealthChecks", "type": "list"},
         ],
         "type": "structure",
     },
@@ -1242,6 +1244,14 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "ClusterOrchestrator": {
+        "members": [{"name": "Eks", "shape": "ClusterOrchestratorEksConfig", "type": "structure"}],
+        "type": "structure",
+    },
+    "ClusterOrchestratorEksConfig": {
+        "members": [{"name": "ClusterArn", "shape": "EksClusterArn", "type": "string"}],
+        "type": "structure",
+    },
     "ClusterSummaries": {
         "member_shape": "ClusterSummary",
         "member_type": "structure",
@@ -1659,6 +1669,8 @@ SHAPE_DAG = {
             },
             {"name": "VpcConfig", "shape": "VpcConfig", "type": "structure"},
             {"name": "Tags", "shape": "TagList", "type": "list"},
+            {"name": "Orchestrator", "shape": "ClusterOrchestrator", "type": "structure"},
+            {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
         ],
         "type": "structure",
     },
@@ -3818,6 +3830,8 @@ SHAPE_DAG = {
             {"name": "FailureMessage", "shape": "String", "type": "string"},
             {"name": "InstanceGroups", "shape": "ClusterInstanceGroupDetailsList", "type": "list"},
             {"name": "VpcConfig", "shape": "VpcConfig", "type": "structure"},
+            {"name": "Orchestrator", "shape": "ClusterOrchestrator", "type": "structure"},
+            {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
         ],
         "type": "structure",
     },
@@ -10775,6 +10789,11 @@ SHAPE_DAG = {
         "members": [{"name": "Groups", "shape": "Groups", "type": "list"}],
         "type": "structure",
     },
+    "OnStartDeepHealthChecks": {
+        "member_shape": "DeepHealthCheckType",
+        "member_type": "string",
+        "type": "list",
+    },
     "OnlineStoreConfig": {
         "members": [
             {"name": "SecurityConfig", "shape": "OnlineStoreSecurityConfig", "type": "structure"},
@@ -13766,6 +13785,7 @@ SHAPE_DAG = {
                 "shape": "ClusterInstanceGroupSpecifications",
                 "type": "list",
             },
+            {"name": "NodeRecovery", "shape": "ClusterNodeRecovery", "type": "string"},
         ],
         "type": "structure",
     },

sagemaker_core/main/resources.py

@@ -3198,6 +3198,8 @@ class Cluster(Base):
         creation_time: The time when the SageMaker Cluster is created.
         failure_message: The failure message of the SageMaker HyperPod cluster.
         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.
 
     """
 
@@ -3208,6 +3210,8 @@ class Cluster(Base):
     failure_message: Optional[str] = Unassigned()
     instance_groups: Optional[List[ClusterInstanceGroupDetails]] = Unassigned()
     vpc_config: Optional[VpcConfig] = Unassigned()
+    orchestrator: Optional[ClusterOrchestrator] = Unassigned()
+    node_recovery: Optional[str] = Unassigned()
 
     def get_name(self) -> str:
         attributes = vars(self)
@@ -3252,6 +3256,8 @@ class Cluster(Base):
         instance_groups: List[ClusterInstanceGroupSpecification],
         vpc_config: Optional[VpcConfig] = Unassigned(),
         tags: Optional[List[Tag]] = Unassigned(),
+        orchestrator: Optional[ClusterOrchestrator] = Unassigned(),
+        node_recovery: Optional[str] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> Optional["Cluster"]:
@@ -3263,6 +3269,8 @@ class Cluster(Base):
             instance_groups: The instance groups to be created in the SageMaker HyperPod cluster.
             vpc_config:
             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.
             session: Boto3 session.
             region: Region name.
 
@@ -3296,6 +3304,8 @@ class Cluster(Base):
             "InstanceGroups": instance_groups,
             "VpcConfig": vpc_config,
             "Tags": tags,
+            "Orchestrator": orchestrator,
+            "NodeRecovery": node_recovery,
         }
 
         operation_input_args = Base.populate_chained_attributes(
@@ -3406,6 +3416,7 @@ class Cluster(Base):
     def update(
         self,
         instance_groups: List[ClusterInstanceGroupSpecification],
+        node_recovery: Optional[str] = Unassigned(),
     ) -> Optional["Cluster"]:
         """
         Update a Cluster resource
@@ -3434,6 +3445,7 @@ class Cluster(Base):
         operation_input_args = {
             "ClusterName": self.cluster_name,
             "InstanceGroups": instance_groups,
+            "NodeRecovery": node_recovery,
         }
         logger.debug(f"Input request: {operation_input_args}")
         # serialize the input request

sagemaker_core/main/shapes.py

@@ -2675,6 +2675,7 @@ class ClusterInstanceGroupDetails(Base):
     execution_role: The execution role for the instance group to assume.
     threads_per_core: The number you specified to TreadsPerCore in CreateCluster for enabling or disabling multithreading. For instance types that support multithreading, you can specify 1 for disabling multithreading and 2 for enabling multithreading. For more information, see the reference table of CPU cores and threads per CPU core per instance type in the Amazon Elastic Compute Cloud User Guide.
     instance_storage_configs: The additional storage configurations for the instances in the SageMaker HyperPod cluster instance group.
+    on_start_deep_health_checks: A flag indicating whether deep health checks should be performed when the cluster instance group is created or updated.
     """
 
     current_count: Optional[int] = Unassigned()
@@ -2685,6 +2686,7 @@ class ClusterInstanceGroupDetails(Base):
     execution_role: Optional[str] = Unassigned()
     threads_per_core: Optional[int] = Unassigned()
     instance_storage_configs: Optional[List[ClusterInstanceStorageConfig]] = Unassigned()
+    on_start_deep_health_checks: Optional[List[str]] = Unassigned()
 
 
 class ClusterInstanceGroupSpecification(Base):
@@ -2701,6 +2703,7 @@ class ClusterInstanceGroupSpecification(Base):
     execution_role: Specifies an IAM execution role to be assumed by the instance group.
     threads_per_core: Specifies the value for Threads per core. For instance types that support multithreading, you can specify 1 for disabling multithreading and 2 for enabling multithreading. For instance types that doesn't support multithreading, specify 1. For more information, see the reference table of CPU cores and threads per CPU core per instance type in the Amazon Elastic Compute Cloud User Guide.
     instance_storage_configs: Specifies the additional storage configurations for the instances in the SageMaker HyperPod cluster instance group.
+    on_start_deep_health_checks: A flag indicating whether deep health checks should be performed when the cluster instance group is created or updated.
     """
 
     instance_count: int
@@ -2710,6 +2713,7 @@ class ClusterInstanceGroupSpecification(Base):
     execution_role: str
     threads_per_core: Optional[int] = Unassigned()
     instance_storage_configs: Optional[List[ClusterInstanceStorageConfig]] = Unassigned()
+    on_start_deep_health_checks: Optional[List[str]] = Unassigned()
 
 
 class ClusterInstancePlacement(Base):
@@ -2796,6 +2800,32 @@ class ClusterNodeSummary(Base):
     instance_status: ClusterInstanceStatusDetails
 
 
+class ClusterOrchestratorEksConfig(Base):
+    """
+    ClusterOrchestratorEksConfig
+      The configuration settings for the Amazon EKS cluster used as the orchestrator for the SageMaker HyperPod cluster.
+
+    Attributes
+    ----------------------
+    cluster_arn: The Amazon Resource Name (ARN) of the Amazon EKS cluster associated with the SageMaker HyperPod cluster.
+    """
+
+    cluster_arn: str
+
+
+class ClusterOrchestrator(Base):
+    """
+    ClusterOrchestrator
+      The type of orchestrator used for the SageMaker HyperPod cluster.
+
+    Attributes
+    ----------------------
+    eks: The Amazon EKS cluster used as the orchestrator for the SageMaker HyperPod cluster.
+    """
+
+    eks: ClusterOrchestratorEksConfig
+
+
 class ClusterSummary(Base):
     """
     ClusterSummary

{sagemaker_core-1.0.3.dist-info → sagemaker_core-1.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.3
+Version: 1.0.4
 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.3.dist-info → sagemaker_core-1.0.4.dist-info}/RECORD

@@ -6,15 +6,15 @@ sagemaker_core/main/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 sagemaker_core/main/config_schema.py,sha256=IDvL1YuMOwVg4nb0D4ysU51tc1UjlQASyz1QZoncs7I,56907
 sagemaker_core/main/exceptions.py,sha256=87DUlrmHxaWoiYNlpNY9ixxFMPRk_dIGPsA2e_xdVwQ,5602
 sagemaker_core/main/intelligent_defaults_helper.py,sha256=5SDM6UavZtp-k5LhqRL7GRIDgzFB5UsC_p7YuiSPK9A,8334
-sagemaker_core/main/resources.py,sha256=_zix3PucNO60z3XVLEKCfSm5zWLyNWJaaGk868-rz5w,1298170
-sagemaker_core/main/shapes.py,sha256=NgrfvUwVDRIYHn1jB2NgDGQ631-S0cA22AlAE7Tw8FA,682244
+sagemaker_core/main/resources.py,sha256=3ivOSUiWqH8VWmOwxz1Oa45aet7CTsc8mZott-ge7yc,1299296
+sagemaker_core/main/shapes.py,sha256=mxsgFQYYMCWrZm_GF3w_wwgfTon0AgheZoXKu17ux1c,683409
 sagemaker_core/main/user_agent.py,sha256=4sZybDXkzRoZnOnVDQ8p8zFTfiRJdsH7amDWInVQ4xU,2708
 sagemaker_core/main/utils.py,sha256=I7WsC-DdbnAZg_PtqcLPywXlqjAwj7ZYbeHhv93Z1O4,18840
 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=2DjmeD2uND307UqDefvVEpE0rZ8yfFU3Bi3TvQCQveI,7658
 sagemaker_core/main/code_injection/constants.py,sha256=2ICExGge8vAWx7lSTW0JGh-bH1korkvpOpDu5M63eI4,980
-sagemaker_core/main/code_injection/shape_dag.py,sha256=-iWikIF2hoNnoVkJGfi_h3OTKqM4rygHIulaQjZrZ1g,646997
+sagemaker_core/main/code_injection/shape_dag.py,sha256=mQR562skpJVs3VbNS-dtGvF94JXVcRBZS5b0cpKH3x0,648111
 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
@@ -27,8 +27,8 @@ sagemaker_core/tools/resources_extractor.py,sha256=B9Twlf2vb7AkZUKLnxSdQZ_rNLPz0
 sagemaker_core/tools/shapes_codegen.py,sha256=_ve959bwH8usZ6dPlpXxi2on9t0hLpcmhRWnaWHCWMQ,11745
 sagemaker_core/tools/shapes_extractor.py,sha256=4KjgDmhlPM4G1f1NeYbORKlXs1s7Q_sm_NK31S_ROQ0,11950
 sagemaker_core/tools/templates.py,sha256=IvL9ykLvOXDNCMnIZn30o-6Cmx4fw-BIJz3nl9LD4dc,24614
-sagemaker_core-1.0.3.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.3.dist-info/METADATA,sha256=t4gnhUg41sSyQ0NtNQ2HGhce5TfXXlLC7aGXsGO6Usw,4877
-sagemaker_core-1.0.3.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
-sagemaker_core-1.0.3.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.3.dist-info/RECORD,,
+sagemaker_core-1.0.4.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sagemaker_core-1.0.4.dist-info/METADATA,sha256=62vOPMIkSDkfR_ekA4qjnxJ96kC4hbzlhAtd2w52Qk0,4877
+sagemaker_core-1.0.4.dist-info/WHEEL,sha256=cVxcB9AmuTcXqmwrtPhNK88dr7IR_b6qagTj0UvIEbY,91
+sagemaker_core-1.0.4.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
+sagemaker_core-1.0.4.dist-info/RECORD,,