sagemaker-core 1.0.39__tar.gz → 1.0.41__tar.gz

{sagemaker_core-1.0.39/src/sagemaker_core.egg-info → sagemaker_core-1.0.41}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.39
+Version: 1.0.41
 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.41/VERSION

@@ -0,0 +1 @@
+1.0.41

{sagemaker_core-1.0.39 → sagemaker_core-1.0.41}/src/sagemaker_core/main/code_injection/shape_dag.py

@@ -1134,6 +1134,77 @@ SHAPE_DAG = {
         "member_type": "structure",
         "type": "list",
     },
+    "CfnCreateTemplateProvider": {
+        "members": [
+            {"name": "TemplateName", "shape": "CfnTemplateName", "type": "string"},
+            {"name": "TemplateURL", "shape": "CfnTemplateURL", "type": "string"},
+            {"name": "RoleARN", "shape": "RoleArn", "type": "string"},
+            {"name": "Parameters", "shape": "CfnStackCreateParameters", "type": "list"},
+        ],
+        "type": "structure",
+    },
+    "CfnStackCreateParameter": {
+        "members": [
+            {"name": "Key", "shape": "CfnStackParameterKey", "type": "string"},
+            {"name": "Value", "shape": "CfnStackParameterValue", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "CfnStackCreateParameters": {
+        "member_shape": "CfnStackCreateParameter",
+        "member_type": "structure",
+        "type": "list",
+    },
+    "CfnStackDetail": {
+        "members": [
+            {"name": "Name", "shape": "CfnStackName", "type": "string"},
+            {"name": "Id", "shape": "CfnStackId", "type": "string"},
+            {"name": "StatusMessage", "shape": "CfnStackStatusMessage", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "CfnStackParameter": {
+        "members": [
+            {"name": "Key", "shape": "CfnStackParameterKey", "type": "string"},
+            {"name": "Value", "shape": "CfnStackParameterValue", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "CfnStackParameters": {
+        "member_shape": "CfnStackParameter",
+        "member_type": "structure",
+        "type": "list",
+    },
+    "CfnStackUpdateParameter": {
+        "members": [
+            {"name": "Key", "shape": "CfnStackParameterKey", "type": "string"},
+            {"name": "Value", "shape": "CfnStackParameterValue", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "CfnStackUpdateParameters": {
+        "member_shape": "CfnStackUpdateParameter",
+        "member_type": "structure",
+        "type": "list",
+    },
+    "CfnTemplateProviderDetail": {
+        "members": [
+            {"name": "TemplateName", "shape": "CfnTemplateName", "type": "string"},
+            {"name": "TemplateURL", "shape": "CfnTemplateURL", "type": "string"},
+            {"name": "RoleARN", "shape": "RoleArn", "type": "string"},
+            {"name": "Parameters", "shape": "CfnStackParameters", "type": "list"},
+            {"name": "StackDetail", "shape": "CfnStackDetail", "type": "structure"},
+        ],
+        "type": "structure",
+    },
+    "CfnUpdateTemplateProvider": {
+        "members": [
+            {"name": "TemplateName", "shape": "CfnTemplateName", "type": "string"},
+            {"name": "TemplateURL", "shape": "CfnTemplateURL", "type": "string"},
+            {"name": "Parameters", "shape": "CfnStackUpdateParameters", "type": "list"},
+        ],
+        "type": "structure",
+    },
     "Channel": {
         "members": [
             {"name": "ChannelName", "shape": "ChannelName", "type": "string"},
@@ -2975,6 +3046,7 @@ SHAPE_DAG = {
                 "type": "structure",
             },
             {"name": "Tags", "shape": "TagList", "type": "list"},
+            {"name": "TemplateProviders", "shape": "CreateTemplateProviderList", "type": "list"},
         ],
         "type": "structure",
     },
@@ -3032,6 +3104,21 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "CreateTemplateProvider": {
+        "members": [
+            {
+                "name": "CfnTemplateProvider",
+                "shape": "CfnCreateTemplateProvider",
+                "type": "structure",
+            }
+        ],
+        "type": "structure",
+    },
+    "CreateTemplateProviderList": {
+        "member_shape": "CreateTemplateProvider",
+        "member_type": "structure",
+        "type": "list",
+    },
     "CreateTrainingJobRequest": {
         "members": [
             {"name": "TrainingJobName", "shape": "TrainingJobName", "type": "string"},
@@ -5713,6 +5800,11 @@ SHAPE_DAG = {
                 "type": "structure",
             },
             {"name": "ProjectStatus", "shape": "ProjectStatus", "type": "string"},
+            {
+                "name": "TemplateProviderDetails",
+                "shape": "TemplateProviderDetailList",
+                "type": "list",
+            },
             {"name": "CreatedBy", "shape": "UserContext", "type": "structure"},
             {"name": "CreationTime", "shape": "Timestamp", "type": "timestamp"},
             {"name": "LastModifiedTime", "shape": "Timestamp", "type": "timestamp"},
@@ -12643,6 +12735,11 @@ SHAPE_DAG = {
             {"name": "ProjectStatus", "shape": "ProjectStatus", "type": "string"},
             {"name": "CreatedBy", "shape": "UserContext", "type": "structure"},
             {"name": "CreationTime", "shape": "Timestamp", "type": "timestamp"},
+            {
+                "name": "TemplateProviderDetails",
+                "shape": "TemplateProviderDetailList",
+                "type": "list",
+            },
             {"name": "Tags", "shape": "TagList", "type": "list"},
             {"name": "LastModifiedTime", "shape": "Timestamp", "type": "timestamp"},
             {"name": "LastModifiedBy", "shape": "UserContext", "type": "structure"},
@@ -14110,6 +14207,21 @@ SHAPE_DAG = {
         "type": "structure",
     },
     "TaskKeywords": {"member_shape": "TaskKeyword", "member_type": "string", "type": "list"},
+    "TemplateProviderDetail": {
+        "members": [
+            {
+                "name": "CfnTemplateProviderDetail",
+                "shape": "CfnTemplateProviderDetail",
+                "type": "structure",
+            }
+        ],
+        "type": "structure",
+    },
+    "TemplateProviderDetailList": {
+        "member_shape": "TemplateProviderDetail",
+        "member_type": "structure",
+        "type": "list",
+    },
     "TensorBoardAppSettings": {
         "members": [{"name": "DefaultResourceSpec", "shape": "ResourceSpec", "type": "structure"}],
         "type": "structure",
@@ -15596,6 +15708,11 @@ SHAPE_DAG = {
                 "type": "structure",
             },
             {"name": "Tags", "shape": "TagList", "type": "list"},
+            {
+                "name": "TemplateProvidersToUpdate",
+                "shape": "UpdateTemplateProviderList",
+                "type": "list",
+            },
         ],
         "type": "structure",
     },
@@ -15616,6 +15733,21 @@ SHAPE_DAG = {
         "members": [{"name": "SpaceArn", "shape": "SpaceArn", "type": "string"}],
         "type": "structure",
     },
+    "UpdateTemplateProvider": {
+        "members": [
+            {
+                "name": "CfnTemplateProvider",
+                "shape": "CfnUpdateTemplateProvider",
+                "type": "structure",
+            }
+        ],
+        "type": "structure",
+    },
+    "UpdateTemplateProviderList": {
+        "member_shape": "UpdateTemplateProvider",
+        "member_type": "structure",
+        "type": "list",
+    },
     "UpdateTrainingJobRequest": {
         "members": [
             {"name": "TrainingJobName", "shape": "TrainingJobName", "type": "string"},

{sagemaker_core-1.0.39 → sagemaker_core-1.0.41}/src/sagemaker_core/main/resources.py

@@ -3275,7 +3275,7 @@ class Cluster(Base):
     def create(
         cls,
         cluster_name: str,
-        instance_groups: List[shapes.ClusterInstanceGroupSpecification],
+        instance_groups: Optional[List[shapes.ClusterInstanceGroupSpecification]] = Unassigned(),
         vpc_config: Optional[shapes.VpcConfig] = Unassigned(),
         tags: Optional[List[shapes.Tag]] = Unassigned(),
         orchestrator: Optional[shapes.ClusterOrchestrator] = Unassigned(),
@@ -3437,7 +3437,7 @@ class Cluster(Base):
     @Base.add_validate_call
     def update(
         self,
-        instance_groups: List[shapes.ClusterInstanceGroupSpecification],
+        instance_groups: Optional[List[shapes.ClusterInstanceGroupSpecification]] = Unassigned(),
         node_recovery: Optional[str] = Unassigned(),
         instance_groups_to_delete: Optional[List[str]] = Unassigned(),
     ) -> Optional["Cluster"]:
@@ -3664,7 +3664,7 @@ class Cluster(Base):
         Parameters:
             creation_time_after: Set a start time for the time range during which you want to list SageMaker HyperPod clusters. Timestamps are formatted according to the ISO 8601 standard.  Acceptable formats include:    YYYY-MM-DDThh:mm:ss.sssTZD (UTC), for example, 2014-10-01T20:30:00.000Z     YYYY-MM-DDThh:mm:ss.sssTZD (with offset), for example, 2014-10-01T12:30:00.000-08:00     YYYY-MM-DD, for example, 2014-10-01    Unix time in seconds, for example, 1412195400. This is also referred to as Unix Epoch time and represents the number of seconds since midnight, January 1, 1970 UTC.   For more information about the timestamp format, see Timestamp in the Amazon Web Services Command Line Interface User Guide.
             creation_time_before: Set an end time for the time range during which you want to list SageMaker HyperPod clusters. A filter that returns nodes in a SageMaker HyperPod cluster created before the specified time. The acceptable formats are the same as the timestamp formats for CreationTimeAfter. For more information about the timestamp format, see Timestamp in the Amazon Web Services Command Line Interface User Guide.
-            max_results: Set the maximum number of SageMaker HyperPod clusters to list.
+            max_results: Specifies the maximum number of clusters to evaluate for the operation (not necessarily the number of matching items). After SageMaker processes the number of clusters up to MaxResults, it stops the operation and returns the matching clusters up to that point. If all the matching clusters are desired, SageMaker will go through all the clusters until NextToken is empty.
             name_contains: Set the maximum number of instances to print in the list.
             next_token: Set the next token to retrieve the list of SageMaker HyperPod clusters.
             sort_by: The field by which to sort results. The default value is CREATION_TIME.
@@ -26320,6 +26320,7 @@ class Project(Base):
         project_description: The description of the project.
         service_catalog_provisioning_details: Information used to provision a service catalog product. For information, see What is Amazon Web Services Service Catalog.
         service_catalog_provisioned_product_details: Information about a provisioned service catalog product.
+        template_provider_details:  An array of template providers associated with the project.
         created_by:
         last_modified_time: The timestamp when project was last modified.
         last_modified_by:
@@ -26337,6 +26338,7 @@ class Project(Base):
         shapes.ServiceCatalogProvisionedProductDetails
     ] = Unassigned()
     project_status: Optional[str] = Unassigned()
+    template_provider_details: Optional[List[shapes.TemplateProviderDetail]] = Unassigned()
     created_by: Optional[shapes.UserContext] = Unassigned()
     creation_time: Optional[datetime.datetime] = Unassigned()
     last_modified_time: Optional[datetime.datetime] = Unassigned()
@@ -26368,6 +26370,7 @@ class Project(Base):
             shapes.ServiceCatalogProvisioningDetails
         ] = Unassigned(),
         tags: Optional[List[shapes.Tag]] = Unassigned(),
+        template_providers: Optional[List[shapes.CreateTemplateProvider]] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> Optional["Project"]:
@@ -26379,6 +26382,7 @@ class Project(Base):
             project_description: A description for the project.
             service_catalog_provisioning_details: The product ID and provisioning artifact ID to provision a service catalog. The provisioning artifact ID will default to the latest provisioning artifact ID of the product, if you don't provide the provisioning artifact ID. For more information, see What is Amazon Web Services Service Catalog.
             tags: An array of key-value pairs that you want to use to organize and track your Amazon Web Services resource costs. For more information, see Tagging Amazon Web Services resources in the Amazon Web Services General Reference Guide.
+            template_providers:  An array of template provider configurations for creating infrastructure resources for the project.
             session: Boto3 session.
             region: Region name.
 
@@ -26411,6 +26415,7 @@ class Project(Base):
             "ProjectDescription": project_description,
             "ServiceCatalogProvisioningDetails": service_catalog_provisioning_details,
             "Tags": tags,
+            "TemplateProviders": template_providers,
         }
 
         operation_input_args = Base.populate_chained_attributes(
@@ -26522,6 +26527,7 @@ class Project(Base):
             shapes.ServiceCatalogProvisioningUpdateDetails
         ] = Unassigned(),
         tags: Optional[List[shapes.Tag]] = Unassigned(),
+        template_providers_to_update: Optional[List[shapes.UpdateTemplateProvider]] = Unassigned(),
     ) -> Optional["Project"]:
         """
         Update a Project resource
@@ -26529,6 +26535,7 @@ class Project(Base):
         Parameters:
             service_catalog_provisioning_update_details: The product ID and provisioning artifact ID to provision a service catalog. The provisioning artifact ID will default to the latest provisioning artifact ID of the product, if you don't provide the provisioning artifact ID. For more information, see What is Amazon Web Services Service Catalog.
             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. In addition, the project must have tag update constraints set in order to include this parameter in the request. For more information, see Amazon Web Services Service Catalog Tag Update Constraints.
+            template_providers_to_update:  The template providers to update in the project.
 
         Returns:
             The Project resource.
@@ -26554,6 +26561,7 @@ class Project(Base):
             "ProjectDescription": project_description,
             "ServiceCatalogProvisioningUpdateDetails": service_catalog_provisioning_update_details,
             "Tags": tags,
+            "TemplateProvidersToUpdate": template_providers_to_update,
         }
         logger.debug(f"Input request: {operation_input_args}")
         # serialize the input request

{sagemaker_core-1.0.39 → sagemaker_core-1.0.41}/src/sagemaker_core/main/shapes.py

@@ -999,7 +999,7 @@ class ResourceConfig(Base):
 
     Attributes
     ----------------------
-    instance_type: The ML compute instance type.   SageMaker Training on Amazon Elastic Compute Cloud (EC2) P4de instances is in preview release starting December 9th, 2022.   Amazon EC2 P4de instances (currently in preview) are powered by 8 NVIDIA A100 GPUs with 80GB high-performance HBM2e GPU memory, which accelerate the speed of training ML models that need to be trained on large datasets of high-resolution data. In this preview release, Amazon SageMaker supports ML training jobs on P4de instances (ml.p4de.24xlarge) to reduce model training time. The ml.p4de.24xlarge instances are available in the following Amazon Web Services Regions.    US East (N. Virginia) (us-east-1)   US West (Oregon) (us-west-2)   To request quota limit increase and start using P4de instances, contact the SageMaker Training service team through your account team.
+    instance_type: The ML compute instance type.
     instance_count: The number of ML compute instances to use. For distributed training, provide a value greater than 1.
     volume_size_in_gb: The size of the ML storage volume that you want to provision.  ML storage volumes store model artifacts and incremental states. Training algorithms might also use the ML storage volume for scratch space. If you want to store the training data in the ML storage volume, choose File as the TrainingInputMode in the algorithm specification.  When using an ML instance with NVMe SSD volumes, SageMaker doesn't provision Amazon EBS General Purpose SSD (gp2) storage. Available storage is fixed to the NVMe-type instance's storage capacity. SageMaker configures storage paths for training datasets, checkpoints, model artifacts, and outputs to use the entire capacity of the instance storage. For example, ML instance families with the NVMe-type instance storage include ml.p4d, ml.g4dn, and ml.g5.  When using an ML instance with the EBS-only storage option and without instance storage, you must define the size of EBS volume through VolumeSizeInGB in the ResourceConfig API. For example, ML instance families that use EBS volumes include ml.c5 and ml.p2.  To look up instance types and their instance storage types and volumes, see Amazon EC2 Instance Types. To find the default local paths defined by the SageMaker training platform, see Amazon SageMaker Training Storage Folders for Training Datasets, Checkpoints, Model Artifacts, and Outputs.
     volume_kms_key_id: The Amazon Web Services KMS key that SageMaker uses to encrypt data on the storage volume attached to the ML compute instance(s) that run the training job.  Certain Nitro-based instances include local storage, dependent on the instance type. Local storage volumes are encrypted using a hardware module on the instance. You can't request a VolumeKmsKeyId when using an instance type with local storage. For a list of instance types that support local instance storage, see Instance Store Volumes. For more information about local instance storage encryption, see SSD Instance Store Volumes.  The VolumeKmsKeyId can be in any of the following formats:   // KMS Key ID  "1234abcd-12ab-34cd-56ef-1234567890ab"    // Amazon Resource Name (ARN) of a KMS Key  "arn:aws:kms:us-west-2:111122223333:key/1234abcd-12ab-34cd-56ef-1234567890ab"
@@ -2972,6 +2972,125 @@ class CategoricalParameterRangeSpecification(Base):
     values: List[str]
 
 
+class CfnStackCreateParameter(Base):
+    """
+    CfnStackCreateParameter
+       A key-value pair that represents a parameter for the CloudFormation stack.
+
+    Attributes
+    ----------------------
+    key:  The name of the CloudFormation parameter.
+    value:  The value of the CloudFormation parameter.
+    """
+
+    key: str
+    value: Optional[str] = Unassigned()
+
+
+class CfnCreateTemplateProvider(Base):
+    """
+    CfnCreateTemplateProvider
+       The CloudFormation template provider configuration for creating infrastructure resources.
+
+    Attributes
+    ----------------------
+    template_name:  A unique identifier for the template within the project.
+    template_url:  The Amazon S3 URL of the CloudFormation template.
+    role_arn:  The IAM role that CloudFormation assumes when creating the stack.
+    parameters:  An array of CloudFormation stack parameters.
+    """
+
+    template_name: str
+    template_url: str
+    role_arn: Optional[str] = Unassigned()
+    parameters: Optional[List[CfnStackCreateParameter]] = Unassigned()
+
+
+class CfnStackDetail(Base):
+    """
+    CfnStackDetail
+       Details about the CloudFormation stack.
+
+    Attributes
+    ----------------------
+    name:  The name of the CloudFormation stack.
+    id:  The unique identifier of the CloudFormation stack.
+    status_message:  A human-readable message about the stack's current status.
+    """
+
+    status_message: str
+    name: Optional[str] = Unassigned()
+    id: Optional[str] = Unassigned()
+
+
+class CfnStackParameter(Base):
+    """
+    CfnStackParameter
+       A key-value pair representing a parameter used in the CloudFormation stack.
+
+    Attributes
+    ----------------------
+    key:  The name of the CloudFormation parameter.
+    value:  The value of the CloudFormation parameter.
+    """
+
+    key: str
+    value: Optional[str] = Unassigned()
+
+
+class CfnStackUpdateParameter(Base):
+    """
+    CfnStackUpdateParameter
+       A key-value pair representing a parameter used in the CloudFormation stack.
+
+    Attributes
+    ----------------------
+    key:  The name of the CloudFormation parameter.
+    value:  The value of the CloudFormation parameter.
+    """
+
+    key: str
+    value: Optional[str] = Unassigned()
+
+
+class CfnTemplateProviderDetail(Base):
+    """
+    CfnTemplateProviderDetail
+       Details about a CloudFormation template provider configuration and associated provisioning information.
+
+    Attributes
+    ----------------------
+    template_name:  The unique identifier of the template within the project.
+    template_url:  The Amazon S3 URL of the CloudFormation template.
+    role_arn:  The IAM role used by CloudFormation to create the stack.
+    parameters:  An array of CloudFormation stack parameters.
+    stack_detail:  Information about the CloudFormation stack created by the template provider.
+    """
+
+    template_name: str
+    template_url: str
+    role_arn: Optional[str] = Unassigned()
+    parameters: Optional[List[CfnStackParameter]] = Unassigned()
+    stack_detail: Optional[CfnStackDetail] = Unassigned()
+
+
+class CfnUpdateTemplateProvider(Base):
+    """
+    CfnUpdateTemplateProvider
+       Contains configuration details for updating an existing CloudFormation template provider in the project.
+
+    Attributes
+    ----------------------
+    template_name:  The unique identifier of the template to update within the project.
+    template_url:  The Amazon S3 URL of the CloudFormation template.
+    parameters:  An array of CloudFormation stack parameters.
+    """
+
+    template_name: str
+    template_url: str
+    parameters: Optional[List[CfnStackUpdateParameter]] = Unassigned()
+
+
 class ChannelSpecification(Base):
     """
     ChannelSpecification
@@ -7354,6 +7473,19 @@ class ServiceCatalogProvisioningDetails(Base):
     provisioning_parameters: Optional[List[ProvisioningParameter]] = Unassigned()
 
 
+class CreateTemplateProvider(Base):
+    """
+    CreateTemplateProvider
+       Contains configuration details for a template provider. Only one type of template provider can be specified.
+
+    Attributes
+    ----------------------
+    cfn_template_provider:  The CloudFormation template provider configuration for creating infrastructure resources.
+    """
+
+    cfn_template_provider: Optional[CfnCreateTemplateProvider] = Unassigned()
+
+
 class SpaceIdleSettings(Base):
     """
     SpaceIdleSettings
@@ -9032,6 +9164,19 @@ class ServiceCatalogProvisionedProductDetails(Base):
     provisioned_product_status_message: Optional[str] = Unassigned()
 
 
+class TemplateProviderDetail(Base):
+    """
+    TemplateProviderDetail
+       Details about a template provider configuration and associated provisioning information.
+
+    Attributes
+    ----------------------
+    cfn_template_provider_detail:  Details about a CloudFormation template provider configuration and associated provisioning information.
+    """
+
+    cfn_template_provider_detail: Optional[CfnTemplateProviderDetail] = Unassigned()
+
+
 class SubscribedWorkteam(Base):
     """
     SubscribedWorkteam
@@ -12245,6 +12390,7 @@ class Project(Base):
     project_status: The status of the project.
     created_by: Who created the project.
     creation_time: A timestamp specifying when the project was created.
+    template_provider_details:  An array of template providers associated with the project.
     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.
     last_modified_time: A timestamp container for when the project was last modified.
     last_modified_by
@@ -12261,6 +12407,7 @@ class Project(Base):
     project_status: Optional[str] = Unassigned()
     created_by: Optional[UserContext] = Unassigned()
     creation_time: Optional[datetime.datetime] = Unassigned()
+    template_provider_details: Optional[List[TemplateProviderDetail]] = Unassigned()
     tags: Optional[List[Tag]] = Unassigned()
     last_modified_time: Optional[datetime.datetime] = Unassigned()
     last_modified_by: Optional[UserContext] = Unassigned()
@@ -12803,3 +12950,16 @@ class VariantProperty(Base):
     """
 
     variant_property_type: str
+
+
+class UpdateTemplateProvider(Base):
+    """
+    UpdateTemplateProvider
+       Contains configuration details for updating an existing template provider in the project.
+
+    Attributes
+    ----------------------
+    cfn_template_provider:  The CloudFormation template provider configuration to update.
+    """
+
+    cfn_template_provider: Optional[CfnUpdateTemplateProvider] = Unassigned()

{sagemaker_core-1.0.39 → sagemaker_core-1.0.41/src/sagemaker_core.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.39
+Version: 1.0.41
 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.39/VERSION

@@ -1 +0,0 @@
-1.0.39