sagemaker-core 1.0.33__py3-none-any.whl → 1.0.35__py3-none-any.whl

sagemaker_core/main/code_injection/codec.py

@@ -22,6 +22,7 @@ from sagemaker_core.main.code_injection.constants import (
     LIST_TYPE,
     MAP_TYPE,
 )
+from io import BytesIO
 
 
 def pascal_to_snake(pascal_str):
@@ -244,6 +245,15 @@ def transform(data, shape, object_instance=None) -> dict:
         elif _member_type == MAP_TYPE:
             _map_type_shape = SHAPE_DAG[_member_shape]
             evaluated_value = _evaluate_map_type(data[_member_name], _map_type_shape)
+        elif _member_type == "blob":
+            blob_data = data[_member_name]
+            if isinstance(blob_data, bytes):
+                evaluated_value = BytesIO(blob_data)
+            elif hasattr(blob_data, "read"):
+                # If it's already a file-like object, use it as is
+                evaluated_value = blob_data
+            else:
+                raise ValueError(f"Unexpected blob data type: {type(blob_data)}")
         else:
             raise ValueError(f"Unexpected member type encountered: {_member_type}")
 

sagemaker_core/main/code_injection/shape_dag.py

@@ -5098,6 +5098,11 @@ SHAPE_DAG = {
             {"name": "MlflowVersion", "shape": "MlflowVersion", "type": "string"},
             {"name": "RoleArn", "shape": "RoleArn", "type": "string"},
             {"name": "TrackingServerStatus", "shape": "TrackingServerStatus", "type": "string"},
+            {
+                "name": "TrackingServerMaintenanceStatus",
+                "shape": "TrackingServerMaintenanceStatus",
+                "type": "string",
+            },
             {"name": "IsActive", "shape": "IsTrackingServerActive", "type": "string"},
             {"name": "TrackingServerUrl", "shape": "TrackingServerUrl", "type": "string"},
             {
@@ -10291,9 +10296,9 @@ SHAPE_DAG = {
     "MetricDatum": {
         "members": [
             {"name": "MetricName", "shape": "AutoMLMetricEnum", "type": "string"},
+            {"name": "StandardMetricName", "shape": "AutoMLMetricExtendedEnum", "type": "string"},
             {"name": "Value", "shape": "Float", "type": "float"},
             {"name": "Set", "shape": "MetricSetSource", "type": "string"},
-            {"name": "StandardMetricName", "shape": "AutoMLMetricExtendedEnum", "type": "string"},
         ],
         "type": "structure",
     },

sagemaker_core/main/resources.py

@@ -6476,7 +6476,6 @@ class DataQualityJobDefinition(Base):
             "monitoring_job_definition_name": "job_definition_name",
             "monitoring_job_definition_arn": "job_definition_arn",
         }
-
         # serialize the input request
         operation_input_args = serialize(operation_input_args)
         logger.debug(f"Serialized input request: {operation_input_args}")
@@ -7369,6 +7368,7 @@ class Domain(Base):
                         "domain_execution_role_arn": {"type": "string"}
                     },
                     "execution_role_identity_config": {"type": "string"},
+                    "unified_studio_settings": {"project_s3_path": {"type": "string"}},
                 },
                 "home_efs_file_system_kms_key_id": {"type": "string"},
                 "subnet_ids": {"type": "array", "items": {"type": "string"}},
@@ -17364,6 +17364,7 @@ class MlflowTrackingServer(Base):
         mlflow_version: The MLflow version used for the described tracking server.
         role_arn: The Amazon Resource Name (ARN) for an IAM role in your account that the described MLflow Tracking Server uses to access the artifact store in Amazon S3.
         tracking_server_status: The current creation status of the described MLflow Tracking Server.
+        tracking_server_maintenance_status:  The current maintenance status of the described MLflow Tracking Server.
         is_active: Whether the described MLflow Tracking Server is currently active.
         tracking_server_url: The URL to connect to the MLflow user interface for the described tracking server.
         weekly_maintenance_window_start: The day and time of the week when weekly maintenance occurs on the described tracking server.
@@ -17382,6 +17383,7 @@ class MlflowTrackingServer(Base):
     mlflow_version: Optional[str] = Unassigned()
     role_arn: Optional[str] = Unassigned()
     tracking_server_status: Optional[str] = Unassigned()
+    tracking_server_maintenance_status: Optional[str] = Unassigned()
     is_active: Optional[str] = Unassigned()
     tracking_server_url: Optional[str] = Unassigned()
     weekly_maintenance_window_start: Optional[str] = Unassigned()
@@ -18664,7 +18666,6 @@ class ModelBiasJobDefinition(Base):
             "monitoring_job_definition_name": "job_definition_name",
             "monitoring_job_definition_arn": "job_definition_arn",
         }
-
         # serialize the input request
         operation_input_args = serialize(operation_input_args)
         logger.debug(f"Serialized input request: {operation_input_args}")
@@ -19895,7 +19896,6 @@ class ModelExplainabilityJobDefinition(Base):
             "monitoring_job_definition_name": "job_definition_name",
             "monitoring_job_definition_arn": "job_definition_arn",
         }
-
         # serialize the input request
         operation_input_args = serialize(operation_input_args)
         logger.debug(f"Serialized input request: {operation_input_args}")
@@ -21514,7 +21514,6 @@ class ModelQualityJobDefinition(Base):
             "monitoring_job_definition_name": "job_definition_name",
             "monitoring_job_definition_arn": "job_definition_arn",
         }
-
         # serialize the input request
         operation_input_args = serialize(operation_input_args)
         logger.debug(f"Serialized input request: {operation_input_args}")

sagemaker_core/main/shapes.py

@@ -1743,15 +1743,15 @@ class MetricDatum(Base):
     Attributes
     ----------------------
     metric_name: The name of the metric.
+    standard_metric_name: The name of the standard metric.   For definitions of the standard metrics, see  Autopilot candidate metrics .
     value: The value of the metric.
     set: The dataset split from which the AutoML job produced the metric.
-    standard_metric_name: The name of the standard metric.   For definitions of the standard metrics, see  Autopilot candidate metrics .
     """
 
     metric_name: Optional[str] = Unassigned()
+    standard_metric_name: Optional[str] = Unassigned()
     value: Optional[float] = Unassigned()
     set: Optional[str] = Unassigned()
-    standard_metric_name: Optional[str] = Unassigned()
 
 
 class CandidateProperties(Base):

sagemaker_core/main/utils.py

@@ -457,13 +457,13 @@ class ResourceIterator(Generic[T]):
         elif (
             len(self.summary_list) > 0
             and self.index >= len(self.summary_list)
-            and self.next_token is None
+            and (not self.next_token)
         ):
             raise StopIteration
 
         # Otherwise, get the next page of summaries by calling the list method with the next token if available
         else:
-            if self.next_token is not None:
+            if self.next_token:
                 response = getattr(self.client, self.list_method)(
                     NextToken=self.next_token, **self.list_method_kwargs
                 )

{sagemaker_core-1.0.33.dist-info → sagemaker_core-1.0.35.dist-info}/METADATA

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

@@ -7,15 +7,15 @@ sagemaker_core/main/config_schema.py,sha256=Wxe2gJash1rrxBomGhSYmII1LmJ3E70LIuSW
 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=jtD_9MUoyJkTDGAAnvEVpaconzPF6kRFBdio_mN_3zs,1427804
-sagemaker_core/main/shapes.py,sha256=YfcsZIpYetJPZHNo-VfajMCaqT6eGYXI_1EvCLr0dGg,739047
+sagemaker_core/main/resources.py,sha256=dNDFf5ANJ8NrVd5sy3Onr_W4Ls-0UEw7RsER7vyLOg8,1428074
+sagemaker_core/main/shapes.py,sha256=JuLWvrd3YSaVMnmERMB0jOPLtpAu98O5icSf7BwBpVU,739047
 sagemaker_core/main/user_agent.py,sha256=BPYDAfDd70ObP-VAjl7aDHALHyGknkpRP21ktVr_LDw,2744
-sagemaker_core/main/utils.py,sha256=tJyHjHFwwmz8LHMlOLf5_Exu4h-kFVnAxvXkbChkuHQ,19215
+sagemaker_core/main/utils.py,sha256=Ge_KdFKWo-qoIXlXkgLbaJbfXMVdUVHTq6BbDQn48Tw,19201
 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/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=v4admWbDBIaum364qkuvq76WM8eOC0wOOfH8KpuA3Gc,704183
+sagemaker_core/main/code_injection/shape_dag.py,sha256=Kdjew_4KCxfAjauG-0fPaHtiA3Ttc8IGzpKcAnQZO7A,704365
 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.33.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.33.dist-info/METADATA,sha256=0AkeGwVa8WdSOps6dTQQZh0_KmS3uRLV_NWmH0PkuEM,4885
-sagemaker_core-1.0.33.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-sagemaker_core-1.0.33.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.33.dist-info/RECORD,,
+sagemaker_core-1.0.35.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sagemaker_core-1.0.35.dist-info/METADATA,sha256=NymLf-CKNE3RMT34LBnDeDLcGMBEfUrWOdSzsWLJ-P4,4885
+sagemaker_core-1.0.35.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sagemaker_core-1.0.35.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
+sagemaker_core-1.0.35.dist-info/RECORD,,

{sagemaker_core-1.0.33.dist-info → sagemaker_core-1.0.35.dist-info}/WHEEL

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