sagemaker-core 1.0.52__py3-none-any.whl → 1.0.54__py3-none-any.whl

sagemaker_core/main/code_injection/shape_dag.py

@@ -1447,7 +1447,9 @@ SHAPE_DAG = {
     },
     "ClusterEbsVolumeConfig": {
         "members": [
-            {"name": "VolumeSizeInGB", "shape": "ClusterEbsVolumeSizeInGB", "type": "integer"}
+            {"name": "VolumeSizeInGB", "shape": "ClusterEbsVolumeSizeInGB", "type": "integer"},
+            {"name": "VolumeKmsKeyId", "shape": "KmsKeyId", "type": "string"},
+            {"name": "RootVolume", "shape": "Boolean", "type": "boolean"},
         ],
         "type": "structure",
     },

sagemaker_core/main/shapes.py

@@ -3414,9 +3414,13 @@ class ClusterEbsVolumeConfig(Base):
     Attributes
     ----------------------
     volume_size_in_gb: The size in gigabytes (GB) of the additional EBS volume to be attached to the instances in the SageMaker HyperPod cluster instance group. The additional EBS volume is attached to each instance within the SageMaker HyperPod cluster instance group and mounted to /opt/sagemaker.
+    volume_kms_key_id: The ID of a KMS key to encrypt the Amazon EBS volume.
+    root_volume: Specifies whether the configuration is for the cluster's root or secondary Amazon EBS volume. You can specify two ClusterEbsVolumeConfig fields to configure both the root and secondary volumes. Set the value to True if you'd like to provide your own customer managed Amazon Web Services KMS key to encrypt the root volume. When True:   The configuration is applied to the root volume.   You can't specify the VolumeSizeInGB field. The size of the root volume is determined for you.   You must specify a KMS key ID for VolumeKmsKeyId to encrypt the root volume with your own KMS key instead of an Amazon Web Services owned KMS key.   Otherwise, by default, the value is False, and the following applies:   The configuration is applied to the secondary volume, while the root volume is encrypted with an Amazon Web Services owned key.   You must specify the VolumeSizeInGB field.   You can optionally specify the VolumeKmsKeyId to encrypt the secondary volume with your own KMS key instead of an Amazon Web Services owned KMS key.
     """
 
     volume_size_in_gb: Optional[int] = Unassigned()
+    volume_kms_key_id: Optional[str] = Unassigned()
+    root_volume: Optional[bool] = Unassigned()
 
 
 class ClusterMetadata(Base):
@@ -5232,8 +5236,8 @@ class S3FileSystemConfig(Base):
     s3_uri: The Amazon S3 URI of the S3 file system configuration.
     """
 
+    s3_uri: str
     mount_path: Optional[str] = Unassigned()
-    s3_uri: Optional[str] = Unassigned()
 
 
 class CustomFileSystemConfig(Base):
@@ -8122,7 +8126,7 @@ class S3FileSystem(Base):
     s3_uri: The Amazon S3 URI that specifies the location in S3 where files are stored, which is mounted within the Studio environment. For example: s3://<bucket-name>/<prefix>/.
     """
 
-    s3_uri: Optional[str] = Unassigned()
+    s3_uri: str
 
 
 class CustomFileSystem(Base):

{sagemaker_core-1.0.52.dist-info → sagemaker_core-1.0.54.dist-info}/METADATA

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

@@ -8,14 +8,14 @@ sagemaker_core/main/default_configs_helper.py,sha256=bg_tgczX_bYzNiSlalJ6TWPTgrQ
 sagemaker_core/main/exceptions.py,sha256=CsiM3V_Gb16grBotnu59LB6tznryPcSvAQDAOOYGc10,5563
 sagemaker_core/main/logs.py,sha256=yfEH7uP91nbE1lefymOlBr81ziBzsDSIOF2Qyd54FJE,6241
 sagemaker_core/main/resources.py,sha256=fjJ7xwqlx0J5_8EH9h-_HDBW20Kyg4UVfF0Ys2tMJ88,1441521
-sagemaker_core/main/shapes.py,sha256=b60sytBhfwqnWY8kt937eDsxVOKymAXMapGOywumReo,784703
+sagemaker_core/main/shapes.py,sha256=XMC6sxYGkIh1UaCM9Hz5S1Ar9WyDSvmwmg-5GKRCMw4,785875
 sagemaker_core/main/user_agent.py,sha256=BPYDAfDd70ObP-VAjl7aDHALHyGknkpRP21ktVr_LDw,2744
 sagemaker_core/main/utils.py,sha256=y1aZzztCApczZSqT0U7_H8zLlpqp1vgoUKcp40mTM2o,19157
 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=2YzJ-iYEzmguVaJVcZeyCR0OpTSR7UOixATrOm4MiBk,8885
 sagemaker_core/main/code_injection/constants.py,sha256=2ICExGge8vAWx7lSTW0JGh-bH1korkvpOpDu5M63eI4,980
-sagemaker_core/main/code_injection/shape_dag.py,sha256=-YYPaCCyGRf-uJXgb6gy313zrIINzVDCBkvEW_7QvEA,745726
+sagemaker_core/main/code_injection/shape_dag.py,sha256=yU5WZX2fjyFe_pNnxOiAWh-hODQPP5pJNmPKuU_xSkU,745881
 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=0lOIH3Rq2CXWkQhK6VenN_TE_v5p852s2kQyb_BeQxA,23460
-sagemaker_core-1.0.52.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.52.dist-info/METADATA,sha256=AxcL3_QLQTtipy1Blv_7HNACjWdc0is5Di3jDWd8vyQ,4871
-sagemaker_core-1.0.52.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sagemaker_core-1.0.52.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.52.dist-info/RECORD,,
+sagemaker_core-1.0.54.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sagemaker_core-1.0.54.dist-info/METADATA,sha256=cjIe5fIpS_Q7gWb9CPhq1sZ0hHnaPtW9WBBwVnYf87c,4871
+sagemaker_core-1.0.54.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sagemaker_core-1.0.54.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
+sagemaker_core-1.0.54.dist-info/RECORD,,