sagemaker-core 1.0.34__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"},
             {

sagemaker_core/main/resources.py

@@ -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()

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.34.dist-info → sagemaker_core-1.0.35.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.34
+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.34.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=0xbgQy8-PH6vzlYyaGzwyo3A_Nij8WedvXYQNy4MFvo,1427888
+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=qlfWAkFaAilQXGIIzKymRbGaeRYUdEM_Wk5QFV_kbIA,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.34.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sagemaker_core-1.0.34.dist-info/METADATA,sha256=McgLTFJ0c7js-mQ3bRNxSBxggXoU-eFz1yQY6SUp8To,4885
-sagemaker_core-1.0.34.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-sagemaker_core-1.0.34.dist-info/top_level.txt,sha256=R3GAZZ1zC5JxqdE_0x2Lu_WYi2Xfke7VsiP3L5zngfA,15
-sagemaker_core-1.0.34.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.34.dist-info → sagemaker_core-1.0.35.dist-info}/WHEEL

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