sagemaker-core 1.0.1__py3-none-any.whl → 1.0.2__py3-none-any.whl

sagemaker_core/main/shapes.py

@@ -18,7 +18,7 @@ from sagemaker_core.main.utils import Unassigned
 
 
 class Base(BaseModel):
-    model_config = ConfigDict(protected_namespaces=(), validate_assignment=True)
+    model_config = ConfigDict(protected_namespaces=(), validate_assignment=True, extra="forbid")
 
 
 class ActionSource(Base):
@@ -3055,7 +3055,7 @@ class ContainerDefinition(Base):
     model_data_url: The S3 path where the model artifacts, which result from model training, are stored. This path must point to a single gzip compressed tar archive (.tar.gz suffix). The S3 path is required for SageMaker built-in algorithms, but not if you use your own algorithms. For more information on built-in algorithms, see Common Parameters.   The model artifacts must be in an S3 bucket that is in the same region as the model or endpoint you are creating.  If you provide a value for this parameter, SageMaker uses Amazon Web Services Security Token Service to download model artifacts from the S3 path you provide. Amazon Web Services STS is activated in your Amazon Web Services account by default. If you previously deactivated Amazon Web Services STS for a region, you need to reactivate Amazon Web Services STS for that region. For more information, see Activating and Deactivating Amazon Web Services STS in an Amazon Web Services Region in the Amazon Web Services Identity and Access Management User Guide.  If you use a built-in algorithm to create a model, SageMaker requires that you provide a S3 path to the model artifacts in ModelDataUrl.
     model_data_source: Specifies the location of ML model data to deploy.  Currently you cannot use ModelDataSource in conjunction with SageMaker batch transform, SageMaker serverless endpoints, SageMaker multi-model endpoints, and SageMaker Marketplace.
     additional_model_data_sources: Data sources that are available to your model in addition to the one that you specify for ModelDataSource when you use the CreateModel action.
-    environment: The environment variables to set in the Docker container. The maximum length of each key and value in the Environment map is 1024 bytes. The maximum length of all keys and values in the map, combined, is 32 KB. If you pass multiple containers to a CreateModel request, then the maximum length of all of their maps, combined, is also 32 KB.
+    environment: The environment variables to set in the Docker container. Don't include any sensitive data in your environment variables. The maximum length of each key and value in the Environment map is 1024 bytes. The maximum length of all keys and values in the map, combined, is 32 KB. If you pass multiple containers to a CreateModel request, then the maximum length of all of their maps, combined, is also 32 KB.
     model_package_name: The name or Amazon Resource Name (ARN) of the model package to use to create the model.
     inference_specification_name: The inference specification name in the model package version.
     multi_model_config: Specifies additional configuration for multi-model endpoints.
@@ -3856,6 +3856,7 @@ class UserSettings(Base):
     custom_posix_user_config: Details about the POSIX identity that is used for file system operations.
     custom_file_system_configs: The settings for assigning a custom file system to a user profile. Permitted users can access this file system in Amazon SageMaker Studio.
     studio_web_portal_settings: Studio settings. If these settings are applied on a user level, they take priority over the settings applied on a domain level.
+    auto_mount_home_efs: Indicates whether auto-mounting of an EFS volume is supported for the user profile. The DefaultAsDomain value is only supported for user profiles. Do not use the DefaultAsDomain value when setting this parameter for a domain.
     """
 
     execution_role: Optional[str] = Unassigned()
@@ -3875,6 +3876,7 @@ class UserSettings(Base):
     custom_posix_user_config: Optional[CustomPosixUserConfig] = Unassigned()
     custom_file_system_configs: Optional[List[CustomFileSystemConfig]] = Unassigned()
     studio_web_portal_settings: Optional[StudioWebPortalSettings] = Unassigned()
+    auto_mount_home_efs: Optional[str] = Unassigned()
 
 
 class RStudioServerProDomainSettings(Base):

sagemaker_core/main/utils.py

@@ -267,7 +267,7 @@ def pascal_to_snake(pascal_str):
 
 
 def is_not_primitive(obj):
-    return not isinstance(obj, (int, float, str, bool, complex))
+    return not isinstance(obj, (int, float, str, bool, datetime.datetime))
 
 
 def is_not_str_dict(obj):
@@ -438,6 +438,11 @@ class ResourceIterator(Generic[T]):
 
                 if self.custom_key_mapping:
                     init_data = {self.custom_key_mapping.get(k, k): v for k, v in init_data.items()}
+
+                # Filter out the fields that are not in the resource class
+                fields = self.resource_cls.__annotations__
+                init_data = {k: v for k, v in init_data.items() if k in fields}
+
                 resource_object = self.resource_cls(**init_data)
 
             # If the resource object has refresh method, refresh and return it

sagemaker_core/tools/resources_codegen.py

@@ -56,7 +56,6 @@ from sagemaker_core.tools.templates import (
     RESOURCE_BASE_CLASS_TEMPLATE,
     RETURN_ITERATOR_TEMPLATE,
     SERIALIZE_INPUT_TEMPLATE,
-    SERIALIZE_LIST_INPUT_TEMPLATE,
     STOP_METHOD_TEMPLATE,
     DELETE_METHOD_TEMPLATE,
     WAIT_FOR_DELETE_METHOD_TEMPLATE,
@@ -897,7 +896,8 @@ class ResourcesCodeGen:
                     resource_name=resource_name,
                     create_args=create_args,
                     resource_lower=resource_lower,
-                    service_name="sagemaker",  # TODO: change service name based on the service - runtime, sagemaker, etc.
+                    # TODO: change service name based on the service - runtime, sagemaker, etc.
+                    service_name="sagemaker",
                     operation_input_args=operation_input_args,
                     operation=operation,
                     get_args=get_args,
@@ -908,7 +908,8 @@ class ResourcesCodeGen:
                     resource_name=resource_name,
                     create_args=create_args,
                     resource_lower=resource_lower,
-                    service_name="sagemaker",  # TODO: change service name based on the service - runtime, sagemaker, etc.
+                    # TODO: change service name based on the service - runtime, sagemaker, etc.
+                    service_name="sagemaker",
                     operation_input_args=operation_input_args,
                     operation=operation,
                     get_args=get_args,
@@ -1074,7 +1075,8 @@ class ResourcesCodeGen:
             resource_name=resource_name,
             import_args=import_args,
             resource_lower=resource_lower,
-            service_name="sagemaker",  # TODO: change service name based on the service - runtime, sagemaker, etc.
+            # TODO: change service name based on the service - runtime, sagemaker, etc.
+            service_name="sagemaker",
             operation_input_args=operation_input_args,
             operation=operation,
             get_args=get_args,
@@ -1378,7 +1380,8 @@ class ResourcesCodeGen:
         formatted_method = GET_METHOD_TEMPLATE.format(
             docstring=docstring,
             resource_name=resource_name,
-            service_name="sagemaker",  # TODO: change service name based on the service - runtime, sagemaker, etc.
+            # TODO: change service name based on the service - runtime, sagemaker, etc.
+            service_name="sagemaker",
             describe_args=describe_args,
             resource_lower=resource_lower,
             operation_input_args=operation_input_args,
@@ -1686,7 +1689,7 @@ class ResourcesCodeGen:
 
         resource_iterator_args = ",\n".join(resource_iterator_args_list)
         resource_iterator_args = add_indent(resource_iterator_args, 8)
-        serialize_operation_input = SERIALIZE_LIST_INPUT_TEMPLATE.format(
+        serialize_operation_input = SERIALIZE_INPUT_TEMPLATE.format(
             operation_input_args=operation_input_args
         )
         initialize_client = INITIALIZE_CLIENT_TEMPLATE.format(service_name=method.service_name)

sagemaker_core/tools/templates.py

@@ -291,6 +291,10 @@ def get(
     operation_input_args = {{
 {operation_input_args}
     }}
+    # serialize the input request
+    operation_input_args = serialize(operation_input_args)
+    logger.debug(f"Serialized input request: {{operation_input_args}}")
+
     client = Base.get_sagemaker_client(session=session, region_name=region, service_name='{service_name}')
     response = client.{operation}(**operation_input_args)
 
@@ -312,6 +316,10 @@ def refresh(
     operation_input_args = {{
 {operation_input_args}
     }}
+    # serialize the input request
+    operation_input_args = serialize(operation_input_args)
+    logger.debug(f"Serialized input request: {{operation_input_args}}")
+
     client = Base.get_sagemaker_client()
     response = client.{operation}(**operation_input_args)
 
@@ -497,6 +505,10 @@ def delete(
     operation_input_args = {{
 {operation_input_args}
     }}
+    # serialize the input request
+    operation_input_args = serialize(operation_input_args)
+    logger.debug(f"Serialized input request: {{operation_input_args}}")
+
     client.{operation}(**operation_input_args)
     
     logger.info(f"Deleting {{self.__class__.__name__}} - {{self.get_name()}}")
@@ -511,6 +523,10 @@ def stop(self) -> None:
     operation_input_args = {{
 {operation_input_args}
     }}
+    # serialize the input request
+    operation_input_args = serialize(operation_input_args)
+    logger.debug(f"Serialized input request: {{operation_input_args}}")
+
     client.{operation}(**operation_input_args)
 
     logger.info(f"Stopping {{self.__class__.__name__}} - {{self.get_name()}}")
@@ -532,7 +548,9 @@ def get_all(
 {operation_input_args}
     }}
 {custom_key_mapping}
-    operation_input_args = {{k: v for k, v in operation_input_args.items() if v is not None and not isinstance(v, Unassigned)}}
+    # serialize the input request
+    operation_input_args = serialize(operation_input_args)
+    logger.debug(f"Serialized input request: {{operation_input_args}}")
     
     return ResourceIterator(
 {resource_iterator_args}
@@ -582,7 +600,9 @@ SERIALIZE_INPUT_TEMPLATE = """
     operation_input_args = {{
 {operation_input_args}
     }}
-    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}}")"""
 
 INITIALIZE_CLIENT_TEMPLATE = """
     client = Base.get_sagemaker_client(session=session, region_name=region, service_name='{service_name}')"""
@@ -604,13 +624,6 @@ DESERIALIZE_RESPONSE_TEMPLATE = """
 DESERIALIZE_RESPONSE_TO_BASIC_TYPE_TEMPLATE = """
     return list(response.values())[0]"""
 
-SERIALIZE_LIST_INPUT_TEMPLATE = """
-    operation_input_args = {{
-{operation_input_args}
-    }}
-    operation_input_args = {{k: v for k, v in operation_input_args.items() if v is not None and not isinstance(v, Unassigned)}}
-    logger.debug(f"Input request: {{operation_input_args}}")"""
-
 RETURN_ITERATOR_TEMPLATE = """
     return ResourceIterator(
 {resource_iterator_args}
@@ -622,7 +635,7 @@ DESERIALIZE_INPUT_AND_RESPONSE_TO_CLS_TEMPLATE = """
 
 RESOURCE_BASE_CLASS_TEMPLATE = """
 class Base(BaseModel):
-    model_config = ConfigDict(protected_namespaces=(), validate_assignment=True)
+    model_config = ConfigDict(protected_namespaces=(), validate_assignment=True, extra="forbid")
     
     @classmethod
     def get_sagemaker_client(cls, session = None, region_name = None, service_name = 'sagemaker'):
@@ -709,7 +722,7 @@ class Base(BaseModel):
 
 SHAPE_BASE_CLASS_TEMPLATE = """
 class {class_name}:
-    model_config = ConfigDict(protected_namespaces=(), validate_assignment=True)
+    model_config = ConfigDict(protected_namespaces=(), validate_assignment=True, extra="forbid")
 """
 
 SHAPE_CLASS_TEMPLATE = '''

{sagemaker_core-1.0.1.dist-info → sagemaker_core-1.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.1
+Version: 1.0.2
 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.1.dist-info → sagemaker_core-1.0.2.dist-info}/RECORD

@@ -6,15 +6,15 @@ sagemaker_core/main/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 sagemaker_core/main/config_schema.py,sha256=IDvL1YuMOwVg4nb0D4ysU51tc1UjlQASyz1QZoncs7I,56907
 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/resources.py,sha256=ttv1ikhwharh6M2Y3BQcyUIki_w2KgaXONri57NoWrI,1262774
-sagemaker_core/main/shapes.py,sha256=dBS_gTcdZWe3oQwgT0Jm5yW4xtXTaoe6W75vcP0kMnk,679075
+sagemaker_core/main/resources.py,sha256=_zix3PucNO60z3XVLEKCfSm5zWLyNWJaaGk868-rz5w,1298170
+sagemaker_core/main/shapes.py,sha256=c0B7r7_xz3m2CDc_Dmh1CjczmhCmwwzL8yaeLoWc7vs,679460
 sagemaker_core/main/user_agent.py,sha256=4sZybDXkzRoZnOnVDQ8p8zFTfiRJdsH7amDWInVQ4xU,2708
-sagemaker_core/main/utils.py,sha256=5FDSaXVi_GdluRfwhlZdVUfb23ROrLzaCG-deCJPDy0,18614
+sagemaker_core/main/utils.py,sha256=I7WsC-DdbnAZg_PtqcLPywXlqjAwj7ZYbeHhv93Z1O4,18840
 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=2DjmeD2uND307UqDefvVEpE0rZ8yfFU3Bi3TvQCQveI,7658
 sagemaker_core/main/code_injection/constants.py,sha256=2ICExGge8vAWx7lSTW0JGh-bH1korkvpOpDu5M63eI4,980
-sagemaker_core/main/code_injection/shape_dag.py,sha256=hqLwJzyLv7MHHr7wZudSn8zG3YbBKTqILUb5UVQCjC8,645192
+sagemaker_core/main/code_injection/shape_dag.py,sha256=oM_xL5qx9nQmWIWbtNurVsz6UXMfpIfrUyw7ZIc3Ry4,645281
 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
@@ -22,13 +22,13 @@ sagemaker_core/tools/codegen.py,sha256=mKWVi2pWnPxyIoWUEPYjEc9Gw7D9bCOrHqa00yzIZ
 sagemaker_core/tools/constants.py,sha256=PCyf5MCkWYL4dDmIMKZw8QD7FcF8HqYwK-gnJ1Fdno0,2989
 sagemaker_core/tools/data_extractor.py,sha256=NpP22uCaWgvn9uOHsFEkgtn3h2nOIKDvJTLihN01f1A,1368
 sagemaker_core/tools/method.py,sha256=4Hmu4UWpiBgUTZljYdW1KIKDduDxf_nfhCyuWgLVMWI,717
-sagemaker_core/tools/resources_codegen.py,sha256=CZTjshFdSek0zTYIMB3HxyGnPmqpixs2Q1XflBS50w0,89954
+sagemaker_core/tools/resources_codegen.py,sha256=m7B2vrkfokLmzND9tCkty7EhVMk1FLevTd2LmJd_GKg,89974
 sagemaker_core/tools/resources_extractor.py,sha256=B9Twlf2vb7AkZUKLnxSdQZ_rNLPz0ROk9kuq-6uB1As,14526
 sagemaker_core/tools/shapes_codegen.py,sha256=_ve959bwH8usZ6dPlpXxi2on9t0hLpcmhRWnaWHCWMQ,11745
 sagemaker_core/tools/shapes_extractor.py,sha256=4KjgDmhlPM4G1f1NeYbORKlXs1s7Q_sm_NK31S_ROQ0,11950
-sagemaker_core/tools/templates.py,sha256=1oMfDD1JBp2ImA6cQwc8w9CHBmAUH2WnEtefPNFwb34,24066
-sagemaker_core-1.0.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.1.dist-info/METADATA,sha256=Ze-1qi1eHRDx14GSKoXYzmBKnjLqjwlb_C9T0Gu99CU,4645
-sagemaker_core-1.0.1.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
-sagemaker_core-1.0.1.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.1.dist-info/RECORD,,
+sagemaker_core/tools/templates.py,sha256=IvL9ykLvOXDNCMnIZn30o-6Cmx4fw-BIJz3nl9LD4dc,24614
+sagemaker_core-1.0.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sagemaker_core-1.0.2.dist-info/METADATA,sha256=2CiwzWSqo3XyV_tlMy8ZWIsvq3Pzovrxx8SY4j9JJLY,4645
+sagemaker_core-1.0.2.dist-info/WHEEL,sha256=uCRv0ZEik_232NlR4YDw4Pv3Ajt5bKvMH13NUU7hFuI,91
+sagemaker_core-1.0.2.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
+sagemaker_core-1.0.2.dist-info/RECORD,,

{sagemaker_core-1.0.1.dist-info → sagemaker_core-1.0.2.dist-info}/WHEEL

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