sagemaker-core 1.0.26__py3-none-any.whl → 1.0.28__py3-none-any.whl

sagemaker_core/main/code_injection/shape_dag.py

@@ -1817,6 +1817,7 @@ SHAPE_DAG = {
             {"name": "AppName", "shape": "AppName", "type": "string"},
             {"name": "Tags", "shape": "TagList", "type": "list"},
             {"name": "ResourceSpec", "shape": "ResourceSpec", "type": "structure"},
+            {"name": "RecoveryMode", "shape": "Boolean", "type": "boolean"},
         ],
         "type": "structure",
     },
@@ -2743,6 +2744,7 @@ SHAPE_DAG = {
             },
             {"name": "OnCreate", "shape": "NotebookInstanceLifecycleConfigList", "type": "list"},
             {"name": "OnStart", "shape": "NotebookInstanceLifecycleConfigList", "type": "list"},
+            {"name": "Tags", "shape": "TagList", "type": "list"},
         ],
         "type": "structure",
     },
@@ -2812,6 +2814,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",
@@ -4052,6 +4055,7 @@ SHAPE_DAG = {
             {"name": "UserProfileName", "shape": "UserProfileName", "type": "string"},
             {"name": "SpaceName", "shape": "SpaceName", "type": "string"},
             {"name": "Status", "shape": "AppStatus", "type": "string"},
+            {"name": "RecoveryMode", "shape": "Boolean", "type": "boolean"},
             {"name": "LastHealthCheckTimestamp", "shape": "Timestamp", "type": "timestamp"},
             {"name": "LastUserActivityTimestamp", "shape": "Timestamp", "type": "timestamp"},
             {"name": "CreationTime", "shape": "Timestamp", "type": "timestamp"},
@@ -5496,7 +5500,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",
@@ -13334,6 +13340,7 @@ SHAPE_DAG = {
         "members": [
             {"name": "Results", "shape": "SearchResultsList", "type": "list"},
             {"name": "NextToken", "shape": "NextToken", "type": "string"},
+            {"name": "TotalHits", "shape": "TotalHits", "type": "structure"},
         ],
         "type": "structure",
     },
@@ -14085,6 +14092,13 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "TotalHits": {
+        "members": [
+            {"name": "Value", "shape": "Long", "type": "long"},
+            {"name": "Relation", "shape": "Relation", "type": "string"},
+        ],
+        "type": "structure",
+    },
     "TrackingServerSummary": {
         "members": [
             {"name": "TrackingServerArn", "shape": "TrackingServerArn", "type": "string"},
@@ -14485,6 +14499,7 @@ SHAPE_DAG = {
             {"name": "InstanceType", "shape": "TransformInstanceType", "type": "string"},
             {"name": "InstanceCount", "shape": "TransformInstanceCount", "type": "integer"},
             {"name": "VolumeKmsKeyId", "shape": "KmsKeyId", "type": "string"},
+            {"name": "TransformAmiVersion", "shape": "TransformAmiVersion", "type": "string"},
         ],
         "type": "structure",
     },

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/main/resources.py

@@ -974,6 +974,7 @@ class App(Base):
         user_profile_name: The user profile name.
         space_name: The name of the space. If this value is not set, then UserProfileName must be set.
         status: The status.
+        recovery_mode:  Indicates whether the application is launched in recovery mode.
         last_health_check_timestamp: The timestamp of the last health check.
         last_user_activity_timestamp: The timestamp of the last user's activity. LastUserActivityTimestamp is also updated when SageMaker AI performs health checks without user activity. As a result, this value is set to the same value as LastHealthCheckTimestamp.
         creation_time: The creation time of the application.  After an application has been shut down for 24 hours, SageMaker AI deletes all metadata for the application. To be considered an update and retain application metadata, applications must be restarted within 24 hours after the previous application has been shut down. After this time window, creation of an application is considered a new application rather than an update of the previous application.
@@ -990,6 +991,7 @@ class App(Base):
     user_profile_name: Optional[str] = Unassigned()
     space_name: Optional[str] = Unassigned()
     status: Optional[str] = Unassigned()
+    recovery_mode: Optional[bool] = Unassigned()
     last_health_check_timestamp: Optional[datetime.datetime] = Unassigned()
     last_user_activity_timestamp: Optional[datetime.datetime] = Unassigned()
     creation_time: Optional[datetime.datetime] = Unassigned()
@@ -1024,6 +1026,7 @@ class App(Base):
         space_name: Optional[Union[str, object]] = Unassigned(),
         tags: Optional[List[Tag]] = Unassigned(),
         resource_spec: Optional[ResourceSpec] = Unassigned(),
+        recovery_mode: Optional[bool] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> Optional["App"]:
@@ -1038,6 +1041,7 @@ class App(Base):
             space_name: The name of the space. If this value is not set, then UserProfileName must be set.
             tags: Each tag consists of a key and an optional value. Tag keys must be unique per resource.
             resource_spec: The instance type and the Amazon Resource Name (ARN) of the SageMaker AI image created on the instance.  The value of InstanceType passed as part of the ResourceSpec in the CreateApp call overrides the value passed as part of the ResourceSpec configured for the user profile or the domain. If InstanceType is not specified in any of those three ResourceSpec values for a KernelGateway app, the CreateApp call fails with a request validation error.
+            recovery_mode:  Indicates whether the application is launched in recovery mode.
             session: Boto3 session.
             region: Region name.
 
@@ -1074,6 +1078,7 @@ class App(Base):
             "AppName": app_name,
             "Tags": tags,
             "ResourceSpec": resource_spec,
+            "RecoveryMode": recovery_mode,
         }
 
         operation_input_args = Base.populate_chained_attributes(
@@ -22868,6 +22873,7 @@ class NotebookInstanceLifecycleConfig(Base):
         notebook_instance_lifecycle_config_name: str,
         on_create: Optional[List[NotebookInstanceLifecycleHook]] = Unassigned(),
         on_start: Optional[List[NotebookInstanceLifecycleHook]] = Unassigned(),
+        tags: Optional[List[Tag]] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
     ) -> Optional["NotebookInstanceLifecycleConfig"]:
@@ -22878,6 +22884,7 @@ class NotebookInstanceLifecycleConfig(Base):
             notebook_instance_lifecycle_config_name: The name of the lifecycle configuration.
             on_create: A shell script that runs only once, when you create a notebook instance. The shell script must be a base64-encoded string.
             on_start: A shell script that runs every time you start a notebook instance, including when you create the notebook instance. The shell script must be a base64-encoded string.
+            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.
             session: Boto3 session.
             region: Region name.
 
@@ -22909,6 +22916,7 @@ class NotebookInstanceLifecycleConfig(Base):
             "NotebookInstanceLifecycleConfigName": notebook_instance_lifecycle_config_name,
             "OnCreate": on_create,
             "OnStart": on_start,
+            "Tags": tags,
         }
 
         operation_input_args = Base.populate_chained_attributes(
@@ -23640,7 +23648,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.
@@ -23657,7 +23667,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()
@@ -23686,7 +23698,10 @@ class PartnerApp(Base):
     def populate_inputs_decorator(create_func):
         @functools.wraps(create_func)
         def wrapper(*args, **kwargs):
-            config_schema_for_resource = {"execution_role_arn": {"type": "string"}}
+            config_schema_for_resource = {
+                "execution_role_arn": {"type": "string"},
+                "kms_key_id": {"type": "string"},
+            }
             return create_func(
                 *args,
                 **Base.get_updated_kwargs_with_configured_attributes(
@@ -23706,6 +23721,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(),
@@ -23723,6 +23739,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.
@@ -23760,6 +23777,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/main/shapes.py

@@ -1120,11 +1120,13 @@ class TransformResources(Base):
     instance_type: The ML compute instance type for the transform job. If you are using built-in algorithms to transform moderately sized datasets, we recommend using ml.m4.xlarge or ml.m5.largeinstance types.
     instance_count: The number of ML compute instances to use in the transform job. The default value is 1, and the maximum is 100. For distributed transform jobs, specify a value greater than 1.
     volume_kms_key_id: The Amazon Web Services Key Management Service (Amazon Web Services KMS) key that Amazon SageMaker uses to encrypt model data on the storage volume attached to the ML compute instance(s) that run the batch transform 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 any of the following formats:   Key ID: 1234abcd-12ab-34cd-56ef-1234567890ab    Key ARN: arn:aws:kms:us-west-2:111122223333:key/1234abcd-12ab-34cd-56ef-1234567890ab    Alias name: alias/ExampleAlias    Alias name ARN: arn:aws:kms:us-west-2:111122223333:alias/ExampleAlias
+    transform_ami_version: Specifies an option from a collection of preconfigured Amazon Machine Image (AMI) images. Each image is configured by Amazon Web Services with a set of software and driver versions.  al2-ami-sagemaker-batch-gpu-470    Accelerator: GPU   NVIDIA driver version: 470    al2-ami-sagemaker-batch-gpu-535    Accelerator: GPU   NVIDIA driver version: 535
     """
 
     instance_type: str
     instance_count: int
     volume_kms_key_id: Optional[str] = Unassigned()
+    transform_ami_version: Optional[str] = Unassigned()
 
 
 class TransformJobDefinition(Base):
@@ -12520,6 +12522,21 @@ class VisibilityConditions(Base):
     value: Optional[str] = Unassigned()
 
 
+class TotalHits(Base):
+    """
+    TotalHits
+      Represents the total number of matching results and indicates how accurate that count is. The Value field provides the count, which may be exact or estimated. The Relation field indicates whether it's an exact figure or a lower bound. This helps understand the full scope of search results, especially when dealing with large result sets.
+
+    Attributes
+    ----------------------
+    value: The total number of matching results. This value may be exact or an estimate, depending on the Relation field.
+    relation: Indicates the relationship between the returned Value and the actual total number of matching results. Possible values are:    EqualTo: The Value is the exact count of matching results.    GreaterThanOrEqualTo: The Value is a lower bound of the actual count of matching results.
+    """
+
+    value: Optional[int] = Unassigned()
+    relation: Optional[str] = Unassigned()
+
+
 class TrainingPlanOffering(Base):
     """
     TrainingPlanOffering

{sagemaker_core-1.0.26.dist-info → sagemaker_core-1.0.28.dist-info}/METADATA

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

@@ -3,19 +3,19 @@ sagemaker_core/_version.py,sha256=t-XEKT8_pvKkl5z_UrfcIU256c_NxaI16qGWMuNVpbM,32
 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=kfQvf_japrsitTwXTdUaYhOPqKTBBwIE_-ebNeWD-yY,57229
+sagemaker_core/main/config_schema.py,sha256=lBwIm5CT_dSXeW5i6cgRzbiLDjs0qtH1FrM78F-ma2U,57382
 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=OA9eIt59ehBWtStaPK82ksNKPTsUW7wdwzPEENFHujw,1412109
-sagemaker_core/main/shapes.py,sha256=I-bWsTA52wKhoA3faX2DiKYgX87YC3ITFvi53raAtTc,730394
+sagemaker_core/main/resources.py,sha256=ezCu5Hqw9OYu8-rLrC6jJPZYEaUAghJ5rxcsb1XkXRw,1413476
+sagemaker_core/main/shapes.py,sha256=tpq9dp4X1aKDOGNQwt3pslrMMAetQfxJv-oLY-ovYvc,731767
 sagemaker_core/main/user_agent.py,sha256=BPYDAfDd70ObP-VAjl7aDHALHyGknkpRP21ktVr_LDw,2744
 sagemaker_core/main/utils.py,sha256=LCFDM6oxf6_e1i-_Dgtkm3ehl7YfoEpJ2kTTFTL6iOU,18471
 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=M5r0DXoFWW_d-DVo-mtK3i9SGhY1KeJviOroLxyIEuU,699091
+sagemaker_core/main/code_injection/shape_dag.py,sha256=YE-5PMUXhjByNNt0ymf6jvYogRBiPpmc2as2-lAb3Xs,699939
 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=_ve959bwH8usZ6dPlpXxi2on9t0hLpcmhRWnaWHCWMQ,11745
 sagemaker_core/tools/shapes_extractor.py,sha256=GFy55JHGW0V8cwN5SL5gGLUxihLGswueyh5iNCn1sYk,12310
 sagemaker_core/tools/templates.py,sha256=yX2RQKeClgYwKS5Qu_mDpnWJIBCuj0yELrdm95aiTpk,23262
-sagemaker_core-1.0.26.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.26.dist-info/METADATA,sha256=u25aq9QHZzKg-YOhmviG8Gzcfq9-XLencvlclmikFbE,4885
-sagemaker_core-1.0.26.dist-info/WHEEL,sha256=DK49LOLCYiurdXXOXwGJm6U4DkHkg4lcxjhqwRa0CP4,91
-sagemaker_core-1.0.26.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.26.dist-info/RECORD,,
+sagemaker_core-1.0.28.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sagemaker_core-1.0.28.dist-info/METADATA,sha256=Lj262X_K44xPQGa6JyHPdVbtzKNlzFAIhBqneT9M79A,4885
+sagemaker_core-1.0.28.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+sagemaker_core-1.0.28.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
+sagemaker_core-1.0.28.dist-info/RECORD,,

{sagemaker_core-1.0.26.dist-info → sagemaker_core-1.0.28.dist-info}/WHEEL

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