sagemaker-core 1.0.30__py3-none-any.whl → 1.0.32__py3-none-any.whl

sagemaker_core/main/code_injection/shape_dag.py

@@ -124,6 +124,10 @@ SHAPE_DAG = {
         "members": [{"name": "AlarmName", "shape": "AlarmName", "type": "string"}],
         "type": "structure",
     },
+    "AlarmDetails": {
+        "members": [{"name": "AlarmName", "shape": "AlarmName", "type": "string"}],
+        "type": "structure",
+    },
     "AlarmList": {"member_shape": "Alarm", "member_type": "structure", "type": "list"},
     "AlgorithmSpecification": {
         "members": [
@@ -734,6 +738,11 @@ SHAPE_DAG = {
         "type": "structure",
     },
     "AutoParameters": {"member_shape": "AutoParameter", "member_type": "structure", "type": "list"},
+    "AutoRollbackAlarms": {
+        "member_shape": "AlarmDetails",
+        "member_type": "structure",
+        "type": "list",
+    },
     "AutoRollbackConfig": {
         "members": [{"name": "Alarms", "shape": "AlarmList", "type": "list"}],
         "type": "structure",
@@ -1069,6 +1078,13 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "CapacitySizeConfig": {
+        "members": [
+            {"name": "Type", "shape": "NodeUnavailabilityType", "type": "string"},
+            {"name": "Value", "shape": "NodeUnavailabilityValue", "type": "integer"},
+        ],
+        "type": "structure",
+    },
     "CaptureContentTypeHeader": {
         "members": [
             {"name": "CsvContentTypes", "shape": "CsvContentTypes", "type": "list"},
@@ -1272,6 +1288,11 @@ SHAPE_DAG = {
                 "type": "string",
             },
             {"name": "OverrideVpcConfig", "shape": "VpcConfig", "type": "structure"},
+            {
+                "name": "ScheduledUpdateConfig",
+                "shape": "ScheduledUpdateConfig",
+                "type": "structure",
+            },
         ],
         "type": "structure",
     },
@@ -1296,6 +1317,11 @@ SHAPE_DAG = {
             {"name": "OnStartDeepHealthChecks", "shape": "OnStartDeepHealthChecks", "type": "list"},
             {"name": "TrainingPlanArn", "shape": "TrainingPlanArn", "type": "string"},
             {"name": "OverrideVpcConfig", "shape": "VpcConfig", "type": "structure"},
+            {
+                "name": "ScheduledUpdateConfig",
+                "shape": "ScheduledUpdateConfig",
+                "type": "structure",
+            },
         ],
         "type": "structure",
     },
@@ -1352,6 +1378,7 @@ SHAPE_DAG = {
             },
             {"name": "InstanceType", "shape": "ClusterInstanceType", "type": "string"},
             {"name": "LaunchTime", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "LastSoftwareUpdateTime", "shape": "Timestamp", "type": "timestamp"},
             {"name": "LifeCycleConfig", "shape": "ClusterLifeCycleConfig", "type": "structure"},
             {"name": "OverrideVpcConfig", "shape": "VpcConfig", "type": "structure"},
             {"name": "ThreadsPerCore", "shape": "ClusterThreadsPerCore", "type": "integer"},
@@ -1379,6 +1406,7 @@ SHAPE_DAG = {
             {"name": "InstanceId", "shape": "String", "type": "string"},
             {"name": "InstanceType", "shape": "ClusterInstanceType", "type": "string"},
             {"name": "LaunchTime", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "LastSoftwareUpdateTime", "shape": "Timestamp", "type": "timestamp"},
             {
                 "name": "InstanceStatus",
                 "shape": "ClusterInstanceStatusDetails",
@@ -3889,6 +3917,22 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "DeploymentConfiguration": {
+        "members": [
+            {
+                "name": "RollingUpdatePolicy",
+                "shape": "RollingDeploymentPolicy",
+                "type": "structure",
+            },
+            {
+                "name": "WaitIntervalInSeconds",
+                "shape": "WaitTimeIntervalInSeconds",
+                "type": "integer",
+            },
+            {"name": "AutoRollbackConfiguration", "shape": "AutoRollbackAlarms", "type": "list"},
+        ],
+        "type": "structure",
+    },
     "DeploymentRecommendation": {
         "members": [
             {"name": "RecommendationStatus", "shape": "RecommendationStatus", "type": "string"},
@@ -6138,6 +6182,11 @@ SHAPE_DAG = {
             },
             {"name": "DockerSettings", "shape": "DockerSettings", "type": "structure"},
             {"name": "AmazonQSettings", "shape": "AmazonQSettings", "type": "structure"},
+            {
+                "name": "UnifiedStudioSettings",
+                "shape": "UnifiedStudioSettings",
+                "type": "structure",
+            },
         ],
         "type": "structure",
     },
@@ -6156,6 +6205,11 @@ SHAPE_DAG = {
             {"name": "SecurityGroupIds", "shape": "DomainSecurityGroupIds", "type": "list"},
             {"name": "DockerSettings", "shape": "DockerSettings", "type": "structure"},
             {"name": "AmazonQSettings", "shape": "AmazonQSettings", "type": "structure"},
+            {
+                "name": "UnifiedStudioSettings",
+                "shape": "UnifiedStudioSettings",
+                "type": "structure",
+            },
         ],
         "type": "structure",
     },
@@ -13174,6 +13228,17 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "RollingDeploymentPolicy": {
+        "members": [
+            {"name": "MaximumBatchSize", "shape": "CapacitySizeConfig", "type": "structure"},
+            {
+                "name": "RollbackMaximumBatchSize",
+                "shape": "CapacitySizeConfig",
+                "type": "structure",
+            },
+        ],
+        "type": "structure",
+    },
     "RollingUpdatePolicy": {
         "members": [
             {"name": "MaximumBatchSize", "shape": "CapacitySize", "type": "structure"},
@@ -13280,6 +13345,13 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "ScheduledUpdateConfig": {
+        "members": [
+            {"name": "ScheduleExpression", "shape": "CronScheduleExpression", "type": "string"},
+            {"name": "DeploymentConfig", "shape": "DeploymentConfiguration", "type": "structure"},
+        ],
+        "type": "structure",
+    },
     "SchedulerConfig": {
         "members": [
             {"name": "PriorityClasses", "shape": "PriorityClassList", "type": "list"},
@@ -13627,6 +13699,7 @@ SHAPE_DAG = {
             },
             {"name": "AppType", "shape": "AppType", "type": "string"},
             {"name": "SpaceStorageSettings", "shape": "SpaceStorageSettings", "type": "structure"},
+            {"name": "SpaceManagedResources", "shape": "FeatureStatus", "type": "string"},
             {"name": "CustomFileSystems", "shape": "CustomFileSystems", "type": "list"},
         ],
         "type": "structure",
@@ -14746,6 +14819,18 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "UnifiedStudioSettings": {
+        "members": [
+            {"name": "StudioWebPortalAccess", "shape": "FeatureStatus", "type": "string"},
+            {"name": "DomainAccountId", "shape": "AccountId", "type": "string"},
+            {"name": "DomainRegion", "shape": "RegionName", "type": "string"},
+            {"name": "DomainId", "shape": "UnifiedStudioDomainId", "type": "string"},
+            {"name": "ProjectId", "shape": "UnifiedStudioProjectId", "type": "string"},
+            {"name": "EnvironmentId", "shape": "UnifiedStudioEnvironmentId", "type": "string"},
+            {"name": "ProjectS3Path", "shape": "S3Uri", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "UnprocessedIdentifiers": {
         "member_shape": "BatchGetRecordIdentifier",
         "member_type": "structure",
@@ -14856,8 +14941,27 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "UpdateClusterSoftwareInstanceGroupSpecification": {
+        "members": [
+            {"name": "InstanceGroupName", "shape": "ClusterInstanceGroupName", "type": "string"}
+        ],
+        "type": "structure",
+    },
+    "UpdateClusterSoftwareInstanceGroups": {
+        "member_shape": "UpdateClusterSoftwareInstanceGroupSpecification",
+        "member_type": "structure",
+        "type": "list",
+    },
     "UpdateClusterSoftwareRequest": {
-        "members": [{"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"}],
+        "members": [
+            {"name": "ClusterName", "shape": "ClusterNameOrArn", "type": "string"},
+            {
+                "name": "InstanceGroups",
+                "shape": "UpdateClusterSoftwareInstanceGroups",
+                "type": "list",
+            },
+            {"name": "DeploymentConfig", "shape": "DeploymentConfiguration", "type": "structure"},
+        ],
         "type": "structure",
     },
     "UpdateClusterSoftwareResponse": {

sagemaker_core/main/config_schema.py

@@ -243,6 +243,9 @@ SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA = {
                                                 "domain_execution_role_arn": {"type": "string"}
                                             },
                                             "execution_role_identity_config": {"type": "string"},
+                                            "unified_studio_settings": {
+                                                "project_s3_path": {"type": "string"}
+                                            },
                                         },
                                         "home_efs_file_system_kms_key_id": {"type": "string"},
                                         "subnet_ids": {

sagemaker_core/main/resources.py

@@ -3833,6 +3833,7 @@ class Cluster(Base):
     @Base.add_validate_call
     def update_software(
         self,
+        deployment_config: Optional[shapes.DeploymentConfiguration] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> None:
@@ -3840,6 +3841,7 @@ class Cluster(Base):
         Updates the platform software of a SageMaker HyperPod cluster for security patching.
 
         Parameters:
+            deployment_config: The configuration to use when updating the AMI versions.
             session: Boto3 session.
             region: Region name.
 
@@ -3859,6 +3861,8 @@ class Cluster(Base):
 
         operation_input_args = {
             "ClusterName": self.cluster_name,
+            "InstanceGroups": self.instance_groups,
+            "DeploymentConfig": deployment_config,
         }
         # serialize the input request
         operation_input_args = serialize(operation_input_args)
@@ -25812,9 +25816,9 @@ class ProcessingJob(Base):
             processing_inputs: An array of inputs configuring the data to download into the processing container.
             processing_output_config: Output configuration for the processing job.
             stopping_condition: The time limit for how long the processing job is allowed to run.
-            environment: The environment variables to set in the Docker container. Up to 100 key and values entries in the map are supported.
+            environment: The environment variables to set in the Docker container. Up to 100 key and values entries in the map are supported.  Do not include any security-sensitive information including account access IDs, secrets, or tokens in any environment fields. As part of the shared responsibility model, you are responsible for any potential exposure, unauthorized access, or compromise of your sensitive data if caused by security-sensitive information included in the request environment variable or plain text fields.
             network_config: Networking options for a processing job, such as whether to allow inbound and outbound network calls to and from processing containers, and the VPC subnets and security groups to use for VPC-enabled processing jobs.
-            tags: (Optional) An array of key-value pairs. For more information, see Using Cost Allocation Tags in the Amazon Web Services Billing and Cost Management User Guide.
+            tags: (Optional) An array of key-value pairs. For more information, see Using Cost Allocation Tags in the Amazon Web Services Billing and Cost Management User Guide.  Do not include any security-sensitive information including account access IDs, secrets, or tokens in any tags. As part of the shared responsibility model, you are responsible for any potential exposure, unauthorized access, or compromise of your sensitive data if caused by security-sensitive information included in the request tag variable or plain text fields.
             experiment_config:
             session: Boto3 session.
             region: Region name.
@@ -27979,7 +27983,7 @@ class TrainingJob(Base):
         profiler_rule_configurations: Configuration information for Amazon SageMaker Debugger rules for profiling system and framework metrics.
         profiler_rule_evaluation_statuses: Evaluation status of Amazon SageMaker Debugger rules for profiling on a training job.
         profiling_status: Profiling status of a training job.
-        environment: The environment variables to set in the Docker container.
+        environment: The environment variables to set in the Docker container.  Do not include any security-sensitive information including account access IDs, secrets, or tokens in any environment fields. As part of the shared responsibility model, you are responsible for any potential exposure, unauthorized access, or compromise of your sensitive data if caused by security-sensitive information included in the request environment variable or plain text fields.
         retry_strategy: The number of times to retry the job when the job fails due to an InternalServerError.
         remote_debug_config: Configuration for remote debugging. To learn more about the remote debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) for remote debugging.
         infra_check_config: Contains information about the infrastructure health check configuration for the training job.
@@ -28122,10 +28126,10 @@ class TrainingJob(Base):
             output_data_config: Specifies the path to the S3 location where you want to store model artifacts. SageMaker creates subfolders for the artifacts.
             resource_config: The resources, including the ML compute instances and ML storage volumes, to use for model training.  ML storage volumes store model artifacts and incremental states. Training algorithms might also use ML storage volumes for scratch space. If you want SageMaker to use the ML storage volume to store the training data, choose File as the TrainingInputMode in the algorithm specification. For distributed training algorithms, specify an instance count greater than 1.
             stopping_condition: Specifies a limit to how long a model training job can run. It also specifies how long a managed Spot training job has to complete. When the job reaches the time limit, SageMaker ends the training job. Use this API to cap model training costs. To stop a job, SageMaker sends the algorithm the SIGTERM signal, which delays job termination for 120 seconds. Algorithms can use this 120-second window to save the model artifacts, so the results of training are not lost.
-            hyper_parameters: Algorithm-specific parameters that influence the quality of the model. You set hyperparameters before you start the learning process. For a list of hyperparameters for each training algorithm provided by SageMaker, see Algorithms.  You can specify a maximum of 100 hyperparameters. Each hyperparameter is a key-value pair. Each key and value is limited to 256 characters, as specified by the Length Constraint.   Do not include any security-sensitive information including account access IDs, secrets or tokens in any hyperparameter field. If the use of security-sensitive credentials are detected, SageMaker will reject your training job request and return an exception error.
+            hyper_parameters: Algorithm-specific parameters that influence the quality of the model. You set hyperparameters before you start the learning process. For a list of hyperparameters for each training algorithm provided by SageMaker, see Algorithms.  You can specify a maximum of 100 hyperparameters. Each hyperparameter is a key-value pair. Each key and value is limited to 256 characters, as specified by the Length Constraint.   Do not include any security-sensitive information including account access IDs, secrets, or tokens in any hyperparameter fields. As part of the shared responsibility model, you are responsible for any potential exposure, unauthorized access, or compromise of your sensitive data if caused by any security-sensitive information included in the request hyperparameter variable or plain text fields.
             input_data_config: An array of Channel objects. Each channel is a named input source. InputDataConfig describes the input data and its location.  Algorithms can accept input data from one or more channels. For example, an algorithm might have two channels of input data, training_data and validation_data. The configuration for each channel provides the S3, EFS, or FSx location where the input data is stored. It also provides information about the stored data: the MIME type, compression method, and whether the data is wrapped in RecordIO format.  Depending on the input mode that the algorithm supports, SageMaker either copies input data files from an S3 bucket to a local directory in the Docker container, or makes it available as input streams. For example, if you specify an EFS location, input data files are available as input streams. They do not need to be downloaded. Your input must be in the same Amazon Web Services region as your training job.
             vpc_config: A VpcConfig object that specifies the VPC that you want your training job to connect to. Control access to and from your training container by configuring the VPC. For more information, see Protect Training Jobs by Using an Amazon Virtual Private Cloud.
-            tags: An array of key-value pairs. You can use tags to categorize your Amazon Web Services resources in different ways, for example, by purpose, owner, or environment. For more information, see Tagging Amazon Web Services Resources.
+            tags: An array of key-value pairs. You can use tags to categorize your Amazon Web Services resources in different ways, for example, by purpose, owner, or environment. For more information, see Tagging Amazon Web Services Resources.  Do not include any security-sensitive information including account access IDs, secrets, or tokens in any tags. As part of the shared responsibility model, you are responsible for any potential exposure, unauthorized access, or compromise of your sensitive data if caused by any security-sensitive information included in the request tag variable or plain text fields.
             enable_network_isolation: Isolates the training container. No inbound or outbound network calls can be made, except for calls between peers within a training cluster for distributed training. If you enable network isolation for training jobs that are configured to use a VPC, SageMaker downloads and uploads customer data and model artifacts through the specified VPC, but the training container does not have network access.
             enable_inter_container_traffic_encryption: To encrypt all communications between ML compute instances in distributed training, choose True. Encryption provides greater security for distributed training, but training might take longer. How long it takes depends on the amount of communication between compute instances, especially if you use a deep learning algorithm in distributed training. For more information, see Protect Communications Between ML Compute Instances in a Distributed Training Job.
             enable_managed_spot_training: To train models using managed spot training, choose True. Managed spot training provides a fully managed and scalable infrastructure for training machine learning models. this option is useful when training jobs can be interrupted and when there is flexibility when the training job is run.  The complete and intermediate results of jobs are stored in an Amazon S3 bucket, and can be used as a starting point to train models incrementally. Amazon SageMaker provides metrics and logs in CloudWatch. They can be used to see when managed spot training jobs are running, interrupted, resumed, or completed.
@@ -28136,7 +28140,7 @@ class TrainingJob(Base):
             experiment_config:
             profiler_config:
             profiler_rule_configurations: Configuration information for Amazon SageMaker Debugger rules for profiling system and framework metrics.
-            environment: The environment variables to set in the Docker container.
+            environment: The environment variables to set in the Docker container.  Do not include any security-sensitive information including account access IDs, secrets, or tokens in any environment fields. As part of the shared responsibility model, you are responsible for any potential exposure, unauthorized access, or compromise of your sensitive data if caused by security-sensitive information included in the request environment variable or plain text fields.
             retry_strategy: The number of times to retry the job when the job fails due to an InternalServerError.
             remote_debug_config: Configuration for remote debugging. To learn more about the remote debugging functionality of SageMaker, see Access a training container through Amazon Web Services Systems Manager (SSM) for remote debugging.
             infra_check_config: Contains information about the infrastructure health check configuration for the training job.

sagemaker_core/main/shapes.py

@@ -712,6 +712,19 @@ class Alarm(Base):
     alarm_name: Optional[str] = Unassigned()
 
 
+class AlarmDetails(Base):
+    """
+    AlarmDetails
+      The details of the alarm to monitor during the AMI update.
+
+    Attributes
+    ----------------------
+    alarm_name: The name of the alarm.
+    """
+
+    alarm_name: str
+
+
 class MetricDefinition(Base):
     """
     MetricDefinition
@@ -2873,6 +2886,21 @@ class CanvasAppSettings(Base):
     emr_serverless_settings: Optional[EmrServerlessSettings] = Unassigned()
 
 
+class CapacitySizeConfig(Base):
+    """
+    CapacitySizeConfig
+      The configuration of the size measurements of the AMI update. Using this configuration, you can specify whether SageMaker should update your instance group by an amount or percentage of instances.
+
+    Attributes
+    ----------------------
+    type: Specifies whether SageMaker should process the update by amount or percentage of instances.
+    value: Specifies the amount or percentage of instances SageMaker updates at a time.
+    """
+
+    type: str
+    value: int
+
+
 class CaptureContentTypeHeader(Base):
     """
     CaptureContentTypeHeader
@@ -3153,6 +3181,53 @@ class ClusterInstanceStorageConfig(Base):
     ebs_volume_config: Optional[ClusterEbsVolumeConfig] = Unassigned()
 
 
+class RollingDeploymentPolicy(Base):
+    """
+    RollingDeploymentPolicy
+      The configurations that SageMaker uses when updating the AMI versions.
+
+    Attributes
+    ----------------------
+    maximum_batch_size: The maximum amount of instances in the cluster that SageMaker can update at a time.
+    rollback_maximum_batch_size: The maximum amount of instances in the cluster that SageMaker can roll back at a time.
+    """
+
+    maximum_batch_size: CapacitySizeConfig
+    rollback_maximum_batch_size: Optional[CapacitySizeConfig] = Unassigned()
+
+
+class DeploymentConfiguration(Base):
+    """
+    DeploymentConfiguration
+      The configuration to use when updating the AMI versions.
+
+    Attributes
+    ----------------------
+    rolling_update_policy: The policy that SageMaker uses when updating the AMI versions of the cluster.
+    wait_interval_in_seconds: The duration in seconds that SageMaker waits before updating more instances in the cluster.
+    auto_rollback_configuration: An array that contains the alarms that SageMaker monitors to know whether to roll back the AMI update.
+    """
+
+    rolling_update_policy: Optional[RollingDeploymentPolicy] = Unassigned()
+    wait_interval_in_seconds: Optional[int] = Unassigned()
+    auto_rollback_configuration: Optional[List[AlarmDetails]] = Unassigned()
+
+
+class ScheduledUpdateConfig(Base):
+    """
+    ScheduledUpdateConfig
+      The configuration object of the schedule that SageMaker follows when updating the AMI.
+
+    Attributes
+    ----------------------
+    schedule_expression: A cron expression that specifies the schedule that SageMaker follows when updating the AMI.
+    deployment_config: The configuration to use when updating the AMI versions.
+    """
+
+    schedule_expression: str
+    deployment_config: Optional[DeploymentConfiguration] = Unassigned()
+
+
 class ClusterInstanceGroupDetails(Base):
     """
     ClusterInstanceGroupDetails
@@ -3173,6 +3248,7 @@ class ClusterInstanceGroupDetails(Base):
     training_plan_arn: The Amazon Resource Name (ARN); of the training plan associated with 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 .
     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_count: Optional[int] = Unassigned()
@@ -3188,6 +3264,7 @@ class ClusterInstanceGroupDetails(Base):
     training_plan_arn: Optional[str] = Unassigned()
     training_plan_status: Optional[str] = Unassigned()
     override_vpc_config: Optional[VpcConfig] = Unassigned()
+    scheduled_update_config: Optional[ScheduledUpdateConfig] = Unassigned()
 
 
 class ClusterInstanceGroupSpecification(Base):
@@ -3207,6 +3284,7 @@ class ClusterInstanceGroupSpecification(Base):
     on_start_deep_health_checks: A flag indicating whether deep health checks should be performed when the cluster instance group is created or updated.
     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.
     """
 
     instance_count: int
@@ -3219,6 +3297,7 @@ class ClusterInstanceGroupSpecification(Base):
     on_start_deep_health_checks: Optional[List[str]] = Unassigned()
     training_plan_arn: Optional[str] = Unassigned()
     override_vpc_config: Optional[VpcConfig] = Unassigned()
+    scheduled_update_config: Optional[ScheduledUpdateConfig] = Unassigned()
 
 
 class ClusterInstancePlacement(Base):
@@ -3263,6 +3342,7 @@ class ClusterNodeDetails(Base):
     instance_status: The status of the instance.
     instance_type: The type of the instance.
     launch_time: The time when the instance is launched.
+    last_software_update_time: The time when the cluster was last updated.
     life_cycle_config: The LifeCycle configuration applied to the instance.
     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.
     threads_per_core: The number of threads per CPU core you specified under CreateCluster.
@@ -3278,6 +3358,7 @@ class ClusterNodeDetails(Base):
     instance_status: Optional[ClusterInstanceStatusDetails] = Unassigned()
     instance_type: Optional[str] = Unassigned()
     launch_time: Optional[datetime.datetime] = Unassigned()
+    last_software_update_time: Optional[datetime.datetime] = Unassigned()
     life_cycle_config: Optional[ClusterLifeCycleConfig] = Unassigned()
     override_vpc_config: Optional[VpcConfig] = Unassigned()
     threads_per_core: Optional[int] = Unassigned()
@@ -3299,6 +3380,7 @@ class ClusterNodeSummary(Base):
     instance_id: The ID of the instance.
     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.
     instance_status: The status of the instance.
     """
 
@@ -3307,6 +3389,7 @@ class ClusterNodeSummary(Base):
     instance_type: str
     launch_time: datetime.datetime
     instance_status: ClusterInstanceStatusDetails
+    last_software_update_time: Optional[datetime.datetime] = Unassigned()
 
 
 class ClusterOrchestratorEksConfig(Base):
@@ -4682,6 +4765,31 @@ class DockerSettings(Base):
     vpc_only_trusted_accounts: Optional[List[str]] = Unassigned()
 
 
+class UnifiedStudioSettings(Base):
+    """
+    UnifiedStudioSettings
+      The settings that apply to an Amazon SageMaker AI domain when you use it in Amazon SageMaker Unified Studio.
+
+    Attributes
+    ----------------------
+    studio_web_portal_access: Sets whether you can access the domain in Amazon SageMaker Studio:  ENABLED  You can access the domain in Amazon SageMaker Studio. If you migrate the domain to Amazon SageMaker Unified Studio, you can access it in both studio interfaces.  DISABLED  You can't access the domain in Amazon SageMaker Studio. If you migrate the domain to Amazon SageMaker Unified Studio, you can access it only in that studio interface.   To migrate a domain to Amazon SageMaker Unified Studio, you specify the UnifiedStudioSettings data type when you use the UpdateDomain action.
+    domain_account_id: The ID of the Amazon Web Services account that has the Amazon SageMaker Unified Studio domain. The default value, if you don't specify an ID, is the ID of the account that has the Amazon SageMaker AI domain.
+    domain_region: The Amazon Web Services Region where the domain is located in Amazon SageMaker Unified Studio. The default value, if you don't specify a Region, is the Region where the Amazon SageMaker AI domain is located.
+    domain_id: The ID of the Amazon SageMaker Unified Studio domain associated with this domain.
+    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.
+    """
+
+    studio_web_portal_access: Optional[str] = Unassigned()
+    domain_account_id: Optional[str] = Unassigned()
+    domain_region: Optional[str] = Unassigned()
+    domain_id: Optional[str] = Unassigned()
+    project_id: Optional[str] = Unassigned()
+    environment_id: Optional[str] = Unassigned()
+    project_s3_path: Optional[str] = Unassigned()
+
+
 class DomainSettings(Base):
     """
     DomainSettings
@@ -4694,6 +4802,7 @@ class DomainSettings(Base):
     execution_role_identity_config: The configuration for attaching a SageMaker AI user profile name to the execution role as a sts:SourceIdentity key.
     docker_settings: A collection of settings that configure the domain's Docker interaction.
     amazon_q_settings: A collection of settings that configure the Amazon Q experience within the domain. The AuthMode that you use to create the domain must be SSO.
+    unified_studio_settings: The settings that apply to an SageMaker AI domain when you use it in Amazon SageMaker Unified Studio.
     """
 
     security_group_ids: Optional[List[str]] = Unassigned()
@@ -4701,6 +4810,7 @@ class DomainSettings(Base):
     execution_role_identity_config: Optional[str] = Unassigned()
     docker_settings: Optional[DockerSettings] = Unassigned()
     amazon_q_settings: Optional[AmazonQSettings] = Unassigned()
+    unified_studio_settings: Optional[UnifiedStudioSettings] = Unassigned()
 
 
 class DefaultSpaceSettings(Base):
@@ -7363,6 +7473,7 @@ class SpaceSettings(Base):
     jupyter_lab_app_settings: The settings for the JupyterLab application.
     app_type: The type of app created within the space. If using the  UpdateSpace API, you can't change the app type of your space by specifying a different value for this field.
     space_storage_settings: The storage settings for a space.
+    space_managed_resources: If you enable this option, SageMaker AI creates the following resources on your behalf when you create the space:   The user profile that possesses the space.   The app that the space contains.
     custom_file_systems: A file system, created by you, that you assign to a space for an Amazon SageMaker AI Domain. Permitted users can access this file system in Amazon SageMaker AI Studio.
     """
 
@@ -7372,6 +7483,7 @@ class SpaceSettings(Base):
     jupyter_lab_app_settings: Optional[SpaceJupyterLabAppSettings] = Unassigned()
     app_type: Optional[str] = Unassigned()
     space_storage_settings: Optional[SpaceStorageSettings] = Unassigned()
+    space_managed_resources: Optional[str] = Unassigned()
     custom_file_systems: Optional[List[CustomFileSystem]] = Unassigned()
 
 
@@ -9370,6 +9482,7 @@ class DomainSettingsForUpdate(Base):
     security_group_ids: The security groups for the Amazon Virtual Private Cloud that the Domain uses for communication between Domain-level apps and user apps.
     docker_settings: A collection of settings that configure the domain's Docker interaction.
     amazon_q_settings: A collection of settings that configure the Amazon Q experience within the domain.
+    unified_studio_settings: The settings that apply to an SageMaker AI domain when you use it in Amazon SageMaker Unified Studio.
     """
 
     r_studio_server_pro_domain_settings_for_update: Optional[
@@ -9379,6 +9492,7 @@ class DomainSettingsForUpdate(Base):
     security_group_ids: Optional[List[str]] = Unassigned()
     docker_settings: Optional[DockerSettings] = Unassigned()
     amazon_q_settings: Optional[AmazonQSettings] = Unassigned()
+    unified_studio_settings: Optional[UnifiedStudioSettings] = Unassigned()
 
 
 class PredefinedMetricSpecification(Base):
@@ -12600,6 +12714,19 @@ class ThroughputConfigUpdate(Base):
     provisioned_write_capacity_units: Optional[int] = Unassigned()
 
 
+class UpdateClusterSoftwareInstanceGroupSpecification(Base):
+    """
+    UpdateClusterSoftwareInstanceGroupSpecification
+      The configuration that describes specifications of the instance groups to update.
+
+    Attributes
+    ----------------------
+    instance_group_name: The name of the instance group to update.
+    """
+
+    instance_group_name: str
+
+
 class VariantProperty(Base):
     """
     VariantProperty

{sagemaker_core-1.0.30.dist-info → sagemaker_core-1.0.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.30
+Version: 1.0.32
 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.30.dist-info → sagemaker_core-1.0.32.dist-info}/RECORD

@@ -3,19 +3,19 @@ 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
 sagemaker_core/main/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sagemaker_core/main/config_schema.py,sha256=lBwIm5CT_dSXeW5i6cgRzbiLDjs0qtH1FrM78F-ma2U,57382
+sagemaker_core/main/config_schema.py,sha256=Wxe2gJash1rrxBomGhSYmII1LmJ3E70LIuSWklX4_7g,57588
 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/logs.py,sha256=yfEH7uP91nbE1lefymOlBr81ziBzsDSIOF2Qyd54FJE,6241
-sagemaker_core/main/resources.py,sha256=xWTgxtjjkxJJfCPbBetDkIp6gFAoOB49dgRYciKab0E,1418508
-sagemaker_core/main/shapes.py,sha256=HmG353jkAGcppGVxknZio_OMktb_uubY2sMpXivmlO8,732587
+sagemaker_core/main/resources.py,sha256=jNX5jp5tmwa0t8PYPxWrRRBAGIn0G-iq3C-7TvBxEPw,1420815
+sagemaker_core/main/shapes.py,sha256=YfcsZIpYetJPZHNo-VfajMCaqT6eGYXI_1EvCLr0dGg,739047
 sagemaker_core/main/user_agent.py,sha256=BPYDAfDd70ObP-VAjl7aDHALHyGknkpRP21ktVr_LDw,2744
 sagemaker_core/main/utils.py,sha256=qTGJDcZwrAQSsdyg8A78x4PKU4Wu1rY6Cn3OIbIspaA,18546
 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=nA51E9iNWHyKou1G23rKSRL4WitdkFRbMuFkyrGHzKU,8428
 sagemaker_core/main/code_injection/constants.py,sha256=2ICExGge8vAWx7lSTW0JGh-bH1korkvpOpDu5M63eI4,980
-sagemaker_core/main/code_injection/shape_dag.py,sha256=HufpRdLVicfx8fQVKlJqp-pkJ_v3rjmUQtcZhIktWxQ,700129
+sagemaker_core/main/code_injection/shape_dag.py,sha256=v4admWbDBIaum364qkuvq76WM8eOC0wOOfH8KpuA3Gc,704183
 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=vIgRWConRGAQ-Mri6LwfkArqWHlL3KXcvbbYa-t_MV4,23414
-sagemaker_core-1.0.30.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.30.dist-info/METADATA,sha256=_QIjOSMM3F06uAMrzR0ClBuj58lMeCryhrpnRsd4C2w,4885
-sagemaker_core-1.0.30.dist-info/WHEEL,sha256=ooBFpIzZCPdw3uqIQsOo4qqbA4ZRPxHnOH7peeONza0,91
-sagemaker_core-1.0.30.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.30.dist-info/RECORD,,
+sagemaker_core-1.0.32.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sagemaker_core-1.0.32.dist-info/METADATA,sha256=N-vlVke2GuSZYAnC04SYBHYOuDIMaGbWBgBYXcqVNgM,4885
+sagemaker_core-1.0.32.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+sagemaker_core-1.0.32.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
+sagemaker_core-1.0.32.dist-info/RECORD,,

{sagemaker_core-1.0.30.dist-info → sagemaker_core-1.0.32.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.0.1)
+Generator: setuptools (80.3.1)
 Root-Is-Purelib: true
 Tag: py3-none-any