sagemaker-core 1.0.25__tar.gz → 1.0.27__tar.gz

{sagemaker_core-1.0.25/src/sagemaker_core.egg-info → sagemaker_core-1.0.27}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.25
+Version: 1.0.27
 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
@@ -28,6 +28,7 @@ Requires-Dist: black<25.0.0,>=24.3.0; extra == "codegen"
 Requires-Dist: pandas<3.0.0,>=2.0.0; extra == "codegen"
 Requires-Dist: pytest<9.0.0,>=8.0.0; extra == "codegen"
 Requires-Dist: pylint<4.0.0,>=3.0.0; extra == "codegen"
+Dynamic: license-file
 
 .. image:: https://github.com/aws/sagemaker-python-sdk/raw/master/branding/icon/sagemaker-banner.png
     :height: 100px

sagemaker_core-1.0.27/VERSION

@@ -0,0 +1 @@
+1.0.27

{sagemaker_core-1.0.25 → sagemaker_core-1.0.27}/src/sagemaker_core/main/code_injection/shape_dag.py

@@ -2070,6 +2070,7 @@ SHAPE_DAG = {
     "CreateDomainResponse": {
         "members": [
             {"name": "DomainArn", "shape": "DomainArn", "type": "string"},
+            {"name": "DomainId", "shape": "DomainId", "type": "string"},
             {"name": "Url", "shape": "String1024", "type": "string"},
         ],
         "type": "structure",
@@ -2811,6 +2812,7 @@ SHAPE_DAG = {
             {"name": "Name", "shape": "PartnerAppName", "type": "string"},
             {"name": "Type", "shape": "PartnerAppType", "type": "string"},
             {"name": "ExecutionRoleArn", "shape": "RoleArn", "type": "string"},
+            {"name": "KmsKeyId", "shape": "KmsKeyId", "type": "string"},
             {
                 "name": "MaintenanceConfig",
                 "shape": "PartnerAppMaintenanceConfig",
@@ -4712,6 +4714,7 @@ SHAPE_DAG = {
             {"name": "HubContentStatus", "shape": "HubContentStatus", "type": "string"},
             {"name": "FailureReason", "shape": "FailureReason", "type": "string"},
             {"name": "CreationTime", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "LastModifiedTime", "shape": "Timestamp", "type": "timestamp"},
         ],
         "type": "structure",
     },
@@ -5494,7 +5497,9 @@ SHAPE_DAG = {
             {"name": "Type", "shape": "PartnerAppType", "type": "string"},
             {"name": "Status", "shape": "PartnerAppStatus", "type": "string"},
             {"name": "CreationTime", "shape": "Timestamp", "type": "timestamp"},
+            {"name": "LastModifiedTime", "shape": "Timestamp", "type": "timestamp"},
             {"name": "ExecutionRoleArn", "shape": "RoleArn", "type": "string"},
+            {"name": "KmsKeyId", "shape": "KmsKeyId", "type": "string"},
             {"name": "BaseUrl", "shape": "String2048", "type": "string"},
             {
                 "name": "MaintenanceConfig",
@@ -7024,6 +7029,10 @@ SHAPE_DAG = {
         "value_shape": "ConfigValue",
         "value_type": "string",
     },
+    "HubAccessConfig": {
+        "members": [{"name": "HubContentArn", "shape": "HubContentArn", "type": "string"}],
+        "type": "structure",
+    },
     "HubContentDependency": {
         "members": [
             {"name": "DependencyOriginPath", "shape": "DependencyOriginPath", "type": "string"},
@@ -7659,6 +7668,7 @@ SHAPE_DAG = {
             {"name": "HubContentDescription", "shape": "HubContentDescription", "type": "string"},
             {"name": "HubContentMarkdown", "shape": "HubContentMarkdown", "type": "string"},
             {"name": "HubContentDocument", "shape": "HubContentDocument", "type": "string"},
+            {"name": "SupportStatus", "shape": "HubContentSupportStatus", "type": "string"},
             {
                 "name": "HubContentSearchKeywords",
                 "shape": "HubContentSearchKeywordList",
@@ -13182,6 +13192,8 @@ SHAPE_DAG = {
             {"name": "S3DataDistributionType", "shape": "S3DataDistribution", "type": "string"},
             {"name": "AttributeNames", "shape": "AttributeNames", "type": "list"},
             {"name": "InstanceGroupNames", "shape": "InstanceGroupNames", "type": "list"},
+            {"name": "ModelAccessConfig", "shape": "ModelAccessConfig", "type": "structure"},
+            {"name": "HubAccessConfig", "shape": "HubAccessConfig", "type": "structure"},
         ],
         "type": "structure",
     },
@@ -14991,6 +15003,47 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "UpdateHubContentReferenceRequest": {
+        "members": [
+            {"name": "HubName", "shape": "HubNameOrArn", "type": "string"},
+            {"name": "HubContentName", "shape": "HubContentName", "type": "string"},
+            {"name": "HubContentType", "shape": "HubContentType", "type": "string"},
+            {"name": "MinVersion", "shape": "HubContentVersion", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "UpdateHubContentReferenceResponse": {
+        "members": [
+            {"name": "HubArn", "shape": "HubArn", "type": "string"},
+            {"name": "HubContentArn", "shape": "HubContentArn", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "UpdateHubContentRequest": {
+        "members": [
+            {"name": "HubName", "shape": "HubNameOrArn", "type": "string"},
+            {"name": "HubContentName", "shape": "HubContentName", "type": "string"},
+            {"name": "HubContentType", "shape": "HubContentType", "type": "string"},
+            {"name": "HubContentVersion", "shape": "HubContentVersion", "type": "string"},
+            {"name": "HubContentDisplayName", "shape": "HubContentDisplayName", "type": "string"},
+            {"name": "HubContentDescription", "shape": "HubContentDescription", "type": "string"},
+            {"name": "HubContentMarkdown", "shape": "HubContentMarkdown", "type": "string"},
+            {
+                "name": "HubContentSearchKeywords",
+                "shape": "HubContentSearchKeywordList",
+                "type": "list",
+            },
+            {"name": "SupportStatus", "shape": "HubContentSupportStatus", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "UpdateHubContentResponse": {
+        "members": [
+            {"name": "HubArn", "shape": "HubArn", "type": "string"},
+            {"name": "HubContentArn", "shape": "HubContentArn", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "UpdateHubRequest": {
         "members": [
             {"name": "HubName", "shape": "HubNameOrArn", "type": "string"},

{sagemaker_core-1.0.25 → sagemaker_core-1.0.27}/src/sagemaker_core/main/config_schema.py

@@ -756,7 +756,10 @@ SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA = {
                                 },
                                 "PartnerApp": {
                                     "type": "object",
-                                    "properties": {"execution_role_arn": {"type": "string"}},
+                                    "properties": {
+                                        "execution_role_arn": {"type": "string"},
+                                        "kms_key_id": {"type": "string"},
+                                    },
                                 },
                                 "Pipeline": {
                                     "type": "object",

{sagemaker_core-1.0.25 → sagemaker_core-1.0.27}/src/sagemaker_core/main/resources.py

@@ -12221,6 +12221,7 @@ class HubContent(Base):
         hub_content_search_keywords: The searchable keywords for the hub content.
         hub_content_dependencies: The location of any dependencies that the hub content has, such as scripts, model artifacts, datasets, or notebooks.
         failure_reason: The failure reason if importing hub content failed.
+        last_modified_time: The last modified time of the hub content.
 
     """
 
@@ -12242,6 +12243,7 @@ class HubContent(Base):
     hub_content_status: Optional[str] = Unassigned()
     failure_reason: Optional[str] = Unassigned()
     creation_time: Optional[datetime.datetime] = Unassigned()
+    last_modified_time: Optional[datetime.datetime] = Unassigned()
     hub_name: Optional[str] = Unassigned()
 
     def get_name(self) -> str:
@@ -12360,6 +12362,63 @@ class HubContent(Base):
         transform(response, "DescribeHubContentResponse", self)
         return self
 
+    @Base.add_validate_call
+    def update(
+        self,
+        hub_content_type: str,
+        hub_content_version: str,
+        hub_content_display_name: Optional[str] = Unassigned(),
+        hub_content_description: Optional[str] = Unassigned(),
+        hub_content_markdown: Optional[str] = Unassigned(),
+        hub_content_search_keywords: Optional[List[str]] = Unassigned(),
+        support_status: Optional[str] = Unassigned(),
+    ) -> Optional["HubContent"]:
+        """
+        Update a HubContent resource
+
+        Returns:
+            The HubContent resource.
+
+        Raises:
+            botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
+                The error message and error code can be parsed from the exception as follows:
+                ```
+                try:
+                    # AWS service call here
+                except botocore.exceptions.ClientError as e:
+                    error_message = e.response['Error']['Message']
+                    error_code = e.response['Error']['Code']
+                ```
+            ResourceInUse: Resource being accessed is in use.
+            ResourceNotFound: Resource being access is not found.
+        """
+
+        logger.info("Updating hub_content resource.")
+        client = Base.get_sagemaker_client()
+
+        operation_input_args = {
+            "HubName": self.hub_name,
+            "HubContentName": self.hub_content_name,
+            "HubContentType": hub_content_type,
+            "HubContentVersion": hub_content_version,
+            "HubContentDisplayName": hub_content_display_name,
+            "HubContentDescription": hub_content_description,
+            "HubContentMarkdown": hub_content_markdown,
+            "HubContentSearchKeywords": hub_content_search_keywords,
+            "SupportStatus": support_status,
+        }
+        logger.debug(f"Input request: {operation_input_args}")
+        # serialize the input request
+        operation_input_args = serialize(operation_input_args)
+        logger.debug(f"Serialized input request: {operation_input_args}")
+
+        # create the resource
+        response = client.update_hub_content(**operation_input_args)
+        logger.debug(f"Response: {response}")
+        self.refresh()
+
+        return self
+
     @Base.add_validate_call
     def delete(
         self,
@@ -12400,7 +12459,7 @@ class HubContent(Base):
     @Base.add_validate_call
     def wait_for_status(
         self,
-        target_status: Literal["Supported", "Deprecated"],
+        target_status: Literal["Supported", "Deprecated", "Restricted"],
         poll: int = 5,
         timeout: Optional[int] = None,
     ) -> None:
@@ -12461,6 +12520,7 @@ class HubContent(Base):
         hub_content_display_name: Optional[str] = Unassigned(),
         hub_content_description: Optional[str] = Unassigned(),
         hub_content_markdown: Optional[str] = Unassigned(),
+        support_status: Optional[str] = Unassigned(),
         hub_content_search_keywords: Optional[List[str]] = Unassigned(),
         tags: Optional[List[Tag]] = Unassigned(),
         session: Optional[Session] = None,
@@ -12479,6 +12539,7 @@ class HubContent(Base):
             hub_content_display_name: The display name of the hub content to import.
             hub_content_description: A description of the hub content to import.
             hub_content_markdown: A string that provides a description of the hub content. This string can include links, tables, and standard markdown formating.
+            support_status: The status of the hub content resource.
             hub_content_search_keywords: The searchable keywords of the hub content.
             tags: Any tags associated with the hub content.
             session: Boto3 session.
@@ -12517,6 +12578,7 @@ class HubContent(Base):
             "HubContentDescription": hub_content_description,
             "HubContentMarkdown": hub_content_markdown,
             "HubContentDocument": hub_content_document,
+            "SupportStatus": support_status,
             "HubContentSearchKeywords": hub_content_search_keywords,
             "Tags": tags,
         }
@@ -12714,6 +12776,56 @@ class HubContentReference(Base):
         transformed_response = transform(response, "CreateHubContentReferenceResponse")
         return cls(**operation_input_args, **transformed_response)
 
+    @Base.add_validate_call
+    def update(
+        self,
+        hub_content_type: str,
+        min_version: Optional[str] = Unassigned(),
+    ) -> Optional["HubContentReference"]:
+        """
+        Update a HubContentReference resource
+
+        Parameters:
+            hub_content_type: The content type of the resource that you want to update. Only specify a ModelReference resource for this API. To update a Model or Notebook resource, use the UpdateHubContent API instead.
+
+        Returns:
+            The HubContentReference resource.
+
+        Raises:
+            botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
+                The error message and error code can be parsed from the exception as follows:
+                ```
+                try:
+                    # AWS service call here
+                except botocore.exceptions.ClientError as e:
+                    error_message = e.response['Error']['Message']
+                    error_code = e.response['Error']['Code']
+                ```
+            ResourceInUse: Resource being accessed is in use.
+            ResourceNotFound: Resource being access is not found.
+        """
+
+        logger.info("Updating hub_content_reference resource.")
+        client = Base.get_sagemaker_client()
+
+        operation_input_args = {
+            "HubName": self.hub_name,
+            "HubContentName": self.hub_content_name,
+            "HubContentType": hub_content_type,
+            "MinVersion": min_version,
+        }
+        logger.debug(f"Input request: {operation_input_args}")
+        # serialize the input request
+        operation_input_args = serialize(operation_input_args)
+        logger.debug(f"Serialized input request: {operation_input_args}")
+
+        # create the resource
+        response = client.update_hub_content_reference(**operation_input_args)
+        logger.debug(f"Response: {response}")
+        self.refresh()
+
+        return self
+
     @Base.add_validate_call
     def delete(
         self,
@@ -23528,7 +23640,9 @@ class PartnerApp(Base):
         type: The type of SageMaker Partner AI App. Must be one of the following: lakera-guard, comet, deepchecks-llm-evaluation, or fiddler.
         status: The status of the SageMaker Partner AI App.
         creation_time: The time that the SageMaker Partner AI App was created.
+        last_modified_time: The time that the SageMaker Partner AI App was last modified.
         execution_role_arn: The ARN of the IAM role associated with the SageMaker Partner AI App.
+        kms_key_id: The Amazon Web Services KMS customer managed key used to encrypt the data at rest associated with SageMaker Partner AI Apps.
         base_url: The URL of the SageMaker Partner AI App that the Application SDK uses to support in-app calls for the user.
         maintenance_config: Maintenance configuration settings for the SageMaker Partner AI App.
         tier: The instance type and size of the cluster attached to the SageMaker Partner AI App.
@@ -23545,7 +23659,9 @@ class PartnerApp(Base):
     type: Optional[str] = Unassigned()
     status: Optional[str] = Unassigned()
     creation_time: Optional[datetime.datetime] = Unassigned()
+    last_modified_time: Optional[datetime.datetime] = Unassigned()
     execution_role_arn: Optional[str] = Unassigned()
+    kms_key_id: Optional[str] = Unassigned()
     base_url: Optional[str] = Unassigned()
     maintenance_config: Optional[PartnerAppMaintenanceConfig] = Unassigned()
     tier: Optional[str] = Unassigned()
@@ -23594,6 +23710,7 @@ class PartnerApp(Base):
         execution_role_arn: str,
         tier: str,
         auth_type: str,
+        kms_key_id: Optional[str] = Unassigned(),
         maintenance_config: Optional[PartnerAppMaintenanceConfig] = Unassigned(),
         application_config: Optional[PartnerAppConfig] = Unassigned(),
         enable_iam_session_based_identity: Optional[bool] = Unassigned(),
@@ -23611,6 +23728,7 @@ class PartnerApp(Base):
             execution_role_arn: The ARN of the IAM role that the partner application uses.
             tier: Indicates the instance type and size of the cluster attached to the SageMaker Partner AI App.
             auth_type: The authorization type that users use to access the SageMaker Partner AI App.
+            kms_key_id: SageMaker Partner AI Apps uses Amazon Web Services KMS to encrypt data at rest using an Amazon Web Services managed key by default. For more control, specify a customer managed key.
             maintenance_config: Maintenance configuration settings for the SageMaker Partner AI App.
             application_config: Configuration settings for the SageMaker Partner AI App.
             enable_iam_session_based_identity: When set to TRUE, the SageMaker Partner AI App sets the Amazon Web Services IAM session name or the authenticated IAM user as the identity of the SageMaker Partner AI App user.
@@ -23648,6 +23766,7 @@ class PartnerApp(Base):
             "Name": name,
             "Type": type,
             "ExecutionRoleArn": execution_role_arn,
+            "KmsKeyId": kms_key_id,
             "MaintenanceConfig": maintenance_config,
             "Tier": tier,
             "ApplicationConfig": application_config,

{sagemaker_core-1.0.25 → sagemaker_core-1.0.27}/src/sagemaker_core/main/shapes.py

@@ -835,6 +835,19 @@ class AlgorithmSummary(Base):
     algorithm_description: Optional[str] = Unassigned()
 
 
+class HubAccessConfig(Base):
+    """
+    HubAccessConfig
+      The configuration for a private hub model reference that points to a public SageMaker JumpStart model. For more information about private hubs, see Private curated hubs for foundation model access control in JumpStart.
+
+    Attributes
+    ----------------------
+    hub_content_arn: The ARN of your private model hub content. This should be a ModelReference resource type that points to a SageMaker JumpStart public hub model.
+    """
+
+    hub_content_arn: str
+
+
 class S3DataSource(Base):
     """
     S3DataSource
@@ -847,6 +860,8 @@ class S3DataSource(Base):
     s3_data_distribution_type: If you want SageMaker to replicate the entire dataset on each ML compute instance that is launched for model training, specify FullyReplicated.  If you want SageMaker to replicate a subset of data on each ML compute instance that is launched for model training, specify ShardedByS3Key. If there are n ML compute instances launched for a training job, each instance gets approximately 1/n of the number of S3 objects. In this case, model training on each machine uses only the subset of training data.  Don't choose more ML compute instances for training than available S3 objects. If you do, some nodes won't get any data and you will pay for nodes that aren't getting any training data. This applies in both File and Pipe modes. Keep this in mind when developing algorithms.  In distributed training, where you use multiple ML compute EC2 instances, you might choose ShardedByS3Key. If the algorithm requires copying training data to the ML storage volume (when TrainingInputMode is set to File), this copies 1/n of the number of objects.
     attribute_names: A list of one or more attribute names to use that are found in a specified augmented manifest file.
     instance_group_names: A list of names of instance groups that get data from the S3 data source.
+    model_access_config
+    hub_access_config: The configuration for a private hub model reference that points to a SageMaker JumpStart public hub model.
     """
 
     s3_data_type: str
@@ -854,6 +869,8 @@ class S3DataSource(Base):
     s3_data_distribution_type: Optional[str] = Unassigned()
     attribute_names: Optional[List[str]] = Unassigned()
     instance_group_names: Optional[List[str]] = Unassigned()
+    model_access_config: Optional[ModelAccessConfig] = Unassigned()
+    hub_access_config: Optional[HubAccessConfig] = Unassigned()
 
 
 class FileSystemDataSource(Base):
@@ -4368,7 +4385,7 @@ class KernelGatewayAppSettings(Base):
     Attributes
     ----------------------
     default_resource_spec: The default instance type and the Amazon Resource Name (ARN) of the default SageMaker AI image used by the KernelGateway app.  The Amazon SageMaker AI Studio UI does not use the default instance type value set here. The default instance type set here is used when Apps are created using the CLI or CloudFormation and the instance type parameter value is not passed.
-    custom_images: A list of custom SageMaker AI images that are configured to run as a KernelGateway app.
+    custom_images: A list of custom SageMaker AI images that are configured to run as a KernelGateway app. The maximum number of custom images are as follows.   On a domain level: 200   On a space level: 5   On a user profile level: 5
     lifecycle_config_arns:  The Amazon Resource Name (ARN) of the Lifecycle Configurations attached to the the user profile or domain.  To remove a Lifecycle Config, you must set LifecycleConfigArns to an empty list.
     """
 
@@ -7101,7 +7118,7 @@ class ProcessingOutputConfig(Base):
     Attributes
     ----------------------
     outputs: An array of outputs configuring the data to upload from the processing container.
-    kms_key_id: The Amazon Web Services Key Management Service (Amazon Web Services KMS) key that Amazon SageMaker uses to encrypt the processing job output. KmsKeyId can be an ID of a KMS key, ARN of a KMS key, alias of a KMS key, or alias of a KMS key. The KmsKeyId is applied to all outputs.
+    kms_key_id: The Amazon Web Services Key Management Service (Amazon Web Services KMS) key that Amazon SageMaker uses to encrypt the processing job output. KmsKeyId can be an ID of a KMS key, ARN of a KMS key, or alias of a KMS key. The KmsKeyId is applied to all outputs.
     """
 
     outputs: List[ProcessingOutput]

{sagemaker_core-1.0.25 → sagemaker_core-1.0.27/src/sagemaker_core.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.25
+Version: 1.0.27
 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
@@ -28,6 +28,7 @@ Requires-Dist: black<25.0.0,>=24.3.0; extra == "codegen"
 Requires-Dist: pandas<3.0.0,>=2.0.0; extra == "codegen"
 Requires-Dist: pytest<9.0.0,>=8.0.0; extra == "codegen"
 Requires-Dist: pylint<4.0.0,>=3.0.0; extra == "codegen"
+Dynamic: license-file
 
 .. image:: https://github.com/aws/sagemaker-python-sdk/raw/master/branding/icon/sagemaker-banner.png
     :height: 100px

sagemaker_core-1.0.25/VERSION

@@ -1 +0,0 @@
-1.0.25