sagemaker-core 1.0.34__tar.gz → 1.0.36__tar.gz

{sagemaker_core-1.0.34/src/sagemaker_core.egg-info → sagemaker_core-1.0.36}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.34
+Version: 1.0.36
 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.36/VERSION

@@ -0,0 +1 @@
+1.0.36

{sagemaker_core-1.0.34 → sagemaker_core-1.0.36}/src/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-1.0.34 → sagemaker_core-1.0.36}/src/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"},
             {
@@ -6291,6 +6296,24 @@ SHAPE_DAG = {
         ],
         "type": "structure",
     },
+    "Ec2CapacityReservation": {
+        "members": [
+            {
+                "name": "Ec2CapacityReservationId",
+                "shape": "Ec2CapacityReservationId",
+                "type": "string",
+            },
+            {"name": "TotalInstanceCount", "shape": "TaskCount", "type": "integer"},
+            {"name": "AvailableInstanceCount", "shape": "TaskCount", "type": "integer"},
+            {"name": "UsedByCurrentEndpoint", "shape": "TaskCount", "type": "integer"},
+        ],
+        "type": "structure",
+    },
+    "Ec2CapacityReservationsList": {
+        "member_shape": "Ec2CapacityReservation",
+        "member_type": "structure",
+        "type": "list",
+    },
     "Edge": {
         "members": [
             {"name": "SourceArn", "shape": "AssociationEntityArn", "type": "string"},
@@ -12388,6 +12411,41 @@ SHAPE_DAG = {
                 "shape": "ProductionVariantInferenceAmiVersion",
                 "type": "string",
             },
+            {
+                "name": "CapacityReservationConfig",
+                "shape": "ProductionVariantCapacityReservationConfig",
+                "type": "structure",
+            },
+        ],
+        "type": "structure",
+    },
+    "ProductionVariantCapacityReservationConfig": {
+        "members": [
+            {
+                "name": "CapacityReservationPreference",
+                "shape": "CapacityReservationPreference",
+                "type": "string",
+            },
+            {"name": "MlReservationArn", "shape": "MlReservationArn", "type": "string"},
+        ],
+        "type": "structure",
+    },
+    "ProductionVariantCapacityReservationSummary": {
+        "members": [
+            {"name": "MlReservationArn", "shape": "MlReservationArn", "type": "string"},
+            {
+                "name": "CapacityReservationPreference",
+                "shape": "CapacityReservationPreference",
+                "type": "string",
+            },
+            {"name": "TotalInstanceCount", "shape": "TaskCount", "type": "integer"},
+            {"name": "AvailableInstanceCount", "shape": "TaskCount", "type": "integer"},
+            {"name": "UsedByCurrentEndpoint", "shape": "TaskCount", "type": "integer"},
+            {
+                "name": "Ec2CapacityReservations",
+                "shape": "Ec2CapacityReservationsList",
+                "type": "list",
+            },
         ],
         "type": "structure",
     },
@@ -12488,6 +12546,11 @@ SHAPE_DAG = {
                 "shape": "ProductionVariantRoutingConfig",
                 "type": "structure",
             },
+            {
+                "name": "CapacityReservationConfig",
+                "shape": "ProductionVariantCapacityReservationSummary",
+                "type": "structure",
+            },
         ],
         "type": "structure",
     },
@@ -14828,6 +14891,11 @@ SHAPE_DAG = {
             {"name": "ProjectId", "shape": "UnifiedStudioProjectId", "type": "string"},
             {"name": "EnvironmentId", "shape": "UnifiedStudioEnvironmentId", "type": "string"},
             {"name": "ProjectS3Path", "shape": "S3Uri", "type": "string"},
+            {
+                "name": "SingleSignOnApplicationArn",
+                "shape": "SingleSignOnApplicationArn",
+                "type": "string",
+            },
         ],
         "type": "structure",
     },

{sagemaker_core-1.0.34 → sagemaker_core-1.0.36}/src/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-1.0.34 → sagemaker_core-1.0.36}/src/sagemaker_core/main/shapes.py

@@ -4779,6 +4779,7 @@ class UnifiedStudioSettings(Base):
     project_id: The ID of the Amazon SageMaker Unified Studio project that corresponds to the domain.
     environment_id: The ID of the environment that Amazon SageMaker Unified Studio associates with the domain.
     project_s3_path: The location where Amazon S3 stores temporary execution data and other artifacts for the project that corresponds to the domain.
+    single_sign_on_application_arn: The ARN of the application managed by SageMaker AI and SageMaker Unified Studio in the Amazon Web Services IAM Identity Center.
     """
 
     studio_web_portal_access: Optional[str] = Unassigned()
@@ -4788,6 +4789,7 @@ class UnifiedStudioSettings(Base):
     project_id: Optional[str] = Unassigned()
     environment_id: Optional[str] = Unassigned()
     project_s3_path: Optional[str] = Unassigned()
+    single_sign_on_application_arn: Optional[str] = Unassigned()
 
 
 class DomainSettings(Base):
@@ -4966,6 +4968,21 @@ class ProductionVariantRoutingConfig(Base):
     routing_strategy: str
 
 
+class ProductionVariantCapacityReservationConfig(Base):
+    """
+    ProductionVariantCapacityReservationConfig
+      Settings for the capacity reservation for the compute instances that SageMaker AI reserves for an endpoint.
+
+    Attributes
+    ----------------------
+    capacity_reservation_preference: Options that you can choose for the capacity reservation. SageMaker AI supports the following options:  capacity-reservations-only  SageMaker AI launches instances only into an ML capacity reservation. If no capacity is available, the instances fail to launch.
+    ml_reservation_arn: The Amazon Resource Name (ARN) that uniquely identifies the ML capacity reservation that SageMaker AI applies when it deploys the endpoint.
+    """
+
+    capacity_reservation_preference: Optional[str] = Unassigned()
+    ml_reservation_arn: Optional[str] = Unassigned()
+
+
 class ProductionVariant(Base):
     """
     ProductionVariant
@@ -4988,6 +5005,7 @@ class ProductionVariant(Base):
     managed_instance_scaling: Settings that control the range in the number of instances that the endpoint provisions as it scales up or down to accommodate traffic.
     routing_config: Settings that control how the endpoint routes incoming traffic to the instances that the endpoint hosts.
     inference_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. Amazon Web Services optimizes these configurations for different machine learning workloads. By selecting an AMI version, you can ensure that your inference environment is compatible with specific software requirements, such as CUDA driver versions, Linux kernel versions, or Amazon Web Services Neuron driver versions. The AMI version names, and their configurations, are the following:  al2-ami-sagemaker-inference-gpu-2    Accelerator: GPU   NVIDIA driver version: 535   CUDA version: 12.2    al2-ami-sagemaker-inference-gpu-2-1    Accelerator: GPU   NVIDIA driver version: 535   CUDA version: 12.2   NVIDIA Container Toolkit with disabled CUDA-compat mounting    al2-ami-sagemaker-inference-gpu-3-1    Accelerator: GPU   NVIDIA driver version: 550   CUDA version: 12.4   NVIDIA Container Toolkit with disabled CUDA-compat mounting    al2-ami-sagemaker-inference-neuron-2    Accelerator: Inferentia2 and Trainium   Neuron driver version: 2.19
+    capacity_reservation_config: Settings for the capacity reservation for the compute instances that SageMaker AI reserves for an endpoint.
     """
 
     variant_name: str
@@ -5005,6 +5023,7 @@ class ProductionVariant(Base):
     managed_instance_scaling: Optional[ProductionVariantManagedInstanceScaling] = Unassigned()
     routing_config: Optional[ProductionVariantRoutingConfig] = Unassigned()
     inference_ami_version: Optional[str] = Unassigned()
+    capacity_reservation_config: Optional[ProductionVariantCapacityReservationConfig] = Unassigned()
 
 
 class DataCaptureConfig(Base):
@@ -8164,6 +8183,48 @@ class ProductionVariantStatus(Base):
     start_time: Optional[datetime.datetime] = Unassigned()
 
 
+class Ec2CapacityReservation(Base):
+    """
+    Ec2CapacityReservation
+      The EC2 capacity reservations that are shared to an ML capacity reservation.
+
+    Attributes
+    ----------------------
+    ec2_capacity_reservation_id: The unique identifier for an EC2 capacity reservation that's part of the ML capacity reservation.
+    total_instance_count: The number of instances that you allocated to the EC2 capacity reservation.
+    available_instance_count: The number of instances that are currently available in the EC2 capacity reservation.
+    used_by_current_endpoint: The number of instances from the EC2 capacity reservation that are being used by the endpoint.
+    """
+
+    ec2_capacity_reservation_id: Optional[str] = Unassigned()
+    total_instance_count: Optional[int] = Unassigned()
+    available_instance_count: Optional[int] = Unassigned()
+    used_by_current_endpoint: Optional[int] = Unassigned()
+
+
+class ProductionVariantCapacityReservationSummary(Base):
+    """
+    ProductionVariantCapacityReservationSummary
+      Details about an ML capacity reservation.
+
+    Attributes
+    ----------------------
+    ml_reservation_arn: The Amazon Resource Name (ARN) that uniquely identifies the ML capacity reservation that SageMaker AI applies when it deploys the endpoint.
+    capacity_reservation_preference: The option that you chose for the capacity reservation. SageMaker AI supports the following options:  capacity-reservations-only  SageMaker AI launches instances only into an ML capacity reservation. If no capacity is available, the instances fail to launch.
+    total_instance_count: The number of instances that you allocated to the ML capacity reservation.
+    available_instance_count: The number of instances that are currently available in the ML capacity reservation.
+    used_by_current_endpoint: The number of instances from the ML capacity reservation that are being used by the endpoint.
+    ec2_capacity_reservations: The EC2 capacity reservations that are shared to this ML capacity reservation, if any.
+    """
+
+    ml_reservation_arn: Optional[str] = Unassigned()
+    capacity_reservation_preference: Optional[str] = Unassigned()
+    total_instance_count: Optional[int] = Unassigned()
+    available_instance_count: Optional[int] = Unassigned()
+    used_by_current_endpoint: Optional[int] = Unassigned()
+    ec2_capacity_reservations: Optional[List[Ec2CapacityReservation]] = Unassigned()
+
+
 class ProductionVariantSummary(Base):
     """
     ProductionVariantSummary
@@ -8182,6 +8243,7 @@ class ProductionVariantSummary(Base):
     desired_serverless_config: The serverless configuration requested for the endpoint update.
     managed_instance_scaling: Settings that control the range in the number of instances that the endpoint provisions as it scales up or down to accommodate traffic.
     routing_config: Settings that control how the endpoint routes incoming traffic to the instances that the endpoint hosts.
+    capacity_reservation_config: Settings for the capacity reservation for the compute instances that SageMaker AI reserves for an endpoint.
     """
 
     variant_name: str
@@ -8195,6 +8257,9 @@ class ProductionVariantSummary(Base):
     desired_serverless_config: Optional[ProductionVariantServerlessConfig] = Unassigned()
     managed_instance_scaling: Optional[ProductionVariantManagedInstanceScaling] = Unassigned()
     routing_config: Optional[ProductionVariantRoutingConfig] = Unassigned()
+    capacity_reservation_config: Optional[ProductionVariantCapacityReservationSummary] = (
+        Unassigned()
+    )
 
 
 class PendingProductionVariantSummary(Base):

{sagemaker_core-1.0.34 → sagemaker_core-1.0.36}/src/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 → sagemaker_core-1.0.36/src/sagemaker_core.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sagemaker-core
-Version: 1.0.34
+Version: 1.0.36
 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/VERSION

@@ -1 +0,0 @@
-1.0.34