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

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

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

@@ -0,0 +1 @@
+1.0.8

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

@@ -6573,7 +6573,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 +13152,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",
     },
@@ -14701,6 +14724,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.8}/src/sagemaker_core/main/resources.py

@@ -16518,6 +16518,7 @@ class Model(Base):
                         "s3_data_source": {
                             "s3_uri": {"type": "string"},
                             "s3_data_type": {"type": "string"},
+                            "manifest_s3_uri": {"type": "string"},
                         }
                     }
                 },
@@ -27953,6 +27954,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.8}/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.8/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.8
 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