sagemaker-core 1.0.7__tar.gz → 1.0.9__tar.gz

{sagemaker_core-1.0.7/src/sagemaker_core.egg-info → sagemaker_core-1.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.7
+Version: 1.0.9
 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.9/VERSION

@@ -0,0 +1 @@
+1.0.9

{sagemaker_core-1.0.7 → sagemaker_core-1.0.9}/src/sagemaker_core/main/code_injection/shape_dag.py

@@ -1884,6 +1884,7 @@ SHAPE_DAG = {
                 "shape": "AppSecurityGroupManagement",
                 "type": "string",
             },
+            {"name": "TagPropagation", "shape": "TagPropagation", "type": "string"},
             {"name": "DefaultSpaceSettings", "shape": "DefaultSpaceSettings", "type": "structure"},
         ],
         "type": "structure",
@@ -4125,6 +4126,7 @@ SHAPE_DAG = {
                 "shape": "AppSecurityGroupManagement",
                 "type": "string",
             },
+            {"name": "TagPropagation", "shape": "TagPropagation", "type": "string"},
             {"name": "DefaultSpaceSettings", "shape": "DefaultSpaceSettings", "type": "structure"},
         ],
         "type": "structure",
@@ -6573,7 +6575,24 @@ SHAPE_DAG = {
     },
     "Groups": {"member_shape": "Group", "member_type": "string", "type": "list"},
     "HiddenAppTypesList": {"member_shape": "AppType", "member_type": "string", "type": "list"},
+    "HiddenInstanceTypesList": {
+        "member_shape": "AppInstanceType",
+        "member_type": "string",
+        "type": "list",
+    },
     "HiddenMlToolsList": {"member_shape": "MlTools", "member_type": "string", "type": "list"},
+    "HiddenSageMakerImage": {
+        "members": [
+            {"name": "SageMakerImageName", "shape": "SageMakerImageName", "type": "string"},
+            {"name": "VersionAliases", "shape": "VersionAliasesList", "type": "list"},
+        ],
+        "type": "structure",
+    },
+    "HiddenSageMakerImageVersionAliasesList": {
+        "member_shape": "HiddenSageMakerImage",
+        "member_type": "structure",
+        "type": "list",
+    },
     "HolidayConfig": {
         "member_shape": "HolidayConfigAttributes",
         "member_type": "structure",
@@ -13135,6 +13154,12 @@ SHAPE_DAG = {
         "members": [
             {"name": "HiddenMlTools", "shape": "HiddenMlToolsList", "type": "list"},
             {"name": "HiddenAppTypes", "shape": "HiddenAppTypesList", "type": "list"},
+            {"name": "HiddenInstanceTypes", "shape": "HiddenInstanceTypesList", "type": "list"},
+            {
+                "name": "HiddenSageMakerImageVersionAliases",
+                "shape": "HiddenSageMakerImageVersionAliasesList",
+                "type": "list",
+            },
         ],
         "type": "structure",
     },
@@ -14059,6 +14084,7 @@ SHAPE_DAG = {
             {"name": "DefaultSpaceSettings", "shape": "DefaultSpaceSettings", "type": "structure"},
             {"name": "SubnetIds", "shape": "Subnets", "type": "list"},
             {"name": "AppNetworkAccessType", "shape": "AppNetworkAccessType", "type": "string"},
+            {"name": "TagPropagation", "shape": "TagPropagation", "type": "string"},
         ],
         "type": "structure",
     },
@@ -14701,6 +14727,11 @@ SHAPE_DAG = {
         "members": [{"name": "Dimension", "shape": "Dimension", "type": "integer"}],
         "type": "structure",
     },
+    "VersionAliasesList": {
+        "member_shape": "ImageVersionAliasPattern",
+        "member_type": "string",
+        "type": "list",
+    },
     "Vertex": {
         "members": [
             {"name": "Arn", "shape": "AssociationEntityArn", "type": "string"},

{sagemaker_core-1.0.7 → sagemaker_core-1.0.9}/src/sagemaker_core/main/resources.py

@@ -6147,6 +6147,7 @@ class Domain(Base):
         vpc_id: The ID of the Amazon Virtual Private Cloud (VPC) that the domain uses for communication.
         kms_key_id: The Amazon Web Services KMS customer managed key used to encrypt the EFS volume attached to the domain.
         app_security_group_management: The entity that creates and manages the required security groups for inter-app communication in VPCOnly mode. Required when CreateDomain.AppNetworkAccessType is VPCOnly and DomainSettings.RStudioServerProDomainSettings.DomainExecutionRoleArn is provided.
+        tag_propagation: Indicates whether custom tag propagation is supported for the domain.
         default_space_settings: The default settings used to create a space.
 
     """
@@ -6172,6 +6173,7 @@ class Domain(Base):
     vpc_id: Optional[str] = Unassigned()
     kms_key_id: Optional[str] = Unassigned()
     app_security_group_management: Optional[str] = Unassigned()
+    tag_propagation: Optional[str] = Unassigned()
     default_space_settings: Optional[DefaultSpaceSettings] = Unassigned()
 
     def get_name(self) -> str:
@@ -6270,6 +6272,7 @@ class Domain(Base):
         home_efs_file_system_kms_key_id: Optional[str] = Unassigned(),
         kms_key_id: Optional[str] = Unassigned(),
         app_security_group_management: Optional[str] = Unassigned(),
+        tag_propagation: Optional[str] = Unassigned(),
         default_space_settings: Optional[DefaultSpaceSettings] = Unassigned(),
         session: Optional[Session] = None,
         region: Optional[str] = None,
@@ -6289,6 +6292,7 @@ class Domain(Base):
             home_efs_file_system_kms_key_id: Use KmsKeyId.
             kms_key_id: SageMaker uses Amazon Web Services KMS to encrypt EFS and EBS volumes attached to the domain with an Amazon Web Services managed key by default. For more control, specify a customer managed key.
             app_security_group_management: The entity that creates and manages the required security groups for inter-app communication in VPCOnly mode. Required when CreateDomain.AppNetworkAccessType is VPCOnly and DomainSettings.RStudioServerProDomainSettings.DomainExecutionRoleArn is provided. If setting up the domain for use with RStudio, this value must be set to Service.
+            tag_propagation: Indicates whether custom tag propagation is supported for the domain. Defaults to DISABLED.
             default_space_settings: The default settings used to create a space.
             session: Boto3 session.
             region: Region name.
@@ -6330,6 +6334,7 @@ class Domain(Base):
             "HomeEfsFileSystemKmsKeyId": home_efs_file_system_kms_key_id,
             "KmsKeyId": kms_key_id,
             "AppSecurityGroupManagement": app_security_group_management,
+            "TagPropagation": tag_propagation,
             "DefaultSpaceSettings": default_space_settings,
         }
 
@@ -6446,6 +6451,7 @@ class Domain(Base):
         default_space_settings: Optional[DefaultSpaceSettings] = Unassigned(),
         subnet_ids: Optional[List[str]] = Unassigned(),
         app_network_access_type: Optional[str] = Unassigned(),
+        tag_propagation: Optional[str] = Unassigned(),
     ) -> Optional["Domain"]:
         """
         Update a Domain resource
@@ -6482,6 +6488,7 @@ class Domain(Base):
             "DefaultSpaceSettings": default_space_settings,
             "SubnetIds": subnet_ids,
             "AppNetworkAccessType": app_network_access_type,
+            "TagPropagation": tag_propagation,
         }
         logger.debug(f"Input request: {operation_input_args}")
         # serialize the input request
@@ -16518,6 +16525,7 @@ class Model(Base):
                         "s3_data_source": {
                             "s3_uri": {"type": "string"},
                             "s3_data_type": {"type": "string"},
+                            "manifest_s3_uri": {"type": "string"},
                         }
                     }
                 },
@@ -27953,6 +27961,55 @@ class TrialComponent(Base):
         response = client.batch_put_metrics(**operation_input_args)
         logger.debug(f"Response: {response}")
 
+    @classmethod
+    @Base.add_validate_call
+    def batch_get_metrics(
+        cls,
+        metric_queries: List[MetricQuery],
+        session: Optional[Session] = None,
+        region: Optional[str] = None,
+    ) -> Optional[BatchGetMetricsResponse]:
+        """
+        Used to retrieve training metrics from SageMaker.
+
+        Parameters:
+            metric_queries: Queries made to retrieve training metrics from SageMaker.
+            session: Boto3 session.
+            region: Region name.
+
+        Returns:
+            BatchGetMetricsResponse
+
+        Raises:
+            botocore.exceptions.ClientError: This exception is raised for AWS service related errors.
+                The error message and error code can be parsed from the exception as follows:
+                ```
+                try:
+                    # AWS service call here
+                except botocore.exceptions.ClientError as e:
+                    error_message = e.response['Error']['Message']
+                    error_code = e.response['Error']['Code']
+                ```
+        """
+
+        operation_input_args = {
+            "MetricQueries": metric_queries,
+        }
+        # 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="sagemaker-metrics"
+        )
+
+        logger.debug(f"Calling batch_get_metrics API")
+        response = client.batch_get_metrics(**operation_input_args)
+        logger.debug(f"Response: {response}")
+
+        transformed_response = transform(response, "BatchGetMetricsResponse")
+        return BatchGetMetricsResponse(**transformed_response)
+
 
 class UserProfile(Base):
     """

{sagemaker_core-1.0.7 → sagemaker_core-1.0.9}/src/sagemaker_core/main/shapes.py

@@ -406,6 +406,18 @@ class MetricQueryResult(Base):
     message: Optional[str] = Unassigned()
 
 
+class BatchGetMetricsResponse(Base):
+    """
+    BatchGetMetricsResponse
+
+    Attributes
+    ----------------------
+    metric_query_results: The results of a query to retrieve training metrics from SageMaker.
+    """
+
+    metric_query_results: Optional[List[MetricQueryResult]] = Unassigned()
+
+
 class BatchPutMetricsError(Base):
     """
     BatchPutMetricsError
@@ -4304,6 +4316,21 @@ class CustomFileSystemConfig(Base):
     efs_file_system_config: Optional[EFSFileSystemConfig] = Unassigned()
 
 
+class HiddenSageMakerImage(Base):
+    """
+    HiddenSageMakerImage
+      The SageMaker images that are hidden from the Studio user interface. You must specify the SageMaker image name and version aliases.
+
+    Attributes
+    ----------------------
+    sage_maker_image_name:  The SageMaker image name that you are hiding from the Studio user interface.
+    version_aliases:  The version aliases you are hiding from the Studio user interface.
+    """
+
+    sage_maker_image_name: Optional[str] = Unassigned()
+    version_aliases: Optional[List[str]] = Unassigned()
+
+
 class StudioWebPortalSettings(Base):
     """
     StudioWebPortalSettings
@@ -4313,10 +4340,14 @@ class StudioWebPortalSettings(Base):
     ----------------------
     hidden_ml_tools: The machine learning tools that are hidden from the Studio left navigation pane.
     hidden_app_types: The Applications supported in Studio that are hidden from the Studio left navigation pane.
+    hidden_instance_types:  The instance types you are hiding from the Studio user interface.
+    hidden_sage_maker_image_version_aliases:  The version aliases you are hiding from the Studio user interface.
     """
 
     hidden_ml_tools: Optional[List[str]] = Unassigned()
     hidden_app_types: Optional[List[str]] = Unassigned()
+    hidden_instance_types: Optional[List[str]] = Unassigned()
+    hidden_sage_maker_image_version_aliases: Optional[List[HiddenSageMakerImage]] = Unassigned()
 
 
 class UserSettings(Base):
@@ -5849,15 +5880,15 @@ class HumanTaskConfig(Base):
 
     workteam_arn: str
     ui_config: UiConfig
-    pre_human_task_lambda_arn: str
     task_title: str
     task_description: str
     number_of_human_workers_per_data_object: int
     task_time_limit_in_seconds: int
-    annotation_consolidation_config: AnnotationConsolidationConfig
+    pre_human_task_lambda_arn: Optional[str] = Unassigned()
     task_keywords: Optional[List[str]] = Unassigned()
     task_availability_lifetime_in_seconds: Optional[int] = Unassigned()
     max_concurrent_task_count: Optional[int] = Unassigned()
+    annotation_consolidation_config: Optional[AnnotationConsolidationConfig] = Unassigned()
     public_workforce_task_price: Optional[PublicWorkforceTaskPrice] = Unassigned()
 
 
@@ -9979,7 +10010,7 @@ class LabelingJobSummary(Base):
     labeling_job_status: str
     label_counters: LabelCounters
     workteam_arn: str
-    pre_human_task_lambda_arn: str
+    pre_human_task_lambda_arn: Optional[str] = Unassigned()
     annotation_consolidation_lambda_arn: Optional[str] = Unassigned()
     failure_reason: Optional[str] = Unassigned()
     labeling_job_output: Optional[LabelingJobOutput] = Unassigned()

{sagemaker_core-1.0.7 → sagemaker_core-1.0.9/src/sagemaker_core.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sagemaker-core
-Version: 1.0.7
+Version: 1.0.9
 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.7/VERSION

@@ -1 +0,0 @@
-1.0.7